File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -224,8 +224,14 @@ def test_sorting_s3_nwb_zarr(tmp_path):
224224 # just take 3 random units to test
225225 rng = np .random .default_rng (seed = 2205 )
226226 three_unit_ids = rng .choice (sorting .unit_ids , size = 3 )
227- sorting_sub = sorting .select_units (unit_ids = three_unit_ids )
228- sorting_loaded_sub = sorting_loaded .select_units (unit_ids = three_unit_ids )
227+
228+ first_spike_sample_index = 21864115
229+ sorting_sub = sorting .select_units (unit_ids = three_unit_ids ).frame_slice (
230+ start_frame = first_spike_sample_index , end_frame = first_spike_sample_index + 10_000
231+ )
232+ sorting_loaded_sub = sorting_loaded .select_units (unit_ids = three_unit_ids ).frame_slice (
233+ start_frame = first_spike_sample_index , end_frame = first_spike_sample_index + 10_000
234+ )
229235
230236 check_sortings_equal (sorting_sub , sorting_loaded_sub )
231237
You can’t perform that action at this time.
0 commit comments