fix(doctest): use legacy numpy scalar printing

radiocosmology · Sep 26, 2024 · e056d39 · e056d39
1 parent db040a9
commit e056d39
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/caput/memh5.py b/caput/memh5.py
@@ -80,6 +80,12 @@
 else:
     zarr_available = True
 
+# This needs to be set for doctests to pass on numpy 2.0. In numpy 2.0+,
+# the repr of a numpy scalar include the type information - i.e. a numpy
+# scalar is printed as 'np.float64(3.0)' rather than just '3.0'.
+# https://numpy.org/doc/stable/release/2.0.0-notes.html#representation-of-numpy-scalars-changed
+np.set_printoptions(legacy="1.25")
+
 # Basic Classes
 # -------------
 

diff --git a/caput/mpiarray.py b/caput/mpiarray.py
@@ -254,6 +254,12 @@
 
 logger = logging.getLogger(__name__)
 
+# This needs to be set for doctests to pass on numpy 2.0. In numpy 2.0+,
+# the repr of a numpy scalar include the type information - i.e. a numpy
+# scalar is printed as 'np.float64(3.0)' rather than just '3.0'.
+# https://numpy.org/doc/stable/release/2.0.0-notes.html#representation-of-numpy-scalars-changed
+np.set_printoptions(legacy="1.25")
+
 
 class _global_resolver:
     # Private class implementing the global sampling for MPIArray