Change default numpy2 printing options in test.py.

Change the default numpy 2 printing options that crash doctests on scalars.
nennigb · Jul 22, 2024 · 402f296 · 402f296
1 parent b2ddbd9
commit 402f296
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/polze/test.py b/polze/test.py
@@ -24,6 +24,11 @@
 import cmath
 import sys
 
+# Numpy 2.0 change default printing options making doctest failing.
+# https://numpy.org/neps/nep-0051-scalar-representation.html
+# Use legacy mode for testing
+if np.lib.NumpyVersion(np.__version__) >= '2.0.0b1':
+    np.set_printoptions(legacy="1.25")
 
 # Define a non vectorized function
 # For parallel processing we need a function defined outside the class.