diff options
| author | Eric Wieser <wieser.eric@gmail.com> | 2017-03-09 02:57:18 +0000 |
|---|---|---|
| committer | Eric Wieser <wieser.eric@gmail.com> | 2017-03-09 02:57:18 +0000 |
| commit | 18b7cd9df7a4d960550b18faa14d5473e7d5c3d9 (patch) | |
| tree | a6e1cbeda9b501c2b29121c0b9ca9fa94c95b4bf /numpy/lib/tests/test_polynomial.py | |
| parent | 6a3edf3210b439a4d1a51acb4e01bac017697ee6 (diff) | |
| download | numpy-18b7cd9df7a4d960550b18faa14d5473e7d5c3d9.tar.gz | |
BUG: Prevent crash in poly1d.__eq__
Fixes #8760
Diffstat (limited to 'numpy/lib/tests/test_polynomial.py')
| -rw-r--r-- | numpy/lib/tests/test_polynomial.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_polynomial.py b/numpy/lib/tests/test_polynomial.py index 00dffd3d3..f1e4543bb 100644 --- a/numpy/lib/tests/test_polynomial.py +++ b/numpy/lib/tests/test_polynomial.py @@ -213,6 +213,15 @@ class TestDocs(TestCase): v = np.arange(1, 21) assert_almost_equal(np.poly(v), np.poly(np.diag(v))) + def test_poly_eq(self): + p = np.poly1d([1, 2, 3]) + p2 = np.poly1d([1, 2, 4]) + assert_equal(p == None, False) + assert_equal(p != None, True) + assert_equal(p == p, True) + assert_equal(p == p2, False) + assert_equal(p != p2, True) + if __name__ == "__main__": run_module_suite() |