[med-svn] [python-mne] 57/353: ENH: better tests for apply_inverse_raw

[Yaroslav Halchenko]

This is an automated email from the git hooks/post-receive script.

yoh pushed a commit to tag 0.4
in repository python-mne.

commit 92f36a1457d7f4029c54ba67e1e3bf05031f2436
Author: Martin Luessi <mluessi at nmr.mgh.harvard.edu>
Date:   Fri Jan 20 16:21:13 2012 -0500

    ENH: better tests for apply_inverse_raw
---
 mne/minimum_norm/tests/test_inverse.py | 47 ++++++++++++++++++++++++++++++----
 1 file changed, 42 insertions(+), 5 deletions(-)

diff --git a/mne/minimum_norm/tests/test_inverse.py b/mne/minimum_norm/tests/test_inverse.py
index 5c5e939..5e940a9 100644
--- a/mne/minimum_norm/tests/test_inverse.py
+++ b/mne/minimum_norm/tests/test_inverse.py
@@ -1,8 +1,10 @@
 import os.path as op
+from copy import deepcopy
 import numpy as np
 from numpy.testing import assert_array_almost_equal, assert_equal
 from nose.tools import assert_true
 
+from ...fiff.constants import FIFF
 from ...datasets import sample
 from ...label import read_label, label_sign_flip
 from ...event import read_events
@@ -95,16 +97,51 @@ def test_apply_mne_inverse_raw():
     start = 3
     stop = 10
     _, times = raw[0, start:stop]
-    stc = apply_inverse_raw(raw, inverse_operator, lambda2, dSPM=True,
+    for pick_normal in [False, True]:
+        stc = apply_inverse_raw(raw, inverse_operator, lambda2, dSPM=True,
+                                label=label, start=start, stop=stop, nave=1,
+                                pick_normal=pick_normal, buffer_size=None)
+
+        stc2 = apply_inverse_raw(raw, inverse_operator, lambda2, dSPM=True,
+                                 label=label, start=start, stop=stop, nave=1,
+                                 pick_normal=pick_normal, buffer_size=3)
+
+        if not pick_normal:
+            assert_true(np.all(stc.data > 0))
+            assert_true(np.all(stc2.data > 0))
+
+        assert_array_almost_equal(stc.times, times)
+        assert_array_almost_equal(stc2.times, times)
+
+        assert_array_almost_equal(stc.data, stc2.data)
+
+
+def test_apply_mne_inverse_raw_fixed():
+    """Test MNE with precomputed fixed-orientation inverse operator on Raw"""
+    start = 3
+    stop = 10
+    _, times = raw[0, start:stop]
+
+    # create a fixed-orientation inverse operator
+    inv_op = deepcopy(inverse_operator)
+    inv_op['source_ori'] = FIFF.FIFFV_MNE_FIXED_ORI
+    inv_op['eigen_leads']['data'] = inv_op['eigen_leads']['data'][2::3, :]
+    inv_op['eigen_leads']['nrow'] = inv_op['eigen_leads']['nrow'] / 3
+    inv_op['source_cov']['data'] = inv_op['source_cov']['data'][2::3]
+    inv_op['source_cov']['dim'] = inv_op['source_cov']['dim'] / 3
+
+    inv_op['noisenorm'] = inv_op['noisenorm'][2::3]
+    inv_op['reginv'] = inv_op['reginv'][2::3]
+
+    stc = apply_inverse_raw(raw, inv_op, lambda2, dSPM=True,
                             label=label, start=start, stop=stop, nave=1,
                             pick_normal=False, buffer_size=None)
-    assert_true(np.all(stc.data > 0))
-    assert_array_almost_equal(stc.times, times)
 
-    stc2 = apply_inverse_raw(raw, inverse_operator, lambda2, dSPM=True,
+    stc2 = apply_inverse_raw(raw, inv_op, lambda2, dSPM=True,
                              label=label, start=start, stop=stop, nave=1,
                              pick_normal=False, buffer_size=3)
-    assert_true(np.all(stc2.data > 0))
+
+    assert_array_almost_equal(stc.times, times)
     assert_array_almost_equal(stc2.times, times)
 
     assert_array_almost_equal(stc.data, stc2.data)

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-med/python-mne.git