Skip to content

Commit

Permalink
Make Polynomial.coefficient() accept symbols
Browse files Browse the repository at this point in the history
  • Loading branch information
tueda committed Jul 9, 2017
1 parent 695318e commit 4577a91
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions tests/test_poly.py
Original file line number Diff line number Diff line change
Expand Up @@ -286,6 +286,22 @@ def assert_equal(*args, **kwargs):
assert_raises(ValueError, [x1, 'x'], multiset=False)
assert_raises(ValueError, [x1, x2], multiset=False)

def test_interpret_integers(self):
"""Test for Polynomial._interpret_integers()."""
def assert_raises(error, *args, **kwargs):
self.assertRaises(
error, lambda: Polynomial._interpret_integers(*args, **kwargs))

def assert_equal(*args, **kwargs):
self.assertEqual(
Polynomial._interpret_integers(*args[:-1], **kwargs), args[-1])

assert_equal(17, [17])
assert_raises(TypeError, set([3, 5]), ordered=True)
assert_raises(ValueError, [], nonempty=True)
assert_equal([3, 5, 7], [3, 5, 7])
assert_raises(TypeError, object())

def test_factorize(self):
"""Test polynomial factorization."""
p1 = Polynomial('-3+x*y')
Expand Down Expand Up @@ -373,6 +389,9 @@ def test_coefficient(self):

self.assertEqual(p.coefficient(x, 2, False), 1)

self.assertRaises(ValueError, lambda: p.coefficient('x y', [1, 2, 3]))
self.assertEqual(p.coefficient(['x', 'y'], [1, 1]), Polynomial('3'))

def test_exponent_vectors(self):
"""Test for Polynomial.exponent_vectors()."""
p = Polynomial('1+x')
Expand Down

0 comments on commit 4577a91

Please sign in to comment.