def test_pconnscalar(): parcel_map = create_parcel_map((0, 1)) scalar_map = create_scalar_map((2, )) matrix = ci.Cifti2Matrix() matrix.append(parcel_map) matrix.append(scalar_map) hdr = ci.Cifti2Header(matrix) data = np.random.randn(4, 4, 2) img = ci.Cifti2Image(data, hdr) img.nifti_header.set_intent('NIFTI_INTENT_CONNECTIVITY_PARCELLATED_' 'PARCELLATED_SCALAR') with InTemporaryDirectory(): ci.save(img, 'test.pconnscalar.nii') img2 = ci.load('test.pconnscalar.nii') assert_equal(img.nifti_header.get_intent()[0], 'ConnPPSc') assert_true(isinstance(img2, ci.Cifti2Image)) assert_array_equal(img2.get_fdata(), data) assert_equal(img2.header.matrix.get_index_map(0), img2.header.matrix.get_index_map(1)) check_parcel_map(img2.header.matrix.get_index_map(0)) check_scalar_map(img2.header.matrix.get_index_map(2)) del img2
def test_plabel(): label_map = create_label_map((0, )) parcel_map = create_parcel_map((1, )) matrix = ci.Cifti2Matrix() matrix.append(label_map) matrix.append(parcel_map) hdr = ci.Cifti2Header(matrix) data = np.random.randn(2, 4) img = ci.Cifti2Image(data, hdr) with InTemporaryDirectory(): ci.save(img, 'test.plabel.nii') img2 = ci.load('test.plabel.nii') assert_equal(img.nifti_header.get_intent()[0], 'ConnUnknown') assert_true(isinstance(img2, ci.Cifti2Image)) assert_array_equal(img2.get_fdata(), data) check_label_map(img2.header.matrix.get_index_map(0)) check_parcel_map(img2.header.matrix.get_index_map(1)) del img2
def test_dconn(): mapping = create_geometry_map((0, 1)) matrix = ci.Cifti2Matrix() matrix.append(mapping) hdr = ci.Cifti2Header(matrix) data = np.random.randn(10, 10) img = ci.Cifti2Image(data, hdr) img.nifti_header.set_intent('NIFTI_INTENT_CONNECTIVITY_DENSE') with InTemporaryDirectory(): ci.save(img, 'test.dconn.nii') img2 = nib.load('test.dconn.nii') assert_equal(img2.nifti_header.get_intent()[0], 'ConnDense') assert_true(isinstance(img2, ci.Cifti2Image)) assert_array_equal(img2.get_fdata(), data) assert_equal(img2.header.matrix.get_index_map(0), img2.header.matrix.get_index_map(1)) check_geometry_map(img2.header.matrix.get_index_map(0)) del img2
def test_dpconn(): parcel_map = create_parcel_map((0, )) geometry_map = create_geometry_map((1, )) matrix = ci.Cifti2Matrix() matrix.append(parcel_map) matrix.append(geometry_map) hdr = ci.Cifti2Header(matrix) data = np.random.randn(4, 10) img = ci.Cifti2Image(data, hdr) img.nifti_header.set_intent('NIFTI_INTENT_CONNECTIVITY_DENSE_PARCELLATED') with InTemporaryDirectory(): ci.save(img, 'test.dpconn.nii') img2 = ci.load('test.dpconn.nii') assert_equal(img2.nifti_header.get_intent()[0], 'ConnDenseParcel') assert_true(isinstance(img2, ci.Cifti2Image)) assert_array_equal(img2.get_fdata(), data) check_parcel_map(img2.header.matrix.get_index_map(0)) check_geometry_map(img2.header.matrix.get_index_map(1)) del img2
def test_ptseries(): series_map = create_series_map((0, )) parcel_map = create_parcel_map((1, )) matrix = ci.Cifti2Matrix() matrix.append(series_map) matrix.append(parcel_map) hdr = ci.Cifti2Header(matrix) data = np.random.randn(13, 4) img = ci.Cifti2Image(data, hdr) img.nifti_header.set_intent('NIFTI_INTENT_CONNECTIVITY_PARCELLATED_SERIES') with InTemporaryDirectory(): ci.save(img, 'test.ptseries.nii') img2 = nib.load('test.ptseries.nii') assert_equal(img2.nifti_header.get_intent()[0], 'ConnParcelSries') assert_true(isinstance(img2, ci.Cifti2Image)) assert_array_equal(img2.get_fdata(), data) check_series_map(img2.header.matrix.get_index_map(0)) check_parcel_map(img2.header.matrix.get_index_map(1)) del img2
def test_dscalar(): scalar_map = create_scalar_map((0, )) geometry_map = create_geometry_map((1, )) matrix = ci.Cifti2Matrix() matrix.append(scalar_map) matrix.append(geometry_map) hdr = ci.Cifti2Header(matrix) data = np.random.randn(2, 10) img = ci.Cifti2Image(data, hdr) img.nifti_header.set_intent('NIFTI_INTENT_CONNECTIVITY_DENSE_SCALARS') with InTemporaryDirectory(): ci.save(img, 'test.dscalar.nii') img2 = nib.load('test.dscalar.nii') assert_equal(img2.nifti_header.get_intent()[0], 'ConnDenseScalar') assert_true(isinstance(img2, ci.Cifti2Image)) assert_array_equal(img2.get_fdata(), data) check_scalar_map(img2.header.matrix.get_index_map(0)) check_geometry_map(img2.header.matrix.get_index_map(1)) del img2