Skip to content

Commit e1cfc4b

Browse files
authored
Speed up test_sorting_s3_nwb_zarr (again) (#4538)
1 parent 34c83f7 commit e1cfc4b

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

src/spikeinterface/extractors/tests/test_nwbextractors_streaming.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)