Merge pull request #19 from nennigb/fix-np2-new-scalar-formating

Use numpy legacy printing in tests to maintain doctest.
nennigb · Jul 23, 2024 · 13151c9 · 13151c9
2 parents 4329300 + 3792423
commit 13151c9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pypolsys/test.py b/pypolsys/test.py
@@ -25,6 +25,12 @@
 import os
 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")
+
 
 def sort_ref_found(ref, r):
     """ Sort reference solution and the found solution using