From fa0a9b85073c299609c4d93120c21de2f63e6700 Mon Sep 17 00:00:00 2001
From: paulrignanese <paul.rignanese@gmail.com>
Date: Wed, 29 May 2024 12:17:12 +0100
Subject: [PATCH 1/5] Replace toy_example by generate_ground_truth_recording in
 sorters folder

---
 .../sorters/external/tests/test_docker_containers.py       | 3 ++-
 .../sorters/external/tests/test_kilosort4.py               | 5 ++---
 .../sorters/external/tests/test_pykilosort.py              | 1 -
 .../sorters/external/tests/test_singularity_containers.py  | 3 ++-
 .../external/tests/test_singularity_containers_gpu.py      | 3 ++-
 src/spikeinterface/sorters/external/yass.py                | 2 +-
 src/spikeinterface/sorters/tests/common_tests.py           | 6 +++---
 src/spikeinterface/sorters/tests/test_container_tools.py   | 7 ++++---
 src/spikeinterface/sorters/tests/test_launcher.py          | 1 -
 9 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/src/spikeinterface/sorters/external/tests/test_docker_containers.py b/src/spikeinterface/sorters/external/tests/test_docker_containers.py
index 4af25bc9f0..9bf17549bf 100644
--- a/src/spikeinterface/sorters/external/tests/test_docker_containers.py
+++ b/src/spikeinterface/sorters/external/tests/test_docker_containers.py
@@ -4,6 +4,7 @@
 import pytest
 from pathlib import Path
 
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.core.core_tools import is_editable_mode
 import spikeinterface.extractors as se
 import spikeinterface.sorters as ss
@@ -23,7 +24,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording, _ = se.toy_example(duration=30, seed=0, num_channels=64, num_segments=1)
+    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/external/tests/test_kilosort4.py b/src/spikeinterface/sorters/external/tests/test_kilosort4.py
index 87346d1dbb..62a1476407 100644
--- a/src/spikeinterface/sorters/external/tests/test_kilosort4.py
+++ b/src/spikeinterface/sorters/external/tests/test_kilosort4.py
@@ -2,8 +2,7 @@
 import pytest
 from pathlib import Path
 
-from spikeinterface import load_extractor
-from spikeinterface.extractors import toy_example
+from spikeinterface import load_extractor, generate_ground_truth_recording
 from spikeinterface.sorters import Kilosort4Sorter, run_sorter
 from spikeinterface.sorters.tests.common_tests import SorterCommonTestSuite
 
@@ -23,7 +22,7 @@ def setUp(self):
         if (cache_folder / "rec").is_dir():
             recording = load_extractor(cache_folder / "rec")
         else:
-            recording, _ = toy_example(num_channels=32, duration=60, seed=0, num_segments=1)
+            recording, _ = generate_ground_truth_recording(num_channels=32, durations=[60], seed=0)
             recording = recording.save(folder=cache_folder / "rec", verbose=False, format="binary")
         self.recording = recording
         print(self.recording)
diff --git a/src/spikeinterface/sorters/external/tests/test_pykilosort.py b/src/spikeinterface/sorters/external/tests/test_pykilosort.py
index a5a1705613..6f1788f46a 100644
--- a/src/spikeinterface/sorters/external/tests/test_pykilosort.py
+++ b/src/spikeinterface/sorters/external/tests/test_pykilosort.py
@@ -1,7 +1,6 @@
 import unittest
 
 import pytest
-from spikeinterface.extractors import toy_example
 from spikeinterface.sorters import PyKilosortSorter
 from spikeinterface.sorters.tests.common_tests import SorterCommonTestSuite
 
diff --git a/src/spikeinterface/sorters/external/tests/test_singularity_containers.py b/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
index 6c4d096104..3929c6407b 100644
--- a/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
+++ b/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
@@ -3,6 +3,7 @@
 import pytest
 from pathlib import Path
 
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.core.core_tools import is_editable_mode
 import spikeinterface.extractors as se
 import spikeinterface.sorters as ss
