Extend check on clus

SpikeInterface · Sep 4, 2024 · 0ed4876 · 0ed4876
1 parent 464c6e3
commit 0ed4876
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/scripts/test_kilosort4_ci.py b/.github/scripts/test_kilosort4_ci.py
@@ -569,6 +569,7 @@ def monkeypatch_filter_function(self, X, ops=None, ibatch=None):
         results["si"]["st"] = np.load(spikeinterface_output_dir / "sorter_output" / "spike_times.npy")
         results["si"]["clus"] = np.load(spikeinterface_output_dir / "sorter_output" / "spike_clusters.npy")
         assert np.allclose(results["ks"]["st"], results["si"]["st"], rtol=0, atol=1)
+        assert np.array_equal(results["ks"]["clus"], results["si"]["clus"])
 
     ##### Helpers ######
     def _get_kilosort_native_settings(self, recording, paths, param_key, param_value):