Update docstring tests for np.array behaviour in numpy 2

bluesky · Aug 6, 2024 · a5632ed · a5632ed
1 parent ee67653
commit a5632ed
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/docs/how-to/iterate-a-spec.rst b/docs/how-to/iterate-a-spec.rst
@@ -18,9 +18,9 @@ frame. You can get these by using the `Spec.midpoints()` method to produce a
 >>> for d in spec.midpoints():
 ...     print(d)
 ...
-{'x': 1.0}
-{'x': 1.5}
-{'x': 2.0}
+{'x': np.float64(1.0)}
+{'x': np.float64(1.5)}
+{'x': np.float64(2.0)}
 
 This is simple, but not particularly performant, as the numpy arrays of
 points are unpacked point by point into point dictionaries

diff --git a/src/scanspec/core.py b/src/scanspec/core.py
@@ -613,10 +613,10 @@ class Midpoints(Generic[Axis]):
     >>> fy = Frames({"y": np.array([3, 4])})
     >>> mp = Midpoints([fy, fx])
     >>> for p in mp: print(p)
-    {'y': 3, 'x': 1}
-    {'y': 3, 'x': 2}
-    {'y': 4, 'x': 2}
-    {'y': 4, 'x': 1}
+    {'y': np.int64(3), 'x': np.int64(1)}
+    {'y': np.int64(3), 'x': np.int64(2)}
+    {'y': np.int64(4), 'x': np.int64(2)}
+    {'y': np.int64(4), 'x': np.int64(1)}
     """
 
     def __init__(self, stack: List[Frames[Axis]]):