Function test_surface_vector_source_morph

mne/tests/test_morph.py:298–362

Test surface and vector source estimate morph.

(tmp_path)

@testing.requires_testing_data
def test_surface_vector_source_morph(tmp_path):
    """Test surface and vector source estimate morph."""
    pytest.importorskip("h5io")
    inverse_operator_surf = read_inverse_operator(fname_inv_surf)

    stc_surf = read_source_estimate(fname_smorph, subject="sample")
    stc_surf.crop(0.09, 0.1)  # for faster computation

    stc_vec = _real_vec_stc()

    source_morph_surf = compute_source_morph(
        inverse_operator_surf["src"], subjects_dir=subjects_dir, smooth=1, warn=False
    )  # smooth 1 for speed
    src_to_fs = mne.read_source_spaces(
        subjects_dir / "fsaverage" / "bem" / "fsaverage-ico-5-src.fif"
    )
    morph_with_src_to = compute_source_morph(
        inverse_operator_surf["src"],
        subjects_dir=subjects_dir,
        src_to=src_to_fs,
        smooth=1,
        warn=False,
    )
    assert morph_with_src_to.spacing is None
    assert source_morph_surf.subject_from == "sample"
    assert source_morph_surf.subject_to == "fsaverage"
    assert source_morph_surf.kind == "surface"
    assert isinstance(source_morph_surf.src_data, dict)
    assert isinstance(source_morph_surf.src_data["vertices_from"], list)
    assert isinstance(source_morph_surf, SourceMorph)
    stc_surf_morphed = source_morph_surf.apply(stc_surf)
    assert isinstance(stc_surf_morphed, SourceEstimate)
    stc_vec_morphed = source_morph_surf.apply(stc_vec)
    with pytest.raises(ValueError, match="Invalid value for the 'output'"):
        source_morph_surf.apply(stc_surf, output="nifti1")

    # check if correct class after morphing
    assert isinstance(stc_surf_morphed, SourceEstimate)
    assert isinstance(stc_vec_morphed, VectorSourceEstimate)

    # check __repr__
    assert "surface" in repr(source_morph_surf)

    # check loading and saving for surf
    source_morph_surf.save(tmp_path / "42.h5")

    source_morph_surf_r = read_source_morph(tmp_path / "42.h5")

    assert all(
        [
            read == saved
            for read, saved in zip(
                sorted(source_morph_surf_r.__dict__), sorted(source_morph_surf.__dict__)
            )
        ]
    )

    # check wrong subject correction