From 6322499b88e414a97cc6d6642de4529594d22ec5 Mon Sep 17 00:00:00 2001 From: Daniel McCloy Date: Mon, 12 Aug 2024 09:16:35 -0500 Subject: [PATCH] use MRO in test too --- mne/stats/tests/test_cluster_level.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mne/stats/tests/test_cluster_level.py b/mne/stats/tests/test_cluster_level.py index ed0d830bdfd..fc2af127a96 100644 --- a/mne/stats/tests/test_cluster_level.py +++ b/mne/stats/tests/test_cluster_level.py @@ -37,8 +37,8 @@ summarize_clusters_stc, ttest_1samp_no_p, ) -from mne.time_frequency import AverageTFRArray, EpochsTFRArray -from mne.utils import _record_warnings, catch_logging +from mne.time_frequency import AverageTFRArray, BaseTFR, EpochsTFRArray +from mne.utils import GetEpochsMixin, _record_warnings, catch_logging n_space = 50 @@ -910,8 +910,8 @@ def test_new_cluster_api(Inst): pd = pytest.importorskip("pandas") rng = np.random.default_rng(seed=8675309) - is_epo = Inst in (EpochsTFRArray, EpochsArray) - is_tfr = Inst in (EpochsTFRArray, AverageTFRArray) + is_epo = GetEpochsMixin in Inst.__mro__ + is_tfr = BaseTFR in Inst.__mro__ n_epo, n_chan, n_freq, n_times = 6, 3, 4, 5