tests_level_07_1.py

import unittest
import hedy
import sys
import io
import textwrap
from contextlib import contextmanager

@contextmanager
def captured_output():
    new_out, new_err = io.StringIO(), io.StringIO()
    old_out, old_err = sys.stdout, sys.stderr
    try:
        sys.stdout, sys.stderr = new_out, new_err
        yield sys.stdout, sys.stderr
    finally:
        sys.stdout, sys.stderr = old_out, old_err

def run_code(code):
    code = "import random\n" + code
    with captured_output() as (out, err):
        exec(code)
    return out.getvalue().strip()


class TestsLevel7_1(unittest.TestCase):
  def test_print(self):
    result = hedy.transpile("print 'ik heet'", 7, 1)
    self.assertEqual("print('ik heet')", result)

  def test_print_with_var(self):
    result = hedy.transpile("naam is Hedy\nprint 'ik heet' naam", 7, 1)
    self.assertEqual("naam = 'Hedy'\nprint('ik heet'+str(naam))", result)

  def test_print_with_calc_no_spaces(self):
    result = hedy.transpile("print '5 keer 5 is ' 5*5", 7, 1)
    self.assertEqual("print('5 keer 5 is '+str(int(5) * int(5)))", result)

  def test_print_calculation_times_directly(self):
    code = textwrap.dedent("""\
    nummer is 5
    nummertwee is 6
    print nummer * nummertwee""")

    result = hedy.transpile(code, 7, 1)

    expected = textwrap.dedent("""\
    nummer = '5'
    nummertwee = '6'
    print(str(int(nummer) * int(nummertwee)))""")

    self.assertEqual(expected, result)

    self.assertEqual("30", run_code(result))

  def test_transpile_ask(self):
    result = hedy.transpile("antwoord is ask wat is je lievelingskleur?", 7, 1)
    self.assertEqual(result, "antwoord = input('wat is je lievelingskleur?')")

  def test_if_with_indent(self):
    code = textwrap.dedent("""\
    naam is Hedy
    if naam is Hedy
        print 'koekoek'""")
    expected = textwrap.dedent("""\
    naam = 'Hedy'
    if str(naam) == str('Hedy'):
      print('koekoek')""")
    result = hedy.transpile(code, 7, 1)

    self.assertEqual(expected, result)

  def test_if_else(self):
    code = textwrap.dedent("""\
    antwoord is ask Hoeveel is 10 plus 10?
    if antwoord is 20
        print 'Goedzo!'
        print 'Het antwoord was inderdaad ' antwoord
    else
        print 'Foutje'
        print 'Het antwoord moest zijn ' antwoord""")

    expected = textwrap.dedent("""\
    antwoord = input('Hoeveel is 10 plus 10?')
    if str(antwoord) == str('20'):
      print('Goedzo!')
      print('Het antwoord was inderdaad '+str(antwoord))
    else:
      print('Foutje')
      print('Het antwoord moest zijn '+str(antwoord))""")

    result = hedy.transpile(code, 7, 1)

    self.assertEqual(expected, result)

  def test_print_random(self):
    code = textwrap.dedent("""\
    keuzes is steen, schaar, papier
    computerkeuze is keuzes at random
    print 'computer koos ' computerkeuze""")
    expected = textwrap.dedent("""\
    keuzes = ['steen', 'schaar', 'papier']
    computerkeuze=random.choice(keuzes)
    print('computer koos '+str(computerkeuze))""")
    result = hedy.transpile(code, 7, 1)

    self.assertEqual(expected, result)

  def test_for_loop(self):
    code = textwrap.dedent("""\
    a is 2
    a is 3
    for a is 2 to 4
      a is a + 2
      b is b + 2""")
    expected = textwrap.dedent("""\
    a = '2'
    a = '3'
    for a in range(int(2), int(4)+1):
      a = int(a) + int(2)
      b = int(b) + int(2)""")

    result = hedy.transpile(code, 7, 1)

    self.assertEqual(expected, result)

  def test_if__else(self):
    code = textwrap.dedent("""\
    a is 5
    if a is 1
      x is 2
    else
      x is 222""")
    expected = textwrap.dedent("""\
    a = '5'
    if str(a) == str('1'):
      x = '2'
    else:
      x = '222'""")

    result = hedy.transpile(code, 7, 1)

    self.assertEqual(expected, result)

  def test_forloop(self):
    code = textwrap.dedent("""\
    for i is 1 to 10
      print i
    print 'wie niet weg is is gezien'""")
    expected = textwrap.dedent("""\
    for i in range(int(1), int(10)+1):
      print(str(i))
    print('wie niet weg is is gezien')""")

    result = hedy.transpile(code, 7, 1)

    self.assertEqual(expected, result)


#  def test_if_under_else_in_for(self):
#    code = textwrap.dedent("""\
#    for i is 0 to 10
#      antwoord is ask Wat is 5*5
#      if antwoord is 24
#          print 'Dat is fout!'
#      else
#          print 'Dat is goed!'
#      if antwoord is 25
#          i is 10""")
#
#    expected = textwrap.dedent("""\
#    for i in range(int(0), int(10)+1):
#      antwoord = input('Wat is 5*5')
#      if str(antwoord) == str('24'):
#        print('Dat is fout!')
#      else:
#        print('Dat is goed!')
#      if str(antwoord) == str('25'):
#        i = '10'""")
#
#    result = hedy.transpile(code, 7, 1)
#
#    self.assertEqual(expected, result)




#programs with issues to see if we catch them properly
# (so this should fail, for now)
# at one point we want a real "Indent" error and a better error message
# for this!

  # def test_level_7_no_indentation(self):
  #   #test that we get a parse error here
  #   code = textwrap.dedent("""\
  #   antwoord is ask Hoeveel is 10 keer tien?
  #   if antwoord is 100
  #   print 'goed zo'
  #   else
  #   print 'bah slecht'""")
  #
  #   with self.assertRaises(Exception) as context:
  #     result = hedy.transpile(code, 7, 1)
  #   self.assertEqual(str(context.exception), 'Parse')