add round trip pickle test

bfgray3 · bfgray3 · commit 866b39e8217e · 2020-09-28T20:38:27.000-04:00
diff --git a/numpy/core/tests/test__exceptions.py b/numpy/core/tests/test__exceptions.py
@@ -48,5 +48,9 @@ def test__total_size(self):
 
 class TestUFuncNoLoopError:
     def test_pickling(self):
-        """ Test that _UFuncNoLoopError can be pickled """
+        """ Test that _UFuncNoLoopError, _ArrayMemoryError can be pickled """
         assert isinstance(pickle.dumps(_UFuncNoLoopError), bytes)
+
+        error = _ArrayMemoryError((1023,), np.dtype(np.uint8))
+        res = pickle.loads(pickle.dumps(error))
+        assert res._total_size == error._total_size
