@ -49,3 +49,5 @@ def calculate_constant_string(string):
def simplify_easy_math_function(type, function):
if type != "v":
return False
if is_string(function) == False:
return function
@ -34,6 +34,8 @@ class read_function(unittest.TestCase):
def test_easy_function_wrong_type(self):
self.assertEqual(simplify_easy_math_function("a", "42x"), False)
def test_easy_function_number_as_function_return(self):
self.assertEqual(simplify_easy_math_function("v", 42), 42)
if __name__ == '__main__':
unittest.main()