Skip to content

Commit 1ffd1c8

Browse files
authored
Fix module level test groups (#4573)
1 parent 4faf243 commit 1ffd1c8

2 files changed

Lines changed: 27 additions & 4 deletions

File tree

pyproject.toml

Lines changed: 23 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,11 @@ test-streaming-extractors = [
244244
{include-group = "test-common"},
245245
"pooch>=1.8.2",
246246
"datalad>=1.0.2",
247+
"remfile",
248+
"requests",
249+
"aiohttp",
250+
"fsspec",
251+
"s3fs",
247252
]
248253

249254
test-preprocessing = [
@@ -254,9 +259,17 @@ test-preprocessing = [
254259
"torch", # preprocessing tests exercise sortingcomponents.motion.dredge
255260
]
256261

257-
test-postprocessing = [{include-group = "test-common"}]
262+
test-postprocessing = [
263+
{include-group = "test-common"},
264+
"pandas<3",
265+
]
258266
test-metrics = [{include-group = "test-common"}]
259-
test-comparison = [{include-group = "test-common"}]
267+
268+
test-comparison = [
269+
{include-group = "test-common"},
270+
"numba>=0.59"
271+
]
272+
260273
test-sorters-internal = [
261274
{include-group = "test-common"},
262275
"torch", # spyking_circus2 template matching
@@ -278,8 +291,15 @@ test-exporters = [{include-group = "test-common"}]
278291
test-sortingcomponents = [
279292
{include-group = "test-common"},
280293
"torch", # neural network peak detection, motion, matching
294+
"huggingface_hub", # for `test_single_channel_toy_denoiser_in_peak_pipeline`
281295
]
282-
test-generation = [{include-group = "test-common"}]
296+
test-generation = [
297+
{include-group = "test-common"},
298+
"fsspec",
299+
"s3fs",
300+
"numba>=0.59", # numba and sklearn needed for motion correction of
301+
"scikit-learn", # syntheic data used in `test_generate_hybrid_motion`
302+
]
283303

284304
# Aggregate group for workflows that run the full test suite (nightly codecov).
285305
test-all = [

src/spikeinterface/generation/tests/test_hybrid_tools.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,10 @@ def test_generate_hybrid_with_sorting():
3939
def test_generate_hybrid_motion():
4040
rec, _ = generate_ground_truth_recording(sampling_frequency=20000, durations=[10], num_channels=16, seed=0)
4141
_, motion_info = correct_motion(
42-
rec, output_motion_info=True, estimate_motion_kwargs={"win_step_um": 20, "win_scale_um": 20}
42+
rec,
43+
output_motion_info=True,
44+
preset="nonrigid_fast_and_accurate",
45+
estimate_motion_kwargs={"win_step_um": 20, "win_scale_um": 20},
4346
)
4447
motion = motion_info["motion"]
4548
hybrid, sorting_hybrid = generate_hybrid_recording(rec, motion=motion, seed=0)

0 commit comments

Comments
 (0)