diff options
author | Eric Wieser <wieser.eric@gmail.com> | 2017-07-08 22:39:55 +0200 |
---|---|---|
committer | Eric Wieser <wieser.eric@gmail.com> | 2017-07-08 22:39:55 +0200 |
commit | 389bd44e32b0eace0d024b126931a0a00d14cffe (patch) | |
tree | 3840e374eb3bdf928c26c40689b8b63e525aa804 /numpy/lib/tests | |
parent | 3c4f17b7d6d44b3076be9d50e305ed71836fa5d9 (diff) | |
download | numpy-389bd44e32b0eace0d024b126931a0a00d14cffe.tar.gz |
BUG: Return the coefficients array directly
Turns out that this was relied upon downstream
We also add a setter for coeffs, so that augmented assignment does not both
change state and raise an exception suggesting state could not be changed.
Diffstat (limited to 'numpy/lib/tests')
-rw-r--r-- | numpy/lib/tests/test_polynomial.py | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/numpy/lib/tests/test_polynomial.py b/numpy/lib/tests/test_polynomial.py index 0725c186d..2aed5c924 100644 --- a/numpy/lib/tests/test_polynomial.py +++ b/numpy/lib/tests/test_polynomial.py @@ -222,19 +222,19 @@ class TestDocs(TestCase): assert_equal(p == p2, False) assert_equal(p != p2, True) - def test_poly_coeffs_immutable(self): - """ Coefficients should not be modifiable """ + def test_poly_coeffs_mutable(self): + """ Coefficients should be modifiable """ p = np.poly1d([1, 2, 3]) - try: - # despite throwing an exception, this used to change state - p.coeffs += 1 - except Exception: - pass - assert_equal(p.coeffs, [1, 2, 3]) + p.coeffs += 1 + assert_equal(p.coeffs, [2, 3, 4]) p.coeffs[2] += 10 - assert_equal(p.coeffs, [1, 2, 3]) + assert_equal(p.coeffs, [2, 3, 14]) + + # this never used to be allowed - let's not add features to deprecated + # APIs + assert_raises(AttributeError, setattr, p, 'coeffs', np.array(1)) if __name__ == "__main__": |