fix tests

src/spikeinterface/qualitymetrics/tests/conftest.py

@@ -8,8 +8,8 @@
 job_kwargs = dict(n_jobs=2, progress_bar=True, chunk_duration="1s")
 
 
-@pytest.fixture(scope="module")
-def small_sorting_analyzer():
+
+def make_small_analyzer():
     recording, sorting = generate_ground_truth_recording(
         durations=[2.0],
         num_units=10,
@@ -34,6 +34,9 @@ def small_sorting_analyzer():
 
     return sorting_analyzer
 
+@pytest.fixture(scope="module")
+def small_sorting_analyzer():
+    return make_small_analyzer()
 
 @pytest.fixture(scope="module")
 def sorting_analyzer_simple():

src/spikeinterface/qualitymetrics/tests/test_pca_metrics.py

@@ -19,7 +19,14 @@ def test_calculate_pc_metrics(small_sorting_analyzer):
             assert not np.all(np.isnan(res1[metric_name].values))
             assert not np.all(np.isnan(res2[metric_name].values))
 
-        assert np.array_equal(res1[metric_name].values, res2[metric_name].values)
+        # import matplotlib.pyplot as plt
+        # fig, ax = plt.subplots()
+        # ax.plot(res1[metric_name].values)
+        # ax.plot(res2[metric_name].values)
+        # ax.plot(res2[metric_name].values - res1[metric_name].values)
+        # plt.show()
+
+        np.testing.assert_almost_equal(res1[metric_name].values, res2[metric_name].values, decimal=4)
 
 
 def test_pca_metrics_multi_processing(small_sorting_analyzer):
@@ -41,3 +48,8 @@ def test_pca_metrics_multi_processing(small_sorting_analyzer):
     res2 = compute_pc_metrics(
         sorting_analyzer, n_jobs=-1, metric_names=metric_names, max_threads_per_worker=2, progress_bar=True
     )
+
+if __name__ == "__main__":
+    from spikeinterface.qualitymetrics.tests.conftest import make_small_analyzer
+    small_sorting_analyzer = make_small_analyzer()
+    test_calculate_pc_metrics(small_sorting_analyzer)