Skip to content

Commit 0c0e72e

Browse files
authored
MAINT: Make MxNE test more robust (#13315)
1 parent 35398fd commit 0c0e72e

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

mne/inverse_sparse/tests/test_mxne_inverse.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,7 @@ def test_mxne_inverse_sure_synthetic(
549549

550550
@pytest.mark.slowtest # slow on Azure
551551
@testing.requires_testing_data
552-
def test_mxne_inverse_sure():
552+
def test_mxne_inverse_sure_meg():
553553
"""Tests SURE criterion for automatic alpha selection on MEG data."""
554554

555555
def data_fun(times):
@@ -558,10 +558,10 @@ def data_fun(times):
558558
return data
559559

560560
n_dipoles = 2
561-
raw = mne.io.read_raw_fif(fname_raw)
562-
info = mne.io.read_info(fname_data)
563-
with info._unlock():
564-
info["projs"] = []
561+
raw = mne.io.read_raw_fif(fname_raw).pick_types("grad", exclude="bads")
562+
raw.del_proj()
563+
info = raw.info
564+
del raw
565565
noise_cov = mne.make_ad_hoc_cov(info)
566566
label_names = ["Aud-lh", "Aud-rh"]
567567
labels = [
@@ -572,10 +572,8 @@ def data_fun(times):
572572
data_path / "MEG" / "sample" / "sample_audvis_trunc-meg-eeg-oct-4-fwd.fif"
573573
)
574574
forward = mne.read_forward_solution(fname_fwd)
575-
forward = mne.pick_types_forward(
576-
forward, meg="grad", eeg=False, exclude=raw.info["bads"]
577-
)
578-
times = np.arange(100, dtype=np.float64) / raw.info["sfreq"] - 0.1
575+
forward = mne.pick_channels_forward(forward, info["ch_names"])
576+
times = np.arange(100, dtype=np.float64) / info["sfreq"] - 0.1
579577
stc = simulate_sparse_stc(
580578
forward["src"],
581579
n_dipoles=n_dipoles,
@@ -584,13 +582,16 @@ def data_fun(times):
584582
labels=labels,
585583
data_fun=data_fun,
586584
)
585+
assert len(stc.vertices) == 2
586+
assert_array_equal(stc.vertices[0], [89259])
587+
assert_array_equal(stc.vertices[1], [70279])
587588
nave = 30
588589
evoked = simulate_evoked(
589590
forward, stc, info, noise_cov, nave=nave, use_cps=False, iir_filter=None
590591
)
591592
evoked = evoked.crop(tmin=0, tmax=10e-3)
592593
stc_ = mixed_norm(
593-
evoked, forward, noise_cov, loose=0.9, n_mxne_iter=5, depth=0.9, random_state=0
594+
evoked, forward, noise_cov, loose=0.9, n_mxne_iter=5, depth=0.9, random_state=1
594595
)
595596
assert len(stc_.vertices) == len(stc.vertices) == 2
596597
for si in range(len(stc_.vertices)):

0 commit comments

Comments
 (0)