From cc7800d0bd4fd9fd68d2e81d02a6f48b1482ddf9 Mon Sep 17 00:00:00 2001 From: Matteo Raso Date: Sat, 26 Nov 2022 16:14:09 -0500 Subject: [PATCH] Added pickle test for polynomials --- numpy/polynomial/tests/test_polynomial.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/numpy/polynomial/tests/test_polynomial.py b/numpy/polynomial/tests/test_polynomial.py index 032bf0e76af9..6b3ef2388f63 100644 --- a/numpy/polynomial/tests/test_polynomial.py +++ b/numpy/polynomial/tests/test_polynomial.py @@ -5,6 +5,7 @@ import numpy as np import numpy.polynomial.polynomial as poly +import pickle from copy import deepcopy from numpy.testing import ( assert_almost_equal, assert_raises, assert_equal, assert_, @@ -47,6 +48,11 @@ def test_copy(self): y = deepcopy(x) assert_equal(x, y) + def test_pickle(self): + x = poly.Polynomial([1, 2, 3]) + y = pickle.loads(pickle.dumps(x)) + assert_equal(x, y) + class TestArithmetic: def test_polyadd(self):