@@ -29,7 +30,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording, _ = se.toy_example(duration=30, seed=0, num_channels=64, num_segments=1)
+    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py b/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
index 8032826172..12920e90d8 100644
--- a/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
+++ b/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
@@ -3,6 +3,7 @@
 
 import pytest
 
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.core.core_tools import is_editable_mode
 import spikeinterface.extractors as se
 import spikeinterface.sorters as ss
@@ -23,7 +24,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording, _ = se.toy_example(duration=30, seed=0, num_channels=64, num_segments=1)
+    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/external/yass.py b/src/spikeinterface/sorters/external/yass.py
index 29fae7c114..9f948a3a79 100644
--- a/src/spikeinterface/sorters/external/yass.py
+++ b/src/spikeinterface/sorters/external/yass.py
@@ -96,7 +96,7 @@ class YassSorter(BaseSorter):
 
                             1.  Retraining Neural Networks (Default)
 
-                            rec, sort = se.toy_example(duration=300)
+                            rec, sort = generate_ground_truth_recording(durations=[300])
                             sorting_yass = ss.run_yass(rec, '/home/cat/Downloads/test2')
 
 
diff --git a/src/spikeinterface/sorters/tests/common_tests.py b/src/spikeinterface/sorters/tests/common_tests.py
index 1aa36a0e43..b3d7a04bcf 100644
--- a/src/spikeinterface/sorters/tests/common_tests.py
+++ b/src/spikeinterface/sorters/tests/common_tests.py
@@ -4,7 +4,7 @@
 from pathlib import Path
 import shutil
 
-from spikeinterface.extractors import toy_example
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.sorters import run_sorter
 from spikeinterface.core.snippets_tools import snippets_from_sorting
 
@@ -24,7 +24,7 @@ class SorterCommonTestSuite:
     SorterClass = None
 
     def setUp(self):
-        recording, sorting_gt = toy_example(num_channels=4, duration=60, seed=0, num_segments=1)
+        recording, sorting_gt = generate_ground_truth_recording(num_channels=4, durations=[60], seed=0)
         rec_folder = cache_folder / "rec"
         if rec_folder.is_dir():
             shutil.rmtree(rec_folder)
@@ -80,7 +80,7 @@ class SnippetsSorterCommonTestSuite:
     SorterClass = None
 
     def setUp(self):
-        recording, sorting_gt = toy_example(num_channels=4, duration=60, seed=0, num_segments=1)
+        recording, sorting_gt = generate_ground_truth_recording(num_channels=4, durations=[60], seed=0)
         snippets_folder = cache_folder / "snippets"
         if snippets_folder.is_dir():
             shutil.rmtree(snippets_folder)
diff --git a/src/spikeinterface/sorters/tests/test_container_tools.py b/src/spikeinterface/sorters/tests/test_container_tools.py
index 2247385839..16d1e0a4a4 100644
--- a/src/spikeinterface/sorters/tests/test_container_tools.py
+++ b/src/spikeinterface/sorters/tests/test_container_tools.py
@@ -5,7 +5,8 @@
 import os
 
 import spikeinterface as si
-from spikeinterface.extractors import toy_example
+from spikeinterface import generate_ground_truth_recording
+
 from spikeinterface.sorters.container_tools import find_recording_folders, ContainerClient, install_package_in_container
 
 ON_GITHUB = bool(os.getenv("GITHUB_ACTIONS"))
@@ -21,10 +22,10 @@ def setup_module():
     for test_dir in test_dirs:
         if test_dir.exists():
             shutil.rmtree(test_dir)
-    rec1, _ = toy_example(num_segments=1)
+    rec1, _ = generate_ground_truth_recording(durations=[10])
     rec1 = rec1.save(folder=cache_folder / "mono")
 
