MRG: Convert CTF STL (mne-tools#5462)

* ENH: Convert CTF STL * DOC: Better comment [ci skip] * DOC: whats_new.rst [ci skip] * FIX: Fix count
kpillay · Oct 11, 2018 · 8cc1068 · 8cc1068
1 parent 3496365
commit 8cc1068
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/doc/whats_new.rst b/doc/whats_new.rst
@@ -55,6 +55,8 @@ Changelog
 
 - Add :func:`mne.head_to_mri` to convert positions from head coordinates to MRI RAS coordinates, by `Joan Massich`_ and `Alex Gramfort`_
 
+- Add improved CTF helmet for :func:`mne.viz.plot_alignment` by `Eric Larson`_
+
 - :func:`mne.combine_evoked` and :func:`mne.grand_average` can now handle input with the same channels in different orders, if required, by `Jona Sassenhagen`_
 
 - Add `split_naming` parameter to the `Raw.save` method to allow for BIDS-compatible raw file name construction by `Teon Brooks`_

diff --git a/examples/visualization/plot_meg_sensors.py b/examples/visualization/plot_meg_sensors.py
@@ -3,7 +3,7 @@
 Plotting sensor layouts of MEG systems
 ======================================
 
-In this example, sensor layouts of different MEG systems are shown.
+Sensor layouts of different MEG systems are shown.
 """
 # Author: Eric Larson <[email protected]>
 #

diff --git a/mne/data/helmets/CTF_275.fif.gz b/mne/data/helmets/CTF_275.fif.gz
diff --git a/mne/tests/test_surface.py b/mne/tests/test_surface.py
@@ -39,9 +39,12 @@ def test_helmet():
     fname_trans = op.join(base_dir, 'tests', 'data',
                           'sample-audvis-raw-trans.txt')
     trans = _get_trans(fname_trans)[0]
-    for fname in [fname_raw, fname_kit_raw, fname_bti_raw, fname_ctf_raw]:
+    for fname, n in [(fname_raw, 304),
+                     (fname_kit_raw, 304),
+                     (fname_bti_raw, 304),
+                     (fname_ctf_raw, 348)]:
         helmet = get_meg_helmet_surf(read_info(fname), trans)
-        assert_equal(len(helmet['rr']), 304)  # they all have 304 verts
+        assert_equal(len(helmet['rr']), n)
         assert_equal(len(helmet['rr']), len(helmet['nn']))