test_variables.py 812 B

1234567891011121314151617181920212223242526272829303132
  1. import unittest
  2. from tests.parser import TestParser, run_expressions
  3. from sympy import Symbol, symbols
  4. class TestVariables(unittest.TestCase):
  5. def setUp(self):
  6. pass
  7. def tearDown(self):
  8. pass
  9. def test_addition(self):
  10. expressions = [('5 + 5', 5 + 5)]
  11. run_expressions(expressions)
  12. def test_addition_of_one_term(self):
  13. a = Symbol('a')
  14. expressions = [('a + 5', 5 + a)]
  15. run_expressions(expressions)
  16. def test_addition_of_two_terms(self):
  17. a, b = symbols('a,b')
  18. expressions = [('4*a + 5*b', 4*a + 5*b)]
  19. run_expressions(expressions)
  20. #def test_short_addition_of_two_terms(self):
  21. # a, b = symbols('a,b')
  22. # expressions = [('4a + 5b', 4.0*a + 5.0*b)]
  23. # run_expressions(expressions, verbose=1)