def test_save_nifti_parcels_map(): """ Test save_nifti_parcels_map functionality """ base_dir = str(Path(__file__).parent / "examples") ID = '002' dir_path = base_dir + '/002/fmri' roi = None network = None array_data = np.arange(24, dtype=np.int16).reshape((2, 3, 4)) affine = np.diag([1, 2, 3, 1]) net_parcels_map_nifti = nib.Nifti1Image(array_data, affine) # Delete pre-existing parcellations dir parc_dir = dir_path + '/parcellations' if os.path.exists(parc_dir): shutil.move(parc_dir, parc_dir + '_tmp') net_parcels_nii_path = utils.save_nifti_parcels_map( ID, dir_path, roi, network, net_parcels_map_nifti) assert os.path.isfile(net_parcels_nii_path) is True # Restore pre-existsing parc dir if os.path.exists(parc_dir + '_tmp'): shutil.rmtree(parc_dir) shutil.move(parc_dir + '_tmp', parc_dir)
def test_save_nifti_parcels_map(): """ Test save_nifti_parcels_map functionality """ base_dir = str(Path(__file__).parent/"examples") ID = '002' dir_path = base_dir + '/002/fmri' roi = None network = None array_data = np.arange(24, dtype=np.int16).reshape((2, 3, 4)) affine = np.diag([1, 2, 3, 1]) net_parcels_map_nifti = nib.Nifti1Image(array_data, affine) net_parcels_nii_path = utils.save_nifti_parcels_map(ID, dir_path, roi, network, net_parcels_map_nifti) assert os.path.isfile(net_parcels_nii_path) is True
def test_save_nifti_parcels_map(): """ Test save_nifti_parcels_map functionality """ import tempfile dir_path = str(tempfile.TemporaryDirectory().name) os.makedirs(dir_path) ID = '002' network = None array_data = np.arange(24, dtype=np.int16).reshape((2, 3, 4)) affine = np.diag([1, 2, 3, 1]) net_parcels_map_nifti = nib.Nifti1Image(array_data, affine) net_parcels_nii_path = utils.save_nifti_parcels_map( ID, dir_path, network, net_parcels_map_nifti) assert os.path.isfile(net_parcels_nii_path) is True
def test_save_nifti_parcels_map(): """ Test save_nifti_parcels_map functionality """ tmp = tempfile.TemporaryDirectory() dir_path = str(tmp.name) os.makedirs(dir_path, exist_ok=True) ID = '002' vox_size = '2mm' subnet = None array_data = np.arange(24, dtype=np.int16).reshape((2, 3, 4)) affine = np.diag([1, 2, 3, 1]) net_parcels_map_nifti = nib.Nifti1Image(array_data, affine) net_parcels_nii_path = utils.save_nifti_parcels_map( ID, dir_path, subnet, net_parcels_map_nifti, vox_size) assert os.path.isfile(net_parcels_nii_path) is True tmp.cleanup()