port plot_probe_map() to new widgets API

SpikeInterface · Sep 19, 2023 · 3d79295 · 3d79295
1 parent e49071e
commit 3d79295
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 79 deletions.
diff --git a/src/spikeinterface/widgets/_legacy_mpl_widgets/__init__.py b/src/spikeinterface/widgets/_legacy_mpl_widgets/__init__.py
@@ -1,7 +1,6 @@
 # basics
 # from .timeseries import plot_timeseries, TracesWidget
 from .rasters import plot_rasters, RasterWidget
-from .probemap import plot_probe_map, ProbeMapWidget
 
 # isi/ccg/acg
 from .isidistribution import plot_isi_distribution, ISIDistributionWidget

diff --git a/src/spikeinterface/widgets/_legacy_mpl_widgets/probemap.py b/src/spikeinterface/widgets/_legacy_mpl_widgets/probemap.py
diff --git a/src/spikeinterface/widgets/tests/test_widgets.py b/src/spikeinterface/widgets/tests/test_widgets.py
@@ -336,6 +336,11 @@ def test_plot_confusion_matrix(self):
             if backend not in self.skip_backends:
                 sw.plot_confusion_matrix(self.gt_comp)
 
+    def test_plot_probe_map(self):
+        possible_backends = list(sw.ProbeMapWidget.get_possible_backends())
+        for backend in possible_backends:
+            if backend not in self.skip_backends:
+                sw.plot_probe_map(self.recording, with_channel_ids=True, with_contact_id=True)
 
 
 if __name__ == "__main__":
@@ -359,7 +364,8 @@ def test_plot_confusion_matrix(self):
     # mytest.test_template_metrics()
     # mytest.test_amplitudes()
     # mytest.test_plot_agreement_matrix()
-    mytest.test_plot_confusion_matrix()
+    # mytest.test_plot_confusion_matrix()
+    mytest.test_plot_probe_map()
 
     # plt.ion()
     plt.show()
diff --git a/src/spikeinterface/widgets/widget_list.py b/src/spikeinterface/widgets/widget_list.py
@@ -9,6 +9,7 @@
 from .confusion_matrix import ConfusionMatrixWidget
 from .crosscorrelograms import CrossCorrelogramsWidget
 from .motion import MotionWidget
+from .probe_map import ProbeMapWidget
 from .quality_metrics import QualityMetricsWidget
 from .sorting_summary import SortingSummaryWidget
 from .spike_locations import SpikeLocationsWidget
@@ -32,6 +33,7 @@
     ConfusionMatrixWidget,
     CrossCorrelogramsWidget,
     MotionWidget,
+    ProbeMapWidget,
     QualityMetricsWidget,
     SortingSummaryWidget,
     SpikeLocationsWidget,
@@ -87,6 +89,7 @@
 plot_confusion_matrix = ConfusionMatrixWidget
 plot_crosscorrelograms = CrossCorrelogramsWidget
 plot_motion = MotionWidget
+plot_probe_map = ProbeMapWidget
 plot_quality_metrics = QualityMetricsWidget
 plot_sorting_summary = SortingSummaryWidget
 plot_spike_locations = SpikeLocationsWidget