test_symmetricsortingcomparison.py

SpikeInterface · May 29, 2024 · 916c26d · 916c26d
1 parent 790715c
commit 916c26d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/spikeinterface/comparison/tests/test_symmetricsortingcomparison.py b/src/spikeinterface/comparison/tests/test_symmetricsortingcomparison.py
@@ -1,6 +1,7 @@
 import numpy as np
 
-from spikeinterface.extractors import NumpySorting, toy_example
+from spikeinterface.core import generate_ground_truth_recording
+from spikeinterface.extractors import NumpySorting
 from spikeinterface.comparison import compare_two_sorters
 
 
@@ -29,7 +30,7 @@ def test_compare_two_sorters():
 
 
 def test_compare_multi_segment():
-    _, sort = toy_example(num_segments=2)
+    _, sort = generate_ground_truth_recording(durations=[10, 10])
 
     cmp_multi = compare_two_sorters(sort, sort)