Function test_fit_methods

mne/preprocessing/tests/test_ica.py:1210–1245

Test fit_params for ICA.

(method, tmp_path)

@pytest.mark.parametrize("method", ["fastica", "picard", "infomax"])
def test_fit_methods(method, tmp_path):
    """Test fit_params for ICA."""
    _skip_check_picard(method)
    fit_params = {}
    # test no side effects
    ICA(fit_params=fit_params, method=method)
    assert fit_params == {}

    # Test I/O roundtrip.
    # Only picard and infomax support the "extended" keyword, so limit the
    # tests to those.
    if method in ["picard", "infomax"]:
        output_fname = tmp_path / "test_ica-ica.fif"

        raw = read_raw_fif(raw_fname).crop(0.5, stop).load_data()
        n_components = 3
        max_iter = 1
        fit_params = dict(extended=True)
        ica = ICA(
            fit_params=fit_params,
            n_components=n_components,
            max_iter=max_iter,
            method=method,
        )
        fit_params_after_instantiation = ica.fit_params

        if method == "infomax":
            ica.fit(raw)
        else:
            with pytest.warns(UserWarning, match="did not converge"):
                ica.fit(raw)

        ica.save(output_fname)
        ica = read_ica(output_fname)

        assert ica.fit_params == fit_params_after_instantiation


@pytest.mark.parametrize(