diff options
author | Jeffrey Yancey <jeffrey@octane5.com> | 2018-08-26 14:50:37 -0600 |
---|---|---|
committer | Jeffrey Yancey <jeffrey@octane5.com> | 2018-08-26 14:50:37 -0600 |
commit | bfc0020621a931726e219a013e08e9f96e767a33 (patch) | |
tree | b0c2156238dce955b00c6dd93cc8e526db5cccfe /numpy/polynomial/tests/test_polynomial.py | |
parent | 52c67eb9a213b9df87f98d6f1f70398e147de425 (diff) | |
download | numpy-bfc0020621a931726e219a013e08e9f96e767a33.tar.gz |
TST: add missing tests for all polynomial subclass pow fns.
Diffstat (limited to 'numpy/polynomial/tests/test_polynomial.py')
-rw-r--r-- | numpy/polynomial/tests/test_polynomial.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/numpy/polynomial/tests/test_polynomial.py b/numpy/polynomial/tests/test_polynomial.py index 67728e35e..b5a3678a4 100644 --- a/numpy/polynomial/tests/test_polynomial.py +++ b/numpy/polynomial/tests/test_polynomial.py @@ -3,6 +3,8 @@ """ from __future__ import division, absolute_import, print_function +from functools import reduce + import numpy as np import numpy.polynomial.polynomial as poly from numpy.testing import ( @@ -102,6 +104,15 @@ class TestArithmetic(object): res = poly.polyadd(poly.polymul(quo, ci), rem) assert_equal(res, tgt, err_msg=msg) + def test_polypow(self): + for i in range(5): + for j in range(5): + msg = "At i=%d, j=%d" % (i, j) + c = np.arange(i + 1) + tgt = reduce(poly.polymul, [c]*j) if j else np.array([1]) + res = poly.polypow(c, j) + assert_equal(trim(res), trim(tgt), err_msg=msg) + class TestEvaluation(object): # coefficients of 1 + 2*x + 3*x**2 |