adapt tests to new poly

This commit is contained in:
Lafrite 2014-11-15 10:16:59 +01:00
parent 12bcf7214a
commit 3e9277d86b
1 changed files with 10 additions and 9 deletions

View File

@ -6,6 +6,7 @@ import unittest
from pymath.polynom import Polynom
from pymath.fraction import Fraction
from pymath.expression import Expression
class TestPolynom(unittest.TestCase):
@ -86,36 +87,36 @@ class TestPolynom(unittest.TestCase):
self.assertEqual(ans, p.postfix)
def test_postfix_arithm_coef(self):
p = Polynom([1,[2, 3, "+"],4])
p = Polynom([1,Expression([2, 3, "+"]),4])
#ans = [1, 2, 3, "+", "x", "*", "+", 4, "x", 2, "^", "*", "+"]
ans = [4, 'x', 2, '^', '*', 2, 3, '+', 'x', '*', '+', 1, '+']
self.assertEqual(ans, p.postfix)
def test_reduce_nilpo(self):
p = Polynom([1, 2, 3])
self.assertEqual(p, p.reduce()[-1])
#def test_reduce_nilpo(self):
# p = Polynom([1, 2, 3])
# self.assertEqual(p, p.reduce()[-1])
def test_reduce(self):
p = Polynom([1, [2, 3], 4])
reducted = Polynom([1, 5, 4])
self.assertEqual(p.reduce()[-1],reducted)
self.assertEqual(str(p.reduce()[-1]),'4 * x ^ 2 + 5 * x + 1')
def test_add_int(self):
p = Polynom([1, 2, 3])
q = (p + 2)[-1]
self.assertEqual(q, Polynom([3, 2, 3]))
self.assertEqual(str(q), '3 * x ^ 2 + 2 * x + 3')
def test_add_frac(self):
p = Polynom([1, 2, 3])
f = Fraction(1, 2)
q = (p + f)[-1]
self.assertEqual(q, Polynom([Fraction(3, 2), 2, 3]))
#ans = repr(Polynom([Expression(Fraction(3, 2)), Expression(2), Expression(3)]))
self.assertEqual(str(q),'3 * x ^ 2 + 2 * x + 3 / 2')
def test_add_poly(self):
p = Polynom([1, 0, 3])
q = Polynom([0, 2, 3])
r = (p + q)[-1]
self.assertEqual(r, Polynom([1, 2, 6]))
self.assertEqual(str(r), '6 * x ^ 2 + 2 * x + 1')
def test_radd_int(self):
pass