Skip to content

Commit 1fb72b3

Browse files
committed
Update
1 parent d753cea commit 1fb72b3

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

tests/test_warping.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -338,4 +338,15 @@ def test_warp_tensor_with_ts():
338338
expected = np.tile((np.arange(1, 5, 1))[:,None], 10)
339339

340340
tensor = nap.warp_tensor(ts, ep, num_bin=10)
341-
np.testing.assert_array_almost_equal(tensor, expected)
341+
np.testing.assert_array_almost_equal(tensor, expected)
342+
343+
def test_warp_tensor_with_tsd():
344+
tsd = get_tsd()
345+
346+
# Equal
347+
ep = nap.IntervalSet(
348+
start=np.arange(20, 100, 20),
349+
end = np.arange(30, 110, 20)
350+
)
351+
expected = np.ones((4,10))
352+
tensor = nap.warp_tensor(tsd, ep, 10)

0 commit comments

Comments
 (0)