Skip to content

Commit e6c76f7

Browse files
committed
revert si notation change
1 parent cf01a25 commit e6c76f7

1 file changed

Lines changed: 11 additions & 10 deletions

File tree

tests/test_singularity_containers.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33

44
import pytest
55

6-
import spikeinterface.full as si
6+
import spikeinterface.extractors as se
7+
import spikeinterface.sorters as ss
78

89
os.environ['SINGULARITY_DISABLE_CACHE'] = 'true'
910

@@ -26,7 +27,7 @@ def work_dir(request, tmp_path):
2627

2728
@pytest.fixture
2829
def 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

3940
def 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

4445
def 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

4950
def 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

5455
def 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

5960
def 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

6465
def 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

6970
def 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

7475
def 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

Comments
 (0)