Update tests/test_utils_npy.py

2025-08-10 22:52:21 +02:00
parent 59cd8b2a8f
commit 21e278447c
1 changed files with 3 additions and 1 deletions
@@ -34,7 +34,9 @@ def test_nice_linspace(start, stop, num, expected):
@pytest.mark.parametrize("start, stop, step, endpoint, expected", [
    (-2, 2, 2, True, np.array([-2., 0., 2.])),                 # symmetric around 0
    (0, 5, 2, True, np.array([0., 2., 4.])),                   # basic
-    (0, 5, 2, False, np.array([0., 2.])),       
+    (0, 5, 2, False, np.array([0., 2.])),     
+    (-1, 2, 1.5, True, np.array([-1.5, 0., 1.5])),              # step doesn't fit exactly
+    (-1,  2, -1.5,  True,  np.array([-1.5, 0., 1.5])),  
    ( 5,  0,  -2,   True,  np.array([ 4.,  2.,  0.])),

 ])