33
44import pytest
55
6- import spikeinterface .full as si
6+ import spikeinterface .extractors as se
7+ import spikeinterface .sorters as ss
78
89os .environ ['SINGULARITY_DISABLE_CACHE' ] = 'true'
910
@@ -26,7 +27,7 @@ def work_dir(request, tmp_path):
2627
2728@pytest .fixture
2829def run_kwargs (work_dir ):
29- test_recording , _ = si .toy_example (
30+ test_recording , _ = se .toy_example (
3031 duration = 30 ,
3132 seed = 0 ,
3233 num_channels = 64 ,
@@ -37,42 +38,42 @@ def run_kwargs(work_dir):
3738
3839
3940def test_spykingcircus (run_kwargs ):
40- sorting = si .run_spykingcircus (output_folder = "spykingcircus" , ** run_kwargs )
41+ sorting = ss .run_spykingcircus (output_folder = "spykingcircus" , ** run_kwargs )
4142 print (sorting )
4243
4344
4445def test_mountainsort4 (run_kwargs ):
45- sorting = si .run_mountainsort4 (output_folder = "mountainsort4" , ** run_kwargs )
46+ sorting = ss .run_mountainsort4 (output_folder = "mountainsort4" , ** run_kwargs )
4647 print (sorting )
4748
4849
4950def test_tridesclous (run_kwargs ):
50- sorting = si .run_tridesclous (output_folder = "tridesclous" , ** run_kwargs )
51+ sorting = ss .run_tridesclous (output_folder = "tridesclous" , ** run_kwargs )
5152 print (sorting )
5253
5354
5455def test_klusta (run_kwargs ):
55- sorting = si .run_klusta (output_folder = "klusta" , ** run_kwargs )
56+ sorting = ss .run_klusta (output_folder = "klusta" , ** run_kwargs )
5657 print (sorting )
5758
5859
5960def test_ironclust (run_kwargs ):
60- sorting = si .run_ironclust (output_folder = "ironclust" , fGpu = False , ** run_kwargs )
61+ sorting = ss .run_ironclust (output_folder = "ironclust" , fGpu = False , ** run_kwargs )
6162 print (sorting )
6263
6364
6465def test_waveclus (run_kwargs ):
65- sorting = si .run_waveclus (output_folder = "waveclus" , ** run_kwargs )
66+ sorting = ss .run_waveclus (output_folder = "waveclus" , ** run_kwargs )
6667 print (sorting )
6768
6869
6970def test_hdsort (run_kwargs ):
70- sorting = si .run_hdsort (output_folder = "hdsort" , ** run_kwargs )
71+ sorting = ss .run_hdsort (output_folder = "hdsort" , ** run_kwargs )
7172 print (sorting )
7273
7374
7475def test_kilosort1 (run_kwargs ):
75- sorting = si .run_kilosort (output_folder = "kilosort" , useGPU = False , ** run_kwargs )
76+ sorting = ss .run_kilosort (output_folder = "kilosort" , useGPU = False , ** run_kwargs )
7677 print (sorting )
7778
7879
0 commit comments