diff --git a/apptools/persistence/tests/test_state_pickler.py b/apptools/persistence/tests/test_state_pickler.py index ee8bb608..d0de88d9 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.alltrue(numpy.ravel(num == obj.numeric)), 1) + numpy.testing.assert_array_equal(num, obj.numeric) 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.alltrue(numpy.ravel(num == obj.numeric)), 1) + numpy.testing.assert_array_equal(num, obj.numeric) self.assertEqual(id(state.ref), id(num)) if TVTK_AVAILABLE: diff --git a/docs/releases/upcoming/341.bugfix.rst b/docs/releases/upcoming/341.bugfix.rst new file mode 100644 index 00000000..cd486f15 --- /dev/null +++ b/docs/releases/upcoming/341.bugfix.rst @@ -0,0 +1 @@ +Replaced uses of ``numpy.alltrue``, for compatibility with NumPy 2.0.