Skip to content

Commit d030131

Browse files
committed
FIX: More
1 parent 6e28aeb commit d030131

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

mne/inverse_sparse/tests/test_mxne_inverse.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -589,8 +589,12 @@ def data_fun(times):
589589
forward, stc, info, noise_cov, nave=nave, use_cps=False, iir_filter=None
590590
)
591591
evoked = evoked.crop(tmin=0, tmax=10e-3)
592-
stc_ = mixed_norm(evoked, forward, noise_cov, loose=0.9, n_mxne_iter=5, depth=0.9)
593-
assert_array_equal(stc_.vertices, stc.vertices)
592+
stc_ = mixed_norm(
593+
evoked, forward, noise_cov, loose=0.9, n_mxne_iter=5, depth=0.9, random_state=0
594+
)
595+
assert len(stc_.vertices) == len(stc.vertices) == 2
596+
for si in range(len(stc_.vertices)):
597+
assert_array_equal(stc_.vertices[si], stc.vertices[si], err_msg=f"{si=}")
594598

595599

596600
@pytest.mark.slowtest # slow on Azure
@@ -609,7 +613,13 @@ def test_mxne_inverse_empty():
609613
cov = read_cov(fname_cov)
610614
with pytest.warns(RuntimeWarning, match="too big"):
611615
stc, residual = mixed_norm(
612-
evoked, forward, cov, n_mxne_iter=3, alpha=99, return_residual=True
616+
evoked,
617+
forward,
618+
cov,
619+
n_mxne_iter=3,
620+
alpha=99,
621+
return_residual=True,
622+
random_state=0,
613623
)
614624
assert stc.data.size == 0
615625
assert stc.vertices[0].size == 0

0 commit comments

Comments
 (0)