test post-decimation

drammock · Feb 14, 2024 · e317590 · e317590
1 parent 20259e3
commit e317590
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/mne/time_frequency/tests/test_tfr.py b/mne/time_frequency/tests/test_tfr.py
@@ -1544,6 +1544,11 @@ def test_tfr_decim(epochs, method, freqs, decim):
     assert tfr.shape[-1] == want
     # Check that decim changes sfreq
     assert tfr.sfreq == epochs.info["sfreq"] / (decim.step or 1)
+    # check after-the-fact decimation. The mixin .decimate method doesn't allow slices
+    if isinstance(decim, int):
+        tfr2 = epochs.compute_tfr(method, freqs=freqs, decim=1)
+        tfr2.decimate(decim)
+        assert tfr == tfr2
 
 
 def test_tfr_arithmetic(epochs):

diff --git a/mne/utils/mixin.py b/mne/utils/mixin.py
@@ -680,10 +680,12 @@ def decimate(self, decim, offset=0, *, verbose=None):
         # appropriately filtered to avoid aliasing
         from ..epochs import BaseEpochs
         from ..evoked import Evoked
-        from ..time_frequency import AverageTFR, EpochsTFR
+        from ..time_frequency import AverageTFR, EpochsTFR, RawTFR
 
         # This should be the list of classes that inherit
-        _validate_type(self, (BaseEpochs, Evoked, EpochsTFR, AverageTFR), "inst")
+        _validate_type(
+            self, (BaseEpochs, Evoked, RawTFR, EpochsTFR, AverageTFR), "inst"
+        )
         decim, offset, new_sfreq = _check_decim(
             self.info, decim, offset, check_filter=not hasattr(self, "freqs")
         )