diff --git a/apptools/persistence/tests/test_state_pickler.py b/apptools/persistence/tests/test_state_pickler.py index 4def8df0..d095c3e3 100644 --- a/apptools/persistence/tests/test_state_pickler.py +++ b/apptools/persistence/tests/test_state_pickler.py @@ -172,7 +172,7 @@ def verify(self, obj, state): base64.decodebytes(data[num_attr]["data"]) ) num = pickle.loads(junk) - self.assertEqual(numpy.all(numpy.ravel(num == obj.numeric)), 1) + numpy.testing.assert_array_equal(num, obj.numeric, strict=True) self.assertIn(data["ref"]["type"], ["reference", "numeric"]) if data["ref"]["type"] == "numeric": @@ -218,7 +218,7 @@ def verify_unpickled(self, obj, state): self.assertEqual(dct["ref"].__metadata__["type"], "instance") num = state.numeric - self.assertEqual(numpy.all(numpy.ravel(num == obj.numeric)), 1) + numpy.testing.assert_array_equal(num, obj.numeric, strict=True) self.assertEqual(id(state.ref), id(num)) if TVTK_AVAILABLE: