Function test_morph_stc_dense

mne/tests/test_morph.py:729–827

Test morphing stc.

()

@pytest.mark.slowtest
@testing.requires_testing_data
def test_morph_stc_dense():
    """Test morphing stc."""
    subject_from = "sample"
    subject_to = "fsaverage"
    stc_from = read_source_estimate(fname_smorph, subject="sample")
    stc_to = read_source_estimate(fname_fmorph)
    # make sure we can specify grade
    stc_from.crop(0.09, 0.1)  # for faster computation
    stc_to.crop(0.09, 0.1)  # for faster computation
    assert_array_equal(
        stc_to.time_as_index([0.09, 0.1], use_rounding=True), [0, len(stc_to.times) - 1]
    )

    # After dep change this to:
    morph = compute_source_morph(
        subject_to=subject_to,
        spacing=3,
        smooth=12,
        src=stc_from,
        subjects_dir=subjects_dir,
        precompute=True,
    )
    assert morph.vol_morph_mat is None  # a no-op for surface
    stc_to1 = morph.apply(stc_from)
    assert_allclose(stc_to.data, stc_to1.data, atol=1e-5)

    mean_from = stc_from.data.mean(axis=0)
    mean_to = stc_to1.data.mean(axis=0)
    assert np.corrcoef(mean_to, mean_from).min() > 0.999

    vertices_to = grade_to_vertices(subject_to, grade=3, subjects_dir=subjects_dir)

    # make sure we can fill by morphing
    with pytest.warns(RuntimeWarning, match="consider increasing"):
        morph = compute_source_morph(
            stc_from,
            subject_from,
            subject_to,
            spacing=None,
            smooth=1,
            subjects_dir=subjects_dir,
        )
    stc_to5 = morph.apply(stc_from)
    assert stc_to5.data.shape[0] == 163842 + 163842

    # Morph vector data
    stc_vec = _real_vec_stc()
    stc_vec_to1 = compute_source_morph(
        stc_vec,
        subject_from,
        subject_to,
        subjects_dir=subjects_dir,
        spacing=vertices_to,
        smooth=1,
        warn=False,
    ).apply(stc_vec)
    assert stc_vec_to1.subject == subject_to
    assert stc_vec_to1.tmin == stc_vec.tmin