-    rec2, _ = toy_example(num_segments=3)
+    rec2, _ = generate_ground_truth_recording(durations=[10, 10, 10])
     rec2 = rec2.save(folder=cache_folder / "multi")
 
 
diff --git a/src/spikeinterface/sorters/tests/test_launcher.py b/src/spikeinterface/sorters/tests/test_launcher.py
index fa6b6986ba..b67385fc7c 100644
--- a/src/spikeinterface/sorters/tests/test_launcher.py
+++ b/src/spikeinterface/sorters/tests/test_launcher.py
@@ -8,7 +8,6 @@
 
 from spikeinterface.core import load_extractor
 
-# from spikeinterface.extractors import toy_example
 from spikeinterface import generate_ground_truth_recording
 from spikeinterface.sorters import run_sorter_jobs, run_sorter_by_property
 

From 2a7f2650d97f08430f4d645d808a9ed1880da39e Mon Sep 17 00:00:00 2001
From: paulrignanese <paul.rignanese@gmail.com>
Date: Wed, 29 May 2024 13:28:50 +0100
Subject: [PATCH 2/5] Replace generate_ground_truth_recording by
 generate_recording in sorters folder when only the recording is required

---
 .../sorters/external/tests/test_docker_containers.py        | 4 ++--
 src/spikeinterface/sorters/external/tests/test_kilosort4.py | 4 ++--
 .../sorters/external/tests/test_singularity_containers.py   | 4 ++--
 .../external/tests/test_singularity_containers_gpu.py       | 6 +++---
 src/spikeinterface/sorters/tests/test_container_tools.py    | 6 +++---
 src/spikeinterface/sorters/tests/test_launcher.py           | 4 ++--
 6 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/src/spikeinterface/sorters/external/tests/test_docker_containers.py b/src/spikeinterface/sorters/external/tests/test_docker_containers.py
index 9bf17549bf..b917ac345d 100644
--- a/src/spikeinterface/sorters/external/tests/test_docker_containers.py
+++ b/src/spikeinterface/sorters/external/tests/test_docker_containers.py
@@ -4,7 +4,7 @@
 import pytest
 from pathlib import Path
 
-from spikeinterface import generate_ground_truth_recording
+from spikeinterface import generate_ground_truth_recording, generate_recording
 from spikeinterface.core.core_tools import is_editable_mode
 import spikeinterface.extractors as se
 import spikeinterface.sorters as ss
@@ -24,7 +24,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
+    test_recording = generate_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/external/tests/test_kilosort4.py b/src/spikeinterface/sorters/external/tests/test_kilosort4.py
index 62a1476407..a5d4e2cf54 100644
--- a/src/spikeinterface/sorters/external/tests/test_kilosort4.py
+++ b/src/spikeinterface/sorters/external/tests/test_kilosort4.py
@@ -2,7 +2,7 @@
 import pytest
 from pathlib import Path
 
-from spikeinterface import load_extractor, generate_ground_truth_recording
+from spikeinterface import load_extractor, generate_ground_truth_recording, generate_recording
 from spikeinterface.sorters import Kilosort4Sorter, run_sorter
 from spikeinterface.sorters.tests.common_tests import SorterCommonTestSuite
 
@@ -22,7 +22,7 @@ def setUp(self):
         if (cache_folder / "rec").is_dir():
             recording = load_extractor(cache_folder / "rec")
         else:
-            recording, _ = generate_ground_truth_recording(num_channels=32, durations=[60], seed=0)
+            recording = generate_recording(num_channels=32, durations=[60], seed=0)
             recording = recording.save(folder=cache_folder / "rec", verbose=False, format="binary")
         self.recording = recording
         print(self.recording)
diff --git a/src/spikeinterface/sorters/external/tests/test_singularity_containers.py b/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
index 3929c6407b..859809139b 100644
--- a/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
+++ b/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
@@ -3,7 +3,7 @@
 import pytest
 from pathlib import Path
 
