diff --git a/pymath/polynom.py b/pymath/polynom.py index fc570a5..21dc7b9 100644 --- a/pymath/polynom.py +++ b/pymath/polynom.py @@ -68,7 +68,11 @@ class Polynom(object): :returns: Expression ready to be simplify """ - postfix_exp = [value if i==self._letter else i for i in self.postfix] + if isNumber(value): + postfix_exp = [value if i==self._letter else i for i in self.postfix] + else: + postfix_exp = [Expression(value) if i==self._letter else i for i in self.postfix] + return Expression(postfix_exp) def feed_coef(self, l_coef): diff --git a/test/test_polynom.py b/test/test_polynom.py index a21ba0a..00b96c5 100644 --- a/test/test_polynom.py +++ b/test/test_polynom.py @@ -43,6 +43,10 @@ class TestPolynom(unittest.TestCase): p = Polynom([-1]) self.assertEqual(p(3).simplified(), -1) + def test_eval_poly(self): + p = Polynom([1, 2]) + self.assertEqual(p("1+h").simplified(), Polynom([3,2])) + #def test_print(self): # p = Polynom([1,2,3]) # ans = "1 + 2 x + 3 x^2"