def test_it_adds_user_defined_function_and_go_on_with_program(self):
tokens = "| a b - | a b my_fun defstack 5 a = 3 b = a b /".split()
ret = eval(tokens, self.env)
self.assertEqual(ret, 5.0/3)
self.assertEqual(self.env["my_fun"].call_stack, ["a", "b", "-"])
self.assertEqual(self.env["a"], 5.0)
self.assertEqual(self.env["b"], 3.0)
def test_it_calls_an_user_defined_function_respecting_args_order(self):
tokens = "5 8 my_fun".split()
self.env.update({
"my_fun": UserDefinedFunc(["a", "b", "-"], ["a", "b"]),
})
ret = eval(tokens, self.env)
self.assertEqual(ret, -3.0)
def test_it_assigns_the_result_of_a_user_defined_function_to_a_variable(self):
tokens = "5 8 user_defined_subtraction my_var =".split()
self.env.update({
"user_defined_subtraction": UserDefinedFunc(["a", "b", "-"], ["a", "b"]),
})
ret = eval(tokens, self.env)
self.assertEqual(self.env["my_var"], -3.0)
def test_it_calls_an_user_defined_function_with_variables_as_arguments(self):
tokens = "c d my_fun".split()
self.env.update({
"my_fun": UserDefinedFunc(["a", "b", "-"], ["a", "b"]),
"c": 5,
"d": 8,
})
ret = eval(tokens, self.env)
self.assertEqual(ret, -3.0)
def test_it_works_with_multiple_user_defined_calls(self):
tokens = "c d my_sub e my_div".split()
self.env.update({
"my_sub": UserDefinedFunc(["a", "b", "-"], ["a", "b"]),
"my_div": UserDefinedFunc(["a", "b", "/"], ["a", "b"]),
"c": 5,
"d": 8,
"e": 2,
})
ret = eval(tokens, self.env)
self.assertEqual(ret, -3.0/2)
def test_it_works_with_variable_assignment(self):
tokens = ["5", "a", "="]
ret = eval(tokens, self.env)
self.assertEqual(ret, None)
self.assertEqual(self.env["a"], 5)
def test_it_works_with_addition(self):
tokens = ["5", "3", "+"]
ret = eval(tokens, self.env)
self.assertEqual(ret, 8.0)
def test_it_adds_a_user_defined_func_to_env__with_the_right_name(self):
tokens = ["|", "a", "b", "+", "|", "a", "b", "my_fun", "defstack"]
ret = eval(tokens, self.env)
self.assertEqual(ret, None)
self.assertTrue(isinstance(self.env["my_fun"], UserDefinedFunc))
self.assertEqual(self.env["my_fun"].call_stack, ["a", "b", "+"])
def test_it_assigns_results_of_function_call_to_variable(self):
tokens = ["5", "1", "min", "var_name", "="]
ret = eval(tokens, self.env)
self.assertEqual(ret, None)
self.assertEqual(self.env["var_name"], 1)
def test_it_assigns_variable_and_evaluates_function_call(self):
tokens = ["5", "var_name", "=", "4", "var_name", "-"]
ret = eval(tokens, self.env)
self.assertEqual(ret, -1)
def test_it_works_with_stacked_operations(self):
tokens = ["5", "3", "+", "4", "-"]
ret = eval(tokens, self.env)
self.assertEqual(ret, 4.0)
def test_it_raises_for_malformed_expressions(self):
expressions = ["4 3 + = ", "1 2 3 min", "0 max"]
for expression in expressions:
tokens = expression.split()
with self.assertRaises(InvalidExpression):
eval(tokens, self.env)
def test_it_raises_for_invalid_variable_names(self):
for var_name in ["defstack", "[", "]", "1", "list", "set"]:
tokens = ["5", var_name, "="]
with self.assertRaises(InvalidVariableName):
ret = eval(tokens, self.env)
def test_it_works_with_max(self):
tokens = ["3", "-5", "max"]
ret = eval(tokens, self.env)
self.assertEqual(ret, 3.0)
def test_it_works_with_multiplication(self):
tokens = ["3", "5", "*"]
ret = eval(tokens, self.env)
self.assertEqual(ret, 15.0)
def test_it_works_with_division(self):
tokens = ["1", "2", "/"]
ret = eval(tokens, self.env)
self.assertEqual(ret, 0.5)
def test_it_works_with_subtraction(self):
tokens = ["10", "12", "-"]
ret = eval(tokens, self.env)
self.assertEqual(ret, -2.0)