diff options
Diffstat (limited to 'numpy/core')
| -rw-r--r-- | numpy/core/tests/test_records.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/numpy/core/tests/test_records.py b/numpy/core/tests/test_records.py index 753dec13e..8c9ce5c70 100644 --- a/numpy/core/tests/test_records.py +++ b/numpy/core/tests/test_records.py @@ -7,6 +7,7 @@ from numpy.compat import asbytes, asunicode import warnings import collections +import pickle class TestFromrecords(TestCase): @@ -146,6 +147,17 @@ class TestRecord(TestCase): y = self.data[['col2', 'col1']] assert_equal(x[0][0], y[0][1]) + def test_pickle_1(self): + # Issue #1529 + a = np.array([(1, [])], dtype=[('a', np.int32), ('b', np.int32, 0)]) + assert_equal(a, pickle.loads(pickle.dumps(a))) + assert_equal(a[0], pickle.loads(pickle.dumps(a[0]))) + + def test_pickle_2(self): + a = self.data + assert_equal(a, pickle.loads(pickle.dumps(a))) + assert_equal(a[0], pickle.loads(pickle.dumps(a[0]))) + def test_find_duplicate(): l1 = [1, 2, 3, 4, 5, 6] |