Put main stuff back

Author: chrishalcrow

src/spikeinterface/qualitymetrics/tests/test_metrics_functions.py

@@ -536,3 +536,32 @@ def test_calculate_sd_ratio(sorting_analyzer_simple):
     assert np.all(list(sd_ratio.keys()) == sorting_analyzer_simple.unit_ids)
     # @aurelien can you check this, this is not working anymore
     # assert np.allclose(list(sd_ratio.values()), 1, atol=0.25, rtol=0)
+
+
+if __name__ == "__main__":
+
+    sorting_analyzer = _sorting_analyzer_simple()
+    print(sorting_analyzer)
+
+    test_unit_structure_in_output(_small_sorting_analyzer())
+
+    # test_calculate_firing_rate_num_spikes(sorting_analyzer)
+
+    test_calculate_snrs(sorting_analyzer)
+    # test_calculate_amplitude_cutoff(sorting_analyzer)
+    # test_calculate_presence_ratio(sorting_analyzer)
+    # test_calculate_amplitude_median(sorting_analyzer)
+    # test_calculate_sliding_rp_violations(sorting_analyzer)
+    # test_calculate_drift_metrics(sorting_analyzer)
+    # test_synchrony_metrics(sorting_analyzer)
+    # test_synchrony_metrics_unit_id_subset(sorting_analyzer)
+    # test_synchrony_metrics_no_unit_ids(sorting_analyzer)
+    # test_calculate_firing_range(sorting_analyzer)
+    # test_calculate_amplitude_cv_metrics(sorting_analyzer)
+    # test_calculate_sd_ratio(sorting_analyzer)
+
+    # sorting_analyzer_violations = _sorting_analyzer_violations()
+    # print(sorting_analyzer_violations)
+    # test_calculate_isi_violations(sorting_analyzer_violations)
+    # test_calculate_sliding_rp_violations(sorting_analyzer_violations)
+    # test_calculate_rp_violations(sorting_analyzer_violations)

src/spikeinterface/qualitymetrics/tests/test_quality_metric_calculator.py

@@ -19,7 +19,6 @@
 
 def test_compute_quality_metrics(sorting_analyzer_simple):
     sorting_analyzer = sorting_analyzer_simple
-    print(sorting_analyzer)
 
     # without PCs
     metrics = compute_quality_metrics(
@@ -245,3 +244,12 @@ def test_empty_units(sorting_analyzer_simple):
 #         for metric_name in metrics.columns:
 #             # NaNs are skipped
 #             assert np.allclose(metrics[metric_name].dropna(), metrics_par[metric_name].dropna())
+
+if __name__ == "__main__":
+
+    sorting_analyzer = get_sorting_analyzer()
+    print(sorting_analyzer)
+
+    test_compute_quality_metrics(sorting_analyzer)
+    test_compute_quality_metrics_recordingless(sorting_analyzer)
+    test_empty_units(sorting_analyzer)