diff --git a/tests/test_utils_npy.py b/tests/test_utils_npy.py index bdf711d67..c2c9abaaa 100644 --- a/tests/test_utils_npy.py +++ b/tests/test_utils_npy.py @@ -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.])), ])