Browse Source

added test for returning the number if the function is no string as well as corresponding functionality

remotes/origin/feature/feature-read-math-functions
fdai7763 11 months ago
parent
commit
63e01cdd1f
  1. 2
      src/main/py/read_math_function.py
  2. 2
      src/test/py/test_read_math_function.py

2
src/main/py/read_math_function.py

@ -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

2
src/test/py/test_read_math_function.py

@ -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()
Loading…
Cancel
Save