diff --git a/tests/test_interface.py b/tests/test_interface.py index 49a88fc..b70c7f8 100644 --- 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 \ No newline at end of file + 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 \ No newline at end of file