def test_expression_variable_names(self):
x = s.create_variable('x')
y = s.create_variable('y')
z = s.create_variable('z')
f = (z - 2 * (x + 1j)) + 3.5 * (y - 2)
variables = f.variable_names
expected = ['x', 'y', 'z']
for v in variables:
self.assertTrue(v in expected)
for v in expected:
self.assertTrue(v in variables)
def test_str_method_of_classes(self):
four = s.create_constant(4)
y = s.create_variable('y')
f = four + 3.5 * (y - 2)
self.assertEqual(str(four), '4')
self.assertEqual(str(y), 'y')
self.assertEqual(str(f), '(4 + (3.5 * (y - 2)))')
def test_expressions(self):
four = s.create_constant(4)
y = s.create_variable('y')
f = four / 2 + 3.5 * (y - 2)
d = (3 + 1j) + y
z = 15 - 2 / four * (12 / f)
self.assertEqual(d.evaluate(y=2), 5 + 1j)
self.assertEqual(f.evaluate(y=2), 2)
self.assertEqual(f.evaluate(y=0), -5)
self.assertEqual(z.evaluate(y=2, x=3), 12.0)
def test_five_plus_three(self):
plus = second.create_operator('+', lambda lhs, rhs: lhs + rhs)
x = second.create_variable('x')
y = second.create_variable('y')
added_expression = second.create_expression((x, plus, y))
self.assertEqual(added_expression.evaluate(x=5, y=3), 8)
def test_constant_evaluation(self):
self.assertEqual(second.create_variable('x').evaluate(x=42), 42)
self.assertEqual(second.create_constant(5).evaluate(), 5)
def test_operators(self):
y = second.create_variable('y')
twelve = second.create_constant(12)
expression = y + twelve
self.assertEqual(expression.evaluate(y=3), 15)
def test_five_plus_three(self):
plus = second.create_operator('+', lambda lhs, rhs: lhs + rhs)
x = second.create_variable('x')
y = second.create_variable('y')
added_expression = second.create_expression((x, plus, y))
self.assertEqual(added_expression.evaluate(x=5, y=3), 8)
def test_constant_evaluation(self):
self.assertEqual(second.create_variable('x').evaluate(x=42), 42)
self.assertEqual(second.create_constant(5).evaluate(), 5)
def test_operators(self):
y = second.create_variable('y')
twelve = second.create_constant(12)
expression = y + twelve
self.assertEqual(expression.evaluate(y=3), 15)
