Update unit tests for lasso select

mne-tools · Jan 7, 2025 · bea101d · bea101d
1 parent 871e15a
commit bea101d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/mne/viz/tests/test_topo.py b/mne/viz/tests/test_topo.py
@@ -295,6 +295,13 @@ def test_plot_topo_image_epochs():
     assert len(qm_cmap) >= 1
     assert qm_cmap[0] is cmap
 
+def test_plot_topo_select():
+    """Test selecting sensors in an ERP topography plot."""
+    # Show topography
+    evoked = _get_epochs().average()
+    plot_evoked_topo(evoked, select=True)
+
+
 
 def test_plot_tfr_topo():
     """Test plotting of TFR data."""

diff --git a/mne/viz/tests/test_utils.py b/mne/viz/tests/test_utils.py
@@ -27,6 +27,7 @@
     centers_to_edges,
     compare_fiff,
     concatenate_images,
+    SelectFromCollection,
 )
 
 base_dir = Path(__file__).parents[2] / "io" / "tests" / "data"