TEST update test to pass with numpy 2 (#535)

cloudpipe · Jun 19, 2024 · bc677da · bc677da
1 parent 27c008e
commit bc677da
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -9,7 +9,7 @@ psutil
 tornado
 # To be able to test numpy specific things
 # but do not build numpy from source on Python nightly
-numpy >=1.18.5; python_version <= '3.8'
+numpy >=1.18.5; python_version <= '3.12'
 # Code coverage uploader for Travis:
 codecov
 coverage

diff --git a/tests/cloudpickle_test.py b/tests/cloudpickle_test.py
@@ -1501,7 +1501,10 @@ def test_importing_multiprocessing_does_not_impact_whichmodule(self):
         )
         out, _ = proc.communicate()
         self.assertEqual(proc.wait(), 0)
-        self.assertEqual(out, b"numpy.core._multiarray_umath\n")
+        assert out.strip() in (
+            b"numpy.core._multiarray_umath",  # numpy 1
+            b"numpy._core._multiarray_umath",  # numpy 2
+        )
 
     def test_unrelated_faulty_module(self):
         # Check that pickling a dynamically defined function or class does not