diff options
| -rw-r--r-- | numpy/core/tests/test_regression.py | 7 | ||||
| -rw-r--r-- | numpy/lib/polynomial.py | 2 |
2 files changed, 8 insertions, 1 deletions
diff --git a/numpy/core/tests/test_regression.py b/numpy/core/tests/test_regression.py index 88187e35a..6d832d683 100644 --- a/numpy/core/tests/test_regression.py +++ b/numpy/core/tests/test_regression.py @@ -702,6 +702,13 @@ class test_regression(NumpyTestCase): from numpy.oldnumeric.random_array import randint randint(0,50,[2,3]) + def check_poly_div(self, level=rlevel): + """Ticket #553""" + u = N.poly1d([1,2,3]) + v = N.poly1d([1,2,3,4,5]) + q,r = N.polydiv(u,v) + assert_equal(q*v + r, u) + def check_poly_eq(self, level=rlevel): """Ticket #554""" x = N.poly1d([1,2,3]) diff --git a/numpy/lib/polynomial.py b/numpy/lib/polynomial.py index 8231cbd87..63ffec629 100644 --- a/numpy/lib/polynomial.py +++ b/numpy/lib/polynomial.py @@ -400,7 +400,7 @@ def polydiv(u, v): m = len(u) - 1 n = len(v) - 1 scale = 1. / v[0] - q = NX.zeros((m-n+1,), float) + q = NX.zeros((max(m-n+1,1),), float) r = u.copy() for k in range(0, m-n+1): d = scale * r[k] |