TST: add test for dxdt on axis 2

Jacob-Stevens-Haas · May 14, 2024 · a76b94b · a76b94b
1 parent f0d566d
commit a76b94b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_interface.py b/tests/test_interface.py
@@ -33,6 +33,10 @@ def test_axis():
     shape2 = ts.shape
     test3 = dxdt(ts, ts, axis=1).shape
     assert shape2 == test3
+    n = 4
+    x3d = np.tile(np.arange(n), (n, n, 1))
+    test_4 = dxdt(x3d, np.arange(n), axis=2)
+    np.testing.assert_array_equal(test_4, np.ones((n,n,n)))
 
 
 def test_empty():