Merge pull request #3510 from mwtoews/master

Add regression tests for pickleable record arrays

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]