test_multisortingcomparison.py

SpikeInterface · May 29, 2024 · 17d08e5 · 17d08e5
1 parent 0968ba5
commit 17d08e5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/spikeinterface/comparison/tests/test_multisortingcomparison.py b/src/spikeinterface/comparison/tests/test_multisortingcomparison.py
@@ -5,7 +5,8 @@
 import pytest
 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_multiple_sorters, MultiSortingComparison
 
 if hasattr(pytest, "global_test_folder"):
@@ -72,7 +73,7 @@ def test_compare_multiple_sorters():
 
 def test_compare_multi_segment():
     num_segments = 3
-    _, sort = toy_example(num_segments=num_segments)
+    _, sort = generate_ground_truth_recording(durations=[10] * num_segments)
 
     cmp_multi = compare_multiple_sorters([sort, sort, sort])