Function test_random_parcellation

mne/tests/test_label.py:1024–1060

Test generation of random cortical parcellation.

()

@testing.requires_testing_data
def test_random_parcellation():
    """Test generation of random cortical parcellation."""
    pytest.importorskip("nibabel")
    hemi = "both"
    n_parcel = 50
    surface = "sphere.reg"
    subject = "sample_ds"
    rng = np.random.RandomState(0)

    # Parcellation
    labels = random_parcellation(
        subject, n_parcel, hemi, subjects_dir, surface=surface, random_state=rng
    )

    # test number of labels
    assert_equal(len(labels), n_parcel)
    if hemi == "both":
        hemi = ["lh", "rh"]
    hemis = np.atleast_1d(hemi)
    for hemi in set(hemis):
        vertices_total = []
        for label in labels:
            if label.hemi == hemi:
                # test that labels are not empty
                assert len(label.vertices) > 0

                # vertices of hemi covered by labels
                vertices_total = np.append(vertices_total, label.vertices)

        # test that labels don't intersect
        assert_equal(len(np.unique(vertices_total)), len(vertices_total))

        surf_fname = subjects_dir / subject / "surf" / (hemi + "." + surface)
        vert, _ = read_surface(surf_fname)

        # Test that labels cover whole surface
        assert_array_equal(np.sort(vertices_total), np.arange(len(vert)))


@testing.requires_testing_data