Exemplo n.º 1
0
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)
Exemplo n.º 2
0
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
Exemplo n.º 3
0
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
Exemplo n.º 4
0
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()