tests.py

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

#this code let's us capture std out to also execute the generated Python
# and check its output
@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):
    with captured_output() as (out, err):
        exec(code)
    return out.getvalue().strip()


class TestsForMultipleLevels(unittest.TestCase):
    max_level = 7

    def test_print_with_list_var_random(self):
        min_level = 2
        max_level = 5
        for i in range(min_level, max_level + 1):
            result = hedy.transpile("dieren is Hond, Kat, Kangoeroe\nprint dieren at random", i)
            self.assertEqual(result, "import random\ndieren = ['Hond', 'Kat', 'Kangoeroe']\nprint(random.choice(dieren))")
            self.assertIn(run_code(result), ['Hond', 'Kat', 'Kangoeroe'])
            print('Passed at level ', i)

        min_level = 6
        max_level = 7
        for i in range(min_level, max_level + 1):
            result = hedy.transpile("dieren is Hond, Kat, Kangoeroe\nprint dieren at random", i)
            self.assertEqual(result, "import random\ndieren = ['Hond', 'Kat', 'Kangoeroe']\nprint(str(random.choice(dieren)))")
            self.assertIn(run_code(result), ['Hond', 'Kat', 'Kangoeroe'])
            print('Passed at level ', i)

    # def test_print_undefined_var(self):
    #     min_level = 7
    #
    #     for i in range(min_level, self.max_level + 1):
    #         with self.assertRaises(Exception) as context:
    #             result = hedy.transpile("print naam", i)
    #         self.assertEqual(str(context.exception), 'VarUndefined')
    #         print('Passed at level ', i)

