def test_function_only_with_two_parameters_assign_to_variable(self):
expression_str = "result = get_some_thing(param1, param2)"
expression_tree = expression_parser.parse(expression_str)
self.assertEqual("get_some_thing", expression_tree.function_name)
self.assertItemsEqual(["param1", "param2"], expression_tree.parameters)
self.assertEqual("result", expression_tree.variable_name)
def test_variable_only(self):
expression_str = "test"
expression_tree = expression_parser.parse(expression_str)
self.assertEqual(expression_tree.variable_name, expression_str)
def test_function_only_with_one_parameter(self):
expression_str = "get_some_thing(param1)"
expression_tree = expression_parser.parse(expression_str)
self.assertEqual("get_some_thing", expression_tree.function_name)
self.assertItemsEqual(["param1"], expression_tree.parameters)
