String formatting for empty expressions

diff --git a/pypol/linear.py b/pypol/linear.py
index 5b5d8aa..6550e92 100644 (file)
--- a/pypol/linear.py
+++ b/pypol/linear.py
@@ -206,6 +206,8 @@ class Expression:
         elif constant < 0:
             constant *= -1
             string += ' - {}'.format(constant)
         elif constant < 0:
             constant *= -1
             string += ' - {}'.format(constant)

+        if string == '':
+            string = '0'

         return string
 
     def _parenstr(self, always=False):
         return string
 
     def _parenstr(self, always=False):

diff --git a/tests/test_linear.py b/tests/test_linear.py
index 3912fd3..4636275 100644 (file)
--- a/tests/test_linear.py
+++ b/tests/test_linear.py
@@ -110,6 +110,7 @@ class TestExpression(unittest.TestCase):
         self.assertEqual(self.e / 2, self.x / 2 - self.y + Fraction(3, 2))
 
     def test_str(self):
         self.assertEqual(self.e / 2, self.x / 2 - self.y + Fraction(3, 2))
 
     def test_str(self):

+        self.assertEqual(str(Expression()), '0')

         self.assertEqual(str(self.x), 'x')
         self.assertEqual(str(-self.x), '-x')
         self.assertEqual(str(self.pi), '22/7')
         self.assertEqual(str(self.x), 'x')
         self.assertEqual(str(-self.x), '-x')
         self.assertEqual(str(self.pi), '22/7')