Add test for temporal bins

SpikeInterface · Jul 17, 2024 · e368ea5 · e368ea5
1 parent 1b55388
commit e368ea5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/spikeinterface/sortingcomponents/motion/tests/test_motion_estimation.py b/src/spikeinterface/sortingcomponents/motion/tests/test_motion_estimation.py
@@ -171,6 +171,10 @@ def test_estimate_motion(dataset):
             else:
                 assert motion.displacement[0].shape[1] > 1
 
+            if rec.has_time_vector():
+                assert np.all(motion.temporal_bins_s[0] >= rec.get_times()[0])
+                assert np.all(motion.temporal_bins_s[0] <= rec.get_times()[-1])
+
             # # Test saving to disk
             # corrected_rec = InterpolateMotionRecording(
             #     recording, motion, temporal_bins, spatial_bins, border_mode="force_extrapolate"