Function test_warnings

mne/preprocessing/tests/test_ica.py:187–215

Test that ICA warns on certain input data conditions.

()

def test_warnings():
    """Test that ICA warns on certain input data conditions."""
    raw = read_raw_fif(raw_fname).crop(0, 5).load_data()
    events = read_events(event_name)
    epochs = Epochs(raw, events=events, baseline=None, preload=True)
    ica = ICA(n_components=2, max_iter=1, method="infomax", random_state=0)

    # not high-passed
    with epochs.info._unlock():
        epochs.info["highpass"] = 0.0
    with pytest.warns(RuntimeWarning, match="should be high-pass filtered"):
        ica.fit(epochs)

    # baselined
    with epochs.info._unlock():
        epochs.info["highpass"] = 1.0
    epochs.baseline = (epochs.tmin, 0)
    with pytest.warns(RuntimeWarning, match="epochs.*were baseline-corrected"):
        ica.fit(epochs)

    # cleaning baseline-corrected data
    with epochs.info._unlock():
        epochs.info["highpass"] = 1.0
    epochs.baseline = None
    ica.fit(epochs)

    epochs.baseline = (epochs.tmin, 0)
    with pytest.warns(RuntimeWarning, match="consider baseline-correcting.*again"):
        ica.apply(epochs)


@pytest.mark.parametrize("n_components", (None, 0.9999, 8, 9, 10))