TST: added a test for negative axis arguments

andgoldschmidt · Jun 5, 2024 · 2bfeb89 · 2bfeb89
1 parent 8cde00d
commit 2bfeb89
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_interface.py b/tests/test_interface.py
@@ -98,4 +98,12 @@ def test_hyperparam_entrypoint():
     func = utils._load_hyperparam_func("kalman.default")
     expected = 1
     result = func(None, None)
-    assert result == expected
+    assert result == expected
+
+
+def test_negative_axis():
+    t = np.arange(3)
+    x = np.ones((2, 3, 2))
+    axis = -2
+    dx = dxdt(x, t, kind='finite_difference', axis=axis, k=1)
+    assert x.shape == dx.shape