diff options
| -rw-r--r-- | numpy/core/tests/test_multiarray.py | 2 | ||||
| -rw-r--r-- | numpy/matrixlib/tests/test_masked_matrix.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index 2d3541fc5..235e5bc2e 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -3562,7 +3562,7 @@ class TestPickling(object): for a in DATA: assert_equal( - a, pickle.loads(a.dumps(protocol=proto)), + a, pickle.loads(pickle.dumps(protocol=proto)), err_msg="%r" % a) def _loads(self, obj): diff --git a/numpy/matrixlib/tests/test_masked_matrix.py b/numpy/matrixlib/tests/test_masked_matrix.py index 219349f93..2f7ea5632 100644 --- a/numpy/matrixlib/tests/test_masked_matrix.py +++ b/numpy/matrixlib/tests/test_masked_matrix.py @@ -80,7 +80,7 @@ class TestMaskedMatrix(object): # Test pickling w/ a subclass of ndarray a = masked_array(np.matrix(list(range(10))), mask=[1, 0, 1, 0, 0] * 2) for proto in range(2, pickle.HIGHEST_PROTOCOL + 1): - a_pickled = pickle.loads(a.dumps(protocol=proto)) + a_pickled = pickle.loads(pickle.dumps(protocol=proto)) assert_equal(a_pickled._mask, a._mask) assert_equal(a_pickled, a) assert_(isinstance(a_pickled._data, np.matrix)) |