-from spikeinterface import generate_ground_truth_recording
+from spikeinterface import generate_ground_truth_recording, generate_recording
 from spikeinterface.core.core_tools import is_editable_mode
 import spikeinterface.extractors as se
 import spikeinterface.sorters as ss
@@ -30,7 +30,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
+    test_recording = generate_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py b/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
index 12920e90d8..deb72625d8 100644
--- a/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
+++ b/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
@@ -3,9 +3,9 @@
 
 import pytest
 
-from spikeinterface import generate_ground_truth_recording
+from spikeinterface import generate_ground_truth_recording, generate_recording
 from spikeinterface.core.core_tools import is_editable_mode
-import spikeinterface.extractors as se
+
 import spikeinterface.sorters as ss
 
 os.environ["SINGULARITY_DISABLE_CACHE"] = "true"
@@ -24,7 +24,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
+    test_recording = generate_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/tests/test_container_tools.py b/src/spikeinterface/sorters/tests/test_container_tools.py
index 16d1e0a4a4..776e62f9fe 100644
--- a/src/spikeinterface/sorters/tests/test_container_tools.py
+++ b/src/spikeinterface/sorters/tests/test_container_tools.py
@@ -5,7 +5,7 @@
 import os
 
 import spikeinterface as si
-from spikeinterface import generate_ground_truth_recording
+from spikeinterface import generate_ground_truth_recording, generate_recording
 
 from spikeinterface.sorters.container_tools import find_recording_folders, ContainerClient, install_package_in_container
 
@@ -22,10 +22,10 @@ def setup_module():
     for test_dir in test_dirs:
         if test_dir.exists():
             shutil.rmtree(test_dir)
-    rec1, _ = generate_ground_truth_recording(durations=[10])
+    rec1 = generate_recording(durations=[10])
     rec1 = rec1.save(folder=cache_folder / "mono")
 
-    rec2, _ = generate_ground_truth_recording(durations=[10, 10, 10])
+    rec2 = generate_recording(durations=[10, 10, 10])
     rec2 = rec2.save(folder=cache_folder / "multi")
 
 
diff --git a/src/spikeinterface/sorters/tests/test_launcher.py b/src/spikeinterface/sorters/tests/test_launcher.py
index b67385fc7c..f96d4db635 100644
--- a/src/spikeinterface/sorters/tests/test_launcher.py
+++ b/src/spikeinterface/sorters/tests/test_launcher.py
@@ -8,7 +8,7 @@
 
 from spikeinterface.core import load_extractor
 
-from spikeinterface import generate_ground_truth_recording
+from spikeinterface import generate_ground_truth_recording, generate_recording
 from spikeinterface.sorters import run_sorter_jobs, run_sorter_by_property
 
 
@@ -27,7 +27,7 @@
 def setup_module():
     base_seed = 42
     for i in range(num_recordings):
-        rec, _ = generate_ground_truth_recording(num_channels=8, durations=[10.0], seed=base_seed + i)
+        rec = generate_recording(num_channels=8, durations=[10.0], seed=base_seed + i)
         rec_folder = cache_folder / f"toy_rec_{i}"
         if rec_folder.is_dir():
             shutil.rmtree(rec_folder)

From 5c793ed88b191cbd7c3413d69b1e114d8871ef21 Mon Sep 17 00:00:00 2001
From: paulrignanese <paul.rignanese@gmail.com>
Date: Wed, 29 May 2024 15:47:59 +0100
Subject: [PATCH 3/5] remove the setting of cache folder as global variable for
 test_phase_shift in preprocessing/tests folder

---
 src/spikeinterface/preprocessing/tests/test_phase_shift.py | 7 -------
 1 file changed, 7 deletions(-)

