diff --git a/src/spikeinterface/extractors/tests/test_nwbextractors_streaming.py b/src/spikeinterface/extractors/tests/test_nwbextractors_streaming.py index 045e7f8211..592aecdb16 100644 --- a/src/spikeinterface/extractors/tests/test_nwbextractors_streaming.py +++ b/src/spikeinterface/extractors/tests/test_nwbextractors_streaming.py @@ -220,12 +220,17 @@ def test_sorting_s3_nwb_zarr(tmp_path): # test to/from dict sorting_loaded = load(sorting.to_dict()) - # just take 3 random units to test rng = np.random.default_rng(seed=2205) three_unit_ids = rng.choice(sorting.unit_ids, size=3) - sorting_sub = sorting.select_units(unit_ids=three_unit_ids) - sorting_loaded_sub = sorting_loaded.select_units(unit_ids=three_unit_ids) + + first_spike_sample_index = 21864115 + sorting_sub = sorting.select_units(unit_ids=three_unit_ids).frame_slice( + start_frame=first_spike_sample_index, end_frame=first_spike_sample_index + 10_000 + ) + sorting_loaded_sub = sorting_loaded.select_units(unit_ids=three_unit_ids).frame_slice( + start_frame=first_spike_sample_index, end_frame=first_spike_sample_index + 10_000 + ) check_sortings_equal(sorting_sub, sorting_loaded_sub)