Minor improvements in testing of new expressions

[linpy.git] / pypol / tests / test_linexprs.py

diff --git a/pypol/tests/test_linexprs.py b/pypol/tests/test_linexprs.py
index 68cad74..c55c842 100644 (file)
--- a/pypol/tests/test_linexprs.py
+++ b/pypol/tests/test_linexprs.py
@@ -19,8 +19,8 @@ class TestExpression(unittest.TestCase):
         self.expr = self.x - 2*self.y + 3
 
     def test_new(self):
         self.expr = self.x - 2*self.y + 3
 
     def test_new(self):

-        self.assertIsInstance(self.x, Symbol)
-        self.assertIsInstance(self.pi, Rational)
+        self.assertIsInstance(Expression(coefficients={self.x: 1}), Symbol)
+        self.assertIsInstance(Expression(constant=self.pi), Rational)

         self.assertNotIsInstance(self.x + self.pi, Symbol)
         self.assertNotIsInstance(self.x + self.pi, Rational)
         xx = Expression({self.x: 2})
         self.assertNotIsInstance(self.x + self.pi, Symbol)
         self.assertNotIsInstance(self.x + self.pi, Rational)
         xx = Expression({self.x: 2})


@@ -242,6 +242,13 @@ class TestDummy(unittest.TestCase):
         self.assertNotEqual(self.x, Dummy('x'))
         self.assertNotEqual(Dummy(), Dummy())
 
         self.assertNotEqual(self.x, Dummy('x'))
         self.assertNotEqual(Dummy(), Dummy())
 

+    def test_repr(self):
+        self.assertEqual(repr(self.x), '_x')
+        dummy1 = Dummy()
+        dummy2 = Dummy()
+        self.assertTrue(repr(dummy1).startswith('_Dummy_'))
+        self.assertNotEqual(repr(dummy1), repr(dummy2))
+

 
 class TestSymbols(unittest.TestCase):
 
 
 class TestSymbols(unittest.TestCase):