Function test_ica_additional

mne/preprocessing/tests/test_ica.py:648–1035

Test additional ICA functionality.

(method, tmp_path, short_raw_epochs)

@pytest.mark.slowtest
@pytest.mark.parametrize("method", ["picard", "fastica"])
def test_ica_additional(method, tmp_path, short_raw_epochs):
    """Test additional ICA functionality."""
    _skip_check_picard(method)
    raw, epochs, epochs_eog = short_raw_epochs
    few_picks = np.arange(5)

    # test if n_components=None works
    ica = ICA(n_components=None, method=method, max_iter=1)
    with _baseline_corrected, pytest.warns(UserWarning, match="did not converge"):
        ica.fit(epochs)
    _assert_ica_attributes(ica, epochs.get_data("data"), limits=(0.05, 20))

    test_cov = read_cov(test_cov_name)
    ica = ICA(noise_cov=test_cov, n_components=3, method=method)
    assert ica.info is None
    with pytest.warns(RuntimeWarning, match="normalize_proj"):
        ica.fit(raw, picks=few_picks)
    _assert_ica_attributes(ica, raw.get_data(np.arange(5)), limits=(1, 90))
    assert isinstance(ica.info, Info)
    assert ica.n_components_ < 5

    ica = ICA(n_components=3, method=method, max_iter=1)
    with pytest.raises(RuntimeError, match="No fit"):
        ica.save("")

    with pytest.warns(Warning, match="converge"):
        ica.fit(raw, np.arange(1, 6))
    _assert_ica_attributes(ica, raw.get_data(np.arange(1, 6)))

    # check Kuiper index threshold
    assert_allclose(ica._get_ctps_threshold(), 0.5)
    with pytest.raises(TypeError, match="str or numeric"):
        ica.find_bads_ecg(raw, threshold=None)
    with pytest.warns(RuntimeWarning, match="is longer than the signal"):
        ica.find_bads_ecg(raw, threshold=0.25)
    # check invalid measure argument
    with pytest.raises(ValueError, match="Invalid value"):
        ica.find_bads_ecg(
            raw, method="correlation", measure="unknown", threshold="auto"
        )
    # check passing a ch_name to find_bads_ecg
    with pytest.warns(RuntimeWarning, match="longer"):
        _, scores_1 = ica.find_bads_ecg(raw, threshold="auto")
    with pytest.warns(RuntimeWarning, match="longer"):
        _, scores_2 = ica.find_bads_ecg(raw, raw.ch_names[1], threshold="auto")
    assert scores_1[0] != scores_2[0]

    # test corrmap
    ica2 = ica.copy()
    ica3 = ica.copy()
    corrmap(
        [ica, ica2], (0, 0), threshold="auto", label="blinks", plot=True, ch_type="mag"
    )
    with pytest.raises(RuntimeError, match="No component detected"):
        corrmap(
            [ica, ica2],
            (0, 0),
            threshold=2,