class TestsLevel1(unittest.TestCase):

    def test_transpile_other(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("abc felienne 123", 1)
        self.assertEqual(str(context.exception), 'Invalid')

    def test_transpile_incomplete(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("print", 1)
        self.assertEqual(str(context.exception), 'Incomplete')



    def test_transpile_incomplete_with_multiple_lines(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("print hallo allemaal\nprint", 1)
        self.assertEqual(str(context.exception), 'Incomplete')


    def test_transpile_other(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("abc felienne 123", 1)
        self.assertEqual(str(context.exception), 'Invalid')
        self.assertEqual(str(context.exception.arguments), "{'invalid_command': 'abc', 'level': 1, 'guessed_command': 'ask'}")

    def test_transpile_incomplete_not_a_keyword(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("groen", 1)
        self.assertEqual(str(context.exception), 'Invalid')


    def test_transpile_print(self):
        result = hedy.transpile("print Hallo welkom bij Hedy!", 1)
        self.assertEqual(result, "print('Hallo welkom bij Hedy!')")
        self.assertEqual(run_code(result), 'Hallo welkom bij Hedy!')

    def test_transpile_ask_Spanish(self):
        result = hedy.transpile("ask ask Cuál es tu color favorito?", 1)
        self.assertEqual(result, "answer = input('ask Cuál es tu color favorito?')")

    def test_lines_may_end_in_spaces(self):
        result = hedy.transpile("print Hallo welkom bij Hedy! ", 1)
        self.assertEqual(result, "print('Hallo welkom bij Hedy! ')")
        self.assertEqual(run_code(result), 'Hallo welkom bij Hedy!')

    def test_lines_may_not_start_with_spaces(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile(" print Hallo welkom bij Hedy! ", 1)
        self.assertEqual('Invalid Space', str(context.exception))

    def test_print_with_comma(self):
        result = hedy.transpile("print iedereen zegt tegen hem: NERD, omdat hij de slimste van de klas is.", 1)
        self.assertEqual(result, "print('iedereen zegt tegen hem: NERD, omdat hij de slimste van de klas is.')")

    def test_word_plus_period(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("word.", 1)
        self.assertEqual('Invalid', str(context.exception))

    def test_two_lines_start_with_spaces(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile(" print Hallo welkom bij Hedy!\n print Hallo welkom bij Hedy!", 1)
        self.assertEqual('Invalid Space', str(context.exception))


    def test_transpile_empty(self):
        with self.assertRaises(hedy.HedyException) as context:
            result = hedy.transpile("", 1)

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

    def test_transpile_print_multiple_lines(self):
        result = hedy.transpile("print Hallo welkom bij Hedy\nprint Mooi hoor", 1)
        self.assertEqual(result, "print('Hallo welkom bij Hedy')\nprint('Mooi hoor')")

    def test_transpile_three_lines(self):
        input = """print Hallo
ask Wat is je lievelingskleur
echo je lievelingskleur is"""
        result = hedy.transpile(input, 1)
        self.assertEqual(result, "print('Hallo')\nanswer = input('Wat is je lievelingskleur')\nprint('je lievelingskleur is'+answer)")

    def test_transpile_echo(self):
        result = hedy.transpile("echo Jouw lievelingskleur is dus...", 1)
        self.assertEqual(result, "print('Jouw lievelingskleur is dus...'+answer)")

    def test_transpile_echo_without_argument(self):
        result = hedy.transpile("echo", 1)
        self.assertEqual(result, "print(answer)")


class TestsLevel2(unittest.TestCase):

    # some commands should not change:
    def test_transpile_other(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("abc felienne 123", 2)
        self.assertEqual(str(context.exception), 'Invalid')





    def test_transpile_ask_Spanish(self):
        result = hedy.transpile("ask ask Cuál es tu color favorito?", 2)
        self.assertEqual(result, "answer = input('ask Cuál es tu color favorito?')")

    def test_transpile_echo_at_level_2(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("echo Jouw lievelingskleur is dus...", 2)
        self.assertEqual(str(context.exception), 'Wrong Level')

    def test_spaces_in_arguments(self):
        result = hedy.transpile("print hallo      wereld", 2)
        self.assertEqual(result, "import random\nprint('hallo'+' '+'wereld')")




    def test_transpile_print(self):
        result = hedy.transpile("print Hallo welkom bij Hedy!", 2)
        self.assertEqual(result, "import random\nprint('Hallo'+' '+'welkom'+' '+'bij'+' '+'Hedy'+'!')")

    def test_transpile_ask(self):
        result = hedy.transpile("kleur is ask wat is je lievelingskleur?", 2)
        self.assertEqual(result, "import random\nkleur = input('wat is je lievelingskleur'+'?')")

    def test_transpile_ask_Spanish(self):
        result = hedy.transpile("color is ask ask Cuál es tu color favorito?", 2)
        self.assertEqual(result, "import random\ncolor = input('ask Cuál es tu color favorito'+'?')")

    def test_transpile_ask_with_print(self):
        result = hedy.transpile("kleur is ask wat is je lievelingskleur?\nprint kleur!", 2)
        self.assertEqual(result, "import random\nkleur = input('wat is je lievelingskleur'+'?')\nprint(kleur+'!')")


    def test_transpile_print_multiple_lines(self):
        result = hedy.transpile("print Hallo welkom bij Hedy!\nprint Mooi hoor", 2)
        self.assertEqual(result, "import random\nprint('Hallo'+' '+'welkom'+' '+'bij'+' '+'Hedy'+'!')\nprint('Mooi'+' '+'hoor')")
        self.assertEqual(run_code(result), "Hallo welkom bij Hedy!\nMooi hoor")

    def test_transpile_assign(self):
        result = hedy.transpile("naam is Felienne", 2)
        self.assertEqual(result, "import random\nnaam = 'Felienne'")

    def test_transpile_assign_2_integer(self):
        result = hedy.transpile("naam is 14", 2)
        self.assertEqual(result, "import random\nnaam = '14'")

    def test_transpile_assign_and_print(self):
        result = hedy.transpile("naam is Felienne\nprint naam", 2)
        self.assertEqual(result, "import random\nnaam = 'Felienne'\nprint(naam)")

    def test_transpile_assign_and_print_more_words(self):
        result = hedy.transpile("naam is Felienne\nprint hallo naam", 2)
        self.assertEqual(result, "import random\nnaam = 'Felienne'\nprint('hallo'+' '+naam)")

    def test_transpile_assign_and_print_punctuation(self):
        result = hedy.transpile("naam is Hedy\nprint Hallo naam!", 2)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('Hallo'+' '+naam+'!')")

    def test_transpile_assign_and_print_in_sentence(self):
        result = hedy.transpile("naam is Hedy\nprint naam is jouw voornaam", 2)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint(naam+' '+'is'+' '+'jouw'+' '+'voornaam')")

    def test_transpile_assign_and_print_something_else(self):
        result = hedy.transpile("naam is Felienne\nprint Hallo", 2)
        self.assertEqual(result, "import random\nnaam = 'Felienne'\nprint('Hallo')")

    def test_set_list_var(self):
        result = hedy.transpile("dieren is Hond, Kat, Kangoeroe", 2)
        self.assertEqual(result, "import random\ndieren = ['Hond', 'Kat', 'Kangoeroe']")

    def test_print_with_list_var(self):
        result = hedy.transpile("dieren is Hond, Kat, Kangoeroe\nprint dieren at 1", 2)
        self.assertEqual(result, "import random\ndieren = ['Hond', 'Kat', 'Kangoeroe']\nprint(dieren[1])")
        self.assertEqual(run_code(result), "Kat")



    def test_failing_car_program(self):
        #note: this the right order for assert: expected, actual
        result = hedy.transpile("""naam is ask wat is de naam van de hoofdpersoon
print naam doet mee aan een race hij krijgt een willekeurige auto""",2)
        self.assertEqual("import random\nnaam = input('wat is de naam van de hoofdpersoon')\nprint(naam+' '+'doet'+' '+'mee'+' '+'aan'+' '+'een'+' '+'race'+' '+'hij'+' '+'krijgt'+' '+'een'+' '+'willekeurige'+' '+'auto')",result)

    def test_windows_line_endings(self):
        result = hedy.transpile("print hallo\r\nprint allemaal", 2)
        self.assertEqual("import random\nprint('hallo')\nprint('allemaal')", result)

class TestsLevel3(unittest.TestCase):
    def test_transpile_other(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("abc felienne 123", 3)
        self.assertEqual(str(context.exception), 'Invalid')

    def test_transpile_print_level_2(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("print felienne 123", 3)
            self.assertEqual(str(context), 'First word is not a command') #hier moet nog we een andere foutmelding komen!

    def test_print(self):
        result = hedy.transpile("print 'hallo wereld!'", 3)
        self.assertEqual(result, "import random\nprint('hallo wereld!')")

    def test_print_with_comma(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet ,'",3)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('ik heet ,')")

    def test_print_with_single_quote(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet \\''",3)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('ik heet \\'')")

    def test_name_with_underscore(self):
        result = hedy.transpile("voor_naam is Hedy\nprint 'ik heet '",3)
        self.assertEqual(result, "import random\nvoor_naam = 'Hedy'\nprint('ik heet ')")

    def test_name_that_is_keyword(self):
        result = hedy.transpile("for is Hedy\nprint 'ik heet ' for ",3)
        self.assertEqual(result, "import random\n_for = 'Hedy'\nprint('ik heet '+_for)")

    def test_print_Spanish(self):
        result = hedy.transpile("print 'Cuál es tu color favorito?'", 3)
        self.assertEqual(result, "import random\nprint('Cuál es tu color favorito?')")

    def test_print_with_list_var(self):
        result = hedy.transpile("dieren is Hond, Kat, Kangoeroe\nprint dieren at 1", 3)
        self.assertEqual(result, "import random\ndieren = ['Hond', 'Kat', 'Kangoeroe']\nprint(dieren[1])")
        self.assertEqual(run_code(result), "Kat")

    def test_print_with_list_var_random(self):
        result = hedy.transpile("dieren is Hond, Kat, Kangoeroe\nprint dieren at random", 3)
        self.assertEqual(result, "import random\ndieren = ['Hond', 'Kat', 'Kangoeroe']\nprint(random.choice(dieren))")
        self.assertIn(run_code(result), ['Hond', 'Kat', 'Kangoeroe'])


    def test_transpile_ask_Spanish(self):
        result = hedy.transpile("color is ask Cuál es tu color favorito?", 3)
        self.assertEqual(result, "import random\ncolor = input('Cuál es tu color favorito?')")

    def test_print_2(self):
        result = hedy.transpile("print 'ik heet henk'", 3)
        self.assertEqual(result, """import random
print('ik heet henk')""")

    def test_print_with_var(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet' naam", 3)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('ik heet'+naam)")

    def test_transpile_ask_with_print(self):
        result = hedy.transpile("kleur is ask wat is je lievelingskleur?\nprint 'jouw lievelingskleur is dus' kleur '!'", 3)
        self.assertEqual(result, "import random\nkleur = input('wat is je lievelingskleur?')\nprint('jouw lievelingskleur is dus'+kleur+'!')")

class TestsLevel4(unittest.TestCase):
    #invalid, ask and print should still work as in level 4
    def test_transpile_other(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("abc felienne 123", 4)
        self.assertEqual(str(context.exception), 'Invalid')

    def test_print_with_var(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet' naam", 4)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('ik heet'+naam)")

    def test_print_with_comma(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet,' naam", 4)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('ik heet,'+naam)")



    def test_transpile_ask_with_print(self):
        result = hedy.transpile("kleur is ask wat is je lievelingskleur?\nprint 'jouw lievelingskleur is dus' kleur '!'", 4)
        self.assertEqual(result, "import random\nkleur = input('wat is je lievelingskleur?')\nprint('jouw lievelingskleur is dus'+kleur+'!')")

    def test_transpile_ask_Spanish(self):
        result = hedy.transpile("color is ask Cuál es tu color favorito?", 4)
        self.assertEqual(result, "import random\ncolor = input('Cuál es tu color favorito?')")

    def test_save_list_access_to_var(self):
        result = hedy.transpile("dieren is Hond, Kat, Kangoeroe\ndier is dieren at random\nprint dier", 4)
        self.assertEqual(result, "import random\ndieren = ['Hond', 'Kat', 'Kangoeroe']\ndier=random.choice(dieren)\nprint(dier)")
        self.assertIn(run_code(result), ['Hond', 'Kat', 'Kangoeroe'])

    def test_print_Spanish(self):
        result = hedy.transpile("print 'Cuál es tu color favorito?'", 4)
        self.assertEqual(result, "import random\nprint('Cuál es tu color favorito?')")

    #now adds if
    def test_print_if_else(self):
        result = hedy.transpile("""naam is Hedy
print 'ik heet' naam
if naam is Hedy print 'leuk' else print 'minder leuk'""", 4)

        expected_result = """import random
naam = 'Hedy'
print('ik heet'+naam)
if naam == 'Hedy':
  print('leuk')
else:
  print('minder leuk')"""
        self.assertEqual(expected_result, result)

    def test_print_if_else_with_ask(self):
        result = hedy.transpile("""kleur is ask Wat is je lievelingskleur?
if kleur is groen print 'mooi!' else print 'niet zo mooi'""", 4)

        expected_result = """import random
kleur = input('Wat is je lievelingskleur?')
if kleur == 'groen':
  print('mooi!')
else:
  print('niet zo mooi')"""

        self.assertEqual(expected_result, result)

    #steen schaar papier
    def test_print_if_else_with_and_var(self):
        result = hedy.transpile("""jouwkeuze is steen
computerkeuze is schaar
if computerkeuze is schaar and jouwkeuze is steen print 'jij wint'""", 4)

        expected_result = """import random
jouwkeuze = 'steen'
computerkeuze = 'schaar'
if computerkeuze == 'schaar' and jouwkeuze == 'steen':
  print('jij wint')"""
        self.assertEqual(expected_result, result)
        self.assertEqual(run_code(result),'jij wint')

    def test_print_if_with_var(self):
        result = hedy.transpile("""jouwkeuze is schaar
computerkeuze is schaar
if computerkeuze is jouwkeuze print 'gelijkspel!'""", 4)

        expected_result = """import random
jouwkeuze = 'schaar'
computerkeuze = 'schaar'
if computerkeuze == jouwkeuze:
  print('gelijkspel!')"""
        self.assertEqual(expected_result, result)
        self.assertEqual(run_code(result),'gelijkspel!')

    def test_if_in_array(self):
        actual = hedy.transpile("items is red, green\nselected is red\nif selected in items print 'found!'", 4)

        expected = """import random
items = ['red', 'green']
selected = 'red'
if selected in items:
  print('found!')"""

        self.assertEqual(expected, actual)
        self.assertEqual('found!', run_code(actual))

class TestsLevel5(unittest.TestCase):
    #print should still work
    def test_print_with_var(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet' naam", 5)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('ik heet'+naam)")

    def test_print_with_comma(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet,' naam", 5)
        self.assertEqual(result, "import random\nnaam = 'Hedy'\nprint('ik heet,'+naam)")

    def test_print_Spanish(self):
        result = hedy.transpile("print 'Cuál es tu color favorito?'", 5)
        self.assertEqual(result, "import random\nprint('Cuál es tu color favorito?')")

    def test_transpile_ask_Spanish(self):
        result = hedy.transpile("color is ask Cuál es tu color favorito?", 5)
        self.assertEqual(result, "import random\ncolor = input('Cuál es tu color favorito?')")

    def test_transpile_other(self):
        with self.assertRaises(Exception) as context:
            result = hedy.transpile("abc felienne 123", 5)
        self.assertEqual(str(context.exception), 'Invalid')

    #todo: a few more things repeated from 4 here?

    #now add repeat
    def test_repeat_basic_print(self):
        result = hedy.transpile("repeat 5 times print 'me wants a cookie!'", 5)
        self.assertEqual(result, """import random
for i in range(int('5')):
  print('me wants a cookie!')""")
        self.assertEqual(run_code(result),'me wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!')

    def test_repeat_with_variable_print(self):
        result = hedy.transpile("n is 5\nrepeat n times print 'me wants a cookie!'", 5)
        self.assertEqual(result, """import random
n = '5'
for i in range(int(n)):
  print('me wants a cookie!')""")
        self.assertEqual(run_code(result),'me wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!')

    def test_repeat_nested_in_if(self):
        result = hedy.transpile("kleur is ask Wat is je lievelingskleur?\nif kleur is groen repeat 3 times print 'mooi!'", 5)
        self.assertEqual(result, """import random
kleur = input('Wat is je lievelingskleur?')
if kleur == 'groen':
  for i in range(int('3')):
    print('mooi!')""")

    def test_repeat_over_9_times(self):
        result = hedy.transpile("repeat 10 times print 'me wants a cookie!'", 5)
        self.assertEqual(result, """import random
for i in range(int('10')):
  print('me wants a cookie!')""")
        self.assertEqual(run_code(result),'me wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!')

class TestsLevel6(unittest.TestCase):
    def test_print_with_var(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet' naam", 6)
        self.assertEqual("import random\nnaam = 'Hedy'\nprint('ik heet'+str(naam))",result)


    def test_transpile_ask(self):
        result = hedy.transpile("antwoord is ask wat is je lievelingskleur?", 6)
        self.assertEqual(result, "import random\nantwoord = input('wat is je lievelingskleur?')")

    def test_repeat_nested_in_if(self):
            result = hedy.transpile("kleur is ask Wat is je lievelingskleur?\nif kleur is groen repeat 3 times print 'mooi!'", 6)
            self.assertEqual(result, """import random
kleur = input('Wat is je lievelingskleur?')
if str(kleur) == str('groen'):
  for i in range(int('3')):
    print('mooi!')""")

def test_repeat_with_variable_print(self):
    result = hedy.transpile("n is 5\nrepeat n times print 'me wants a cookie!'", 6)
    self.assertEqual(result, """import random
n = '5'
for i in range(int(n)):
    print('me wants a cookie!')""")
    self.assertEqual(run_code(result),
                     'me wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!')

    #new tests for calculations
    def test_simple_calculation(self):
        result = hedy.transpile("nummer is 4 + 5", 6)
        self.assertEqual('import random\nnummer = int(4) + int(5)', result)

    def test_simple_calculation(self):
        result = hedy.transpile("nummer is 4+5", 6)
        self.assertEqual('import random\nnummer = int(4) + int(5)', result)

    def test_calculation_and_printing(self):
        result = hedy.transpile("nummer is 4 + 5\nprint nummer", 6)
        self.assertEqual('import random\nnummer = int(4) + int(5)\nprint(str(nummer))', result)
        self.assertEqual(run_code(result), "9")

    def test_calculation_with_vars(self):
        result = hedy.transpile("""nummer is 5
nummertwee is 6
getal is nummer * nummertwee
print getal""", 6)
        self.assertEqual("""import random
nummer = '5'
nummertwee = '6'
getal = int(nummer) * int(nummertwee)
print(str(getal))""", result)
        self.assertEqual(run_code(result), "30")

    def test_print_calculation_times_directly(self):
        result = hedy.transpile("""nummer is 5
nummertwee is 6
print nummer * nummertwee""", 6)
        self.assertEqual("""import random
nummer = '5'
nummertwee = '6'
print(str(int(nummer) * int(nummertwee)))""", result)
        self.assertEqual(run_code(result), "30")

    def test_print_calculation_divide_directly(self):
        result = hedy.transpile("""nummer is 5
nummertwee is 6
print nummer / nummertwee""", 6)
        self.assertEqual("""import random
nummer = '5'
nummertwee = '6'
print(str(int(nummer) // int(nummertwee)))""", result)
        self.assertEqual(run_code(result), "0")

class TestsLevel7(unittest.TestCase):
    def test_print(self):
        result = hedy.transpile("print 'ik heet'", 7)
        self.assertEqual("import random\nprint('ik heet')",result)

    def test_print_with_var(self):
        result = hedy.transpile("naam is Hedy\nprint 'ik heet' naam", 7)
        self.assertEqual("import random\nnaam = '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)
        self.assertEqual("import random\nprint('5 keer 5 is '+str(int(5) * int(5)))",result)





    def test_print_calculation_times_directly(self):
        result = hedy.transpile("""nummer is 5
nummertwee is 6
print nummer * nummertwee""", 7)
        self.assertEqual("""import random
nummer = '5'
nummertwee = '6'
print(str(int(nummer) * int(nummertwee)))""", result)
        self.assertEqual(run_code(result), "30")


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

    def test_if_with_indent(self):
        result = hedy.transpile("""naam is Hedy
if naam is Hedy
    print 'koekoek'""", 7)
        self.assertEqual("""import random
naam = 'Hedy'
if str(naam) == str('Hedy'):
    print('koekoek')""", result)

    def test_repeat_with_indent(self):
        result = hedy.transpile("""repeat 5 times
    print 'koekoek'""", 7)
        self.assertEqual("""import random
for i in range(int(5)):
    print('koekoek')""",result)

    def test_repeat_with_variable_print(self):
        result = hedy.transpile("n is 5\nrepeat n times\n    print 'me wants a cookie!'", 7)
        self.assertEqual(result, """import random
n = '5'
for i in range(int(n)):
    print('me wants a cookie!')""")
        self.assertEqual(run_code(result),
                         'me wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!')

    def test_repeat_nested_in_if(self):
        result = hedy.transpile("""kleur is groen
if kleur is groen
    repeat 3 times
        print 'mooi'""", 7)
        self.assertEqual(result, """import random
kleur = 'groen'
if str(kleur) == str('groen'):
    for i in range(int(3)):
        print('mooi')""")

    def test_if_else(self):
        result = hedy.transpile("""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""", 7)

        self.assertEqual("""import random
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)

    def test_repeat_basic_print(self):
        result = hedy.transpile("""repeat 5 times
    print 'me wants a cookie!'""", 7)
        self.assertEqual(result, """import random
for i in range(int(5)):
    print('me wants a cookie!')""")
        self.assertEqual(run_code(result),'me wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!\nme wants a cookie!')


    def test_print_random(self):
        result = hedy.transpile("""keuzes is steen, schaar, papier
computerkeuze is keuzes at random
print 'computer koos ' computerkeuze""", 7)
        self.assertEqual("""import random
keuzes = ['steen', 'schaar', 'papier']
computerkeuze = random.choice(keuzes)
print('computer koos '+str(computerkeuze))""", result)


    def test_repeat_basic_print_multiple_lines(self):
        result = hedy.transpile("""repeat 5 times
    print 'cookieeee!'
    print 'me wants a cookie!'""", 7)
        self.assertEqual(result, """import random
for i in range(int(5)):
    print('cookieeee!')
    print('me wants a cookie!')""")
        # self.assertEqual(run_code(result),'cookieeee!\nme wants a cookie!\ncookieeee!\nme wants a cookie!\ncookieeee!\nme wants a cookie!\ncookieeee!\nme wants a cookie!\ncookieeee!\nme wants a cookie!')





if __name__ == '__main__':
    unittest.main()



    # python = transpile(tree)
    # print(python)
    #
    # exec(python)