diff --git a/src/spikeinterface/preprocessing/tests/test_phase_shift.py b/src/spikeinterface/preprocessing/tests/test_phase_shift.py
index b1ccc433b3..11f0a9c762 100644
--- a/src/spikeinterface/preprocessing/tests/test_phase_shift.py
+++ b/src/spikeinterface/preprocessing/tests/test_phase_shift.py
@@ -13,13 +13,6 @@
 
 import scipy.fft
 
-if hasattr(pytest, "global_test_folder"):
-    cache_folder = pytest.global_test_folder / "preprocessing"
-else:
-    cache_folder = Path("cache_folder") / "preprocessing"
-
-set_global_tmp_folder(cache_folder)
-
 
 def create_shifted_channel():
     duration = 5.0

From c94f1c74ce255be285e6d8ad397edc5bba1b3dbc Mon Sep 17 00:00:00 2001
From: JoeZiminski <joseph.j.ziminski@gmail.com>
Date: Thu, 30 May 2024 11:15:13 +0100
Subject: [PATCH 4/5] Revert back to generate_ground_truth_recording as
 generate_recording is noise only (no spikes).

---
 .../sorters/external/tests/test_docker_containers.py          | 4 ++--
 src/spikeinterface/sorters/external/tests/test_kilosort4.py   | 4 ++--
 .../sorters/external/tests/test_singularity_containers.py     | 4 ++--
 .../sorters/external/tests/test_singularity_containers_gpu.py | 4 ++--
 src/spikeinterface/sorters/tests/test_container_tools.py      | 4 ++--
 src/spikeinterface/sorters/tests/test_launcher.py             | 4 ++--
 6 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/src/spikeinterface/sorters/external/tests/test_docker_containers.py b/src/spikeinterface/sorters/external/tests/test_docker_containers.py
index b917ac345d..9bf17549bf 100644
--- a/src/spikeinterface/sorters/external/tests/test_docker_containers.py
+++ b/src/spikeinterface/sorters/external/tests/test_docker_containers.py
@@ -4,7 +4,7 @@
 import pytest
 from pathlib import Path
 
-from spikeinterface import generate_ground_truth_recording, generate_recording
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.core.core_tools import is_editable_mode
 import spikeinterface.extractors as se
 import spikeinterface.sorters as ss
@@ -24,7 +24,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording = generate_recording(durations=[30], seed=0, num_channels=64)
+    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/external/tests/test_kilosort4.py b/src/spikeinterface/sorters/external/tests/test_kilosort4.py
index a5d4e2cf54..62a1476407 100644
--- a/src/spikeinterface/sorters/external/tests/test_kilosort4.py
+++ b/src/spikeinterface/sorters/external/tests/test_kilosort4.py
@@ -2,7 +2,7 @@
 import pytest
 from pathlib import Path
 
-from spikeinterface import load_extractor, generate_ground_truth_recording, generate_recording
+from spikeinterface import load_extractor, generate_ground_truth_recording
 from spikeinterface.sorters import Kilosort4Sorter, run_sorter
 from spikeinterface.sorters.tests.common_tests import SorterCommonTestSuite
 
@@ -22,7 +22,7 @@ def setUp(self):
         if (cache_folder / "rec").is_dir():
             recording = load_extractor(cache_folder / "rec")
         else:
-            recording = generate_recording(num_channels=32, durations=[60], seed=0)
+            recording, _ = generate_ground_truth_recording(num_channels=32, durations=[60], seed=0)
             recording = recording.save(folder=cache_folder / "rec", verbose=False, format="binary")
         self.recording = recording
         print(self.recording)
diff --git a/src/spikeinterface/sorters/external/tests/test_singularity_containers.py b/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
index 859809139b..3929c6407b 100644
--- a/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
+++ b/src/spikeinterface/sorters/external/tests/test_singularity_containers.py
@@ -3,7 +3,7 @@
 import pytest
 from pathlib import Path
 
-from spikeinterface import generate_ground_truth_recording, generate_recording
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.core.core_tools import is_editable_mode
 import spikeinterface.extractors as se
 import spikeinterface.sorters as ss
