ups

mne-tools · Sep 12, 2019 · 0252dd3 · 0252dd3
1 parent bf51e5a
commit 0252dd3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/visualization/plot_montage.py b/examples/visualization/plot_montage.py
@@ -25,8 +25,7 @@
 # check all montages
 #
 
-# for current_montage in get_builtin_montages():
-for current_montage in ['easycap-M1', 'easycap-M10']:
+for current_montage in get_builtin_montages():
 
     montage = mne.channels.make_standard_montage(current_montage)
 

diff --git a/mne/channels/tests/test_standard_montage.py b/mne/channels/tests/test_standard_montage.py
@@ -122,7 +122,7 @@ def test_standard_montages_in_head(kind, tol):
 from mne.transforms import _sph_to_cart
 
 
-def _plot_dig_transformation(transformed, original):
+def _plot_dig_transformation(transformed, original, title=''):
     EXPECTED_HEAD_SIZE = 0.085
     from mne.viz.backends.renderer import _Renderer
     def get_data(montage):
@@ -190,9 +190,10 @@ def _plot_fid_coord(renderer, data, color):
     ren.sphere(center=trans_data.eeg, color=(.0, .0, 1.0), scale=0.0022)
 
 
+    ren.text2d(x=0, y=0, text=title, width=.1)
     ren.show()
 
-    # import pdb; pdb.set_trace()
+    return
 
 
 @pytest.mark.skip(reason='this is my plotting tinkering')