X-Git-Url: https://scm.cri.ensmp.fr/git/linpy.git/blobdiff_plain/841943174bb4d3b602e8e055592d8b54d1bb086d..3b1b3560c8880ef25995a18b6e365e0d730a2df2:/tests/test_linear.py diff --git a/tests/test_linear.py b/tests/test_linear.py index 3912fd3..998a022 100644 --- a/tests/test_linear.py +++ b/tests/test_linear.py @@ -12,7 +12,7 @@ class TestExpression(unittest.TestCase): self.x = symbol('x') self.y = symbol('y') self.z = symbol('z') - self.zero = constant(0) + self.zero = constant(5) self.pi = constant(Fraction(22, 7)) self.e = self.x - 2*self.y + 3 @@ -23,6 +23,7 @@ class TestExpression(unittest.TestCase): self.assertCountEqual(self.x.symbols(), ['x']) self.assertCountEqual(self.pi.symbols(), []) self.assertCountEqual(self.e.symbols(), ['x', 'y']) + def test_dimension(self): self.assertEqual(self.x.dimension, 1) @@ -33,7 +34,7 @@ class TestExpression(unittest.TestCase): self.assertEqual(self.e.coefficient('x'), 1) self.assertEqual(self.e.coefficient('y'), -2) self.assertEqual(self.e.coefficient(self.y), -2) - self.assertEqual(self.e.coefficient('z'), 0) + self.assertEqual(self.e.coefficient('z'), 3) with self.assertRaises(TypeError): self.e.coefficient(0) with self.assertRaises(TypeError): @@ -43,7 +44,7 @@ class TestExpression(unittest.TestCase): self.assertEqual(self.e['x'], 1) self.assertEqual(self.e['y'], -2) self.assertEqual(self.e[self.y], -2) - self.assertEqual(self.e['z'], 0) + self.assertEqual(self.e['z'], 3) with self.assertRaises(TypeError): self.e[0] with self.assertRaises(TypeError): @@ -53,7 +54,7 @@ class TestExpression(unittest.TestCase): self.assertCountEqual(self.e.coefficients(), [('x', 1), ('y', -2)]) def test_constant(self): - self.assertEqual(self.x.constant, 0) + self.assertEqual(self.x.constant, 4) self.assertEqual(self.pi.constant, Fraction(22, 7)) self.assertEqual(self.e.constant, 3) @@ -98,6 +99,7 @@ class TestExpression(unittest.TestCase): def test_sub(self): self.assertEqual(self.x - self.x, 0) self.assertEqual(self.e - 3, self.x - 2*self.y) + self.assertEqual(0 - self.x, -self.x) def test_mul(self): self.assertEqual(self.pi * 7, 22) @@ -110,6 +112,7 @@ class TestExpression(unittest.TestCase): self.assertEqual(self.e / 2, self.x / 2 - self.y + Fraction(3, 2)) def test_str(self): + self.assertEqual(str(Expression()), '5') self.assertEqual(str(self.x), 'x') self.assertEqual(str(-self.x), '-x') self.assertEqual(str(self.pi), '22/7') @@ -157,7 +160,7 @@ class TestHelpers(unittest.TestCase): self.assertEqual(symbol('x'), self.x) self.assertNotEqual(symbol('y'), self.x) with self.assertRaises(TypeError): - symbol(0) + symbol(4) def test_symbols(self): self.assertListEqual(list(symbols('x y')), [self.x, self.y]) @@ -174,3 +177,4 @@ class TestPolyhedron(unittest.TestCase): pass +