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)