diff --git a/src/spikeinterface/comparison/tests/test_groundtruthcomparison.py b/src/spikeinterface/comparison/tests/test_groundtruthcomparison.py index 931c989cef..75bbfb5400 100644 --- a/src/spikeinterface/comparison/tests/test_groundtruthcomparison.py +++ b/src/spikeinterface/comparison/tests/test_groundtruthcomparison.py @@ -3,7 +3,7 @@ import pandas as pd -from spikeinterface.extractors import NumpySorting, toy_example +from spikeinterface.extractors import NumpySorting from spikeinterface.comparison import compare_sorter_to_ground_truth diff --git a/src/spikeinterface/comparison/tests/test_multisortingcomparison.py b/src/spikeinterface/comparison/tests/test_multisortingcomparison.py index 7e4c9ac77b..f39b8cd890 100644 --- 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_sorting +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_sorting(durations=[10] * num_segments) cmp_multi = compare_multiple_sorters([sort, sort, sort]) diff --git a/src/spikeinterface/comparison/tests/test_symmetricsortingcomparison.py b/src/spikeinterface/comparison/tests/test_symmetricsortingcomparison.py index abc1902e22..e505ced45e 100644 --- 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_sorting +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_sorting(durations=[10, 10]) cmp_multi = compare_two_sorters(sort, sort) diff --git a/src/spikeinterface/qualitymetrics/tests/test_pca_metrics.py b/src/spikeinterface/qualitymetrics/tests/test_pca_metrics.py index 916e9cefc2..e5196708e0 100644 --- a/src/spikeinterface/qualitymetrics/tests/test_pca_metrics.py +++ b/src/spikeinterface/qualitymetrics/tests/test_pca_metrics.py @@ -11,7 +11,6 @@ create_sorting_analyzer, ) -# from spikeinterface.extractors.toy_example import toy_example from spikeinterface.qualitymetrics.utils import create_ground_truth_pc_distributions from spikeinterface.qualitymetrics import (