@@ -30,7 +30,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording = generate_recording(durations=[30], seed=0, num_channels=64)
+    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py b/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
index deb72625d8..a393fde105 100644
--- a/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
+++ b/src/spikeinterface/sorters/external/tests/test_singularity_containers_gpu.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from spikeinterface import generate_ground_truth_recording, generate_recording
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.core.core_tools import is_editable_mode
 
 import spikeinterface.sorters as ss
@@ -24,7 +24,7 @@ def check_gh_settings():
 
 
 def generate_run_kwargs():
-    test_recording = generate_recording(durations=[30], seed=0, num_channels=64)
+    test_recording, _ = generate_ground_truth_recording(durations=[30], seed=0, num_channels=64)
     test_recording = test_recording.save(name="toy")
     test_recording.set_channel_gains(1)
     test_recording.set_channel_offsets(0)
diff --git a/src/spikeinterface/sorters/tests/test_container_tools.py b/src/spikeinterface/sorters/tests/test_container_tools.py
index 776e62f9fe..030585a431 100644
--- a/src/spikeinterface/sorters/tests/test_container_tools.py
+++ b/src/spikeinterface/sorters/tests/test_container_tools.py
@@ -22,10 +22,10 @@ def setup_module():
     for test_dir in test_dirs:
         if test_dir.exists():
             shutil.rmtree(test_dir)
-    rec1 = generate_recording(durations=[10])
+    rec1, _ = generate_ground_truth_recording(durations=[10])
     rec1 = rec1.save(folder=cache_folder / "mono")
 
-    rec2 = generate_recording(durations=[10, 10, 10])
+    rec2, _ = generate_ground_truth_recording(durations=[10, 10, 10])
     rec2 = rec2.save(folder=cache_folder / "multi")
 
 
diff --git a/src/spikeinterface/sorters/tests/test_launcher.py b/src/spikeinterface/sorters/tests/test_launcher.py
index f96d4db635..b67385fc7c 100644
--- a/src/spikeinterface/sorters/tests/test_launcher.py
+++ b/src/spikeinterface/sorters/tests/test_launcher.py
@@ -8,7 +8,7 @@
 
 from spikeinterface.core import load_extractor
 
-from spikeinterface import generate_ground_truth_recording, generate_recording
+from spikeinterface import generate_ground_truth_recording
 from spikeinterface.sorters import run_sorter_jobs, run_sorter_by_property
 
 
@@ -27,7 +27,7 @@
 def setup_module():
     base_seed = 42
     for i in range(num_recordings):
-        rec = generate_recording(num_channels=8, durations=[10.0], seed=base_seed + i)
+        rec, _ = generate_ground_truth_recording(num_channels=8, durations=[10.0], seed=base_seed + i)
         rec_folder = cache_folder / f"toy_rec_{i}"
         if rec_folder.is_dir():
             shutil.rmtree(rec_folder)

From 5f9670f13e744082ea34171c59ca77de067ef440 Mon Sep 17 00:00:00 2001
From: JoeZiminski <joseph.j.ziminski@gmail.com>
Date: Thu, 30 May 2024 11:21:56 +0100
Subject: [PATCH 5/5] Remove unused generate_recording import.

---
 src/spikeinterface/sorters/tests/test_container_tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/spikeinterface/sorters/tests/test_container_tools.py b/src/spikeinterface/sorters/tests/test_container_tools.py
index 030585a431..16d1e0a4a4 100644
--- a/src/spikeinterface/sorters/tests/test_container_tools.py
+++ b/src/spikeinterface/sorters/tests/test_container_tools.py
@@ -5,7 +5,7 @@
 import os
 
 import spikeinterface as si
-from spikeinterface import generate_ground_truth_recording, generate_recording
+from spikeinterface import generate_ground_truth_recording
 
 from spikeinterface.sorters.container_tools import find_recording_folders, ContainerClient, install_package_in_container