Replace with a more idiomatic test

enthought · Jun 28, 2024 · 15fe026 · 15fe026
1 parent 4f8f916
commit 15fe026
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: