def test_fetch_atlas_msdl(tmp_path, request_mocker): labels = pd.DataFrame( {"x": [1.5, 1.2], "y": [1.5, 1.3], "z": [1.5, 1.4], "name": ["Aud", "DMN"], "net_name": ["Aud", "DMN"]}) root = Path("MSDL_rois") archive = {root / "msdl_rois_labels.csv": labels.to_csv(index=False), root / "msdl_rois.nii": "", root / "README.txt": ""} request_mocker.url_mapping["*MSDL_rois.zip"] = dict_to_archive( archive, "zip") dataset = atlas.fetch_atlas_msdl(data_dir=tmp_path, verbose=0) assert isinstance(dataset.labels, list) assert isinstance(dataset.region_coords, list) assert isinstance(dataset.networks, list) assert isinstance(dataset.maps, str) assert request_mocker.url_count == 1 assert dataset.description != ''
def test_fetch_atlas_msdl(tmp_path, request_mocker): datadir = str(tmp_path / 'msdl_atlas') os.mkdir(datadir) os.mkdir(os.path.join(datadir, 'MSDL_rois')) data_dir = os.path.join(datadir, 'MSDL_rois', 'msdl_rois_labels.csv') csv = np.rec.array([(1.5, 1.5, 1.5, 'Aud', 'Aud'), (1.2, 1.3, 1.4, 'DMN', 'DMN')], dtype=[('x', '<f8'), ('y', '<f8'), ('z', '<f8'), ('name', 'S12'), ('net_name', 'S19')]) with open(data_dir, 'wb') as csv_file: header = '{0}\n'.format(','.join(csv.dtype.names)) csv_file.write(header.encode()) np.savetxt(csv_file, csv, delimiter=',', fmt='%s') dataset = atlas.fetch_atlas_msdl(data_dir=str(tmp_path), verbose=0) assert isinstance(dataset.labels, list) assert isinstance(dataset.region_coords, list) assert isinstance(dataset.networks, list) assert isinstance(dataset.maps, _basestring) assert len(tst.mock_url_request.urls) == 1 assert dataset.description != ''
def test_fetch_atlas_msdl(): datadir = os.path.join(tst.tmpdir, 'msdl_atlas') os.mkdir(datadir) os.mkdir(os.path.join(datadir, 'MSDL_rois')) data_dir = os.path.join(datadir, 'MSDL_rois', 'msdl_rois_labels.csv') csv = np.rec.array([(1.5, 1.5, 1.5, 'Aud', 'Aud'), (1.2, 1.3, 1.4, 'DMN', 'DMN')], dtype=[('x', '<f8'), ('y', '<f8'), ('z', '<f8'), ('name', 'S12'), ('net_name', 'S19')]) with open(data_dir, 'wb') as csv_file: header = '{0}\n'.format(','.join(csv.dtype.names)) csv_file.write(header.encode()) np.savetxt(csv_file, csv, delimiter=',', fmt='%s') dataset = atlas.fetch_atlas_msdl(data_dir=tst.tmpdir, verbose=0) assert_true(isinstance(dataset.labels, list)) assert_true(isinstance(dataset.region_coords, list)) assert_true(isinstance(dataset.networks, list)) assert_true(isinstance(dataset.maps, _basestring)) assert_equal(len(tst.mock_url_request.urls), 1) assert_not_equal(dataset.description, '')
def test_fetch_atlas_msdl(): dataset = atlas.fetch_atlas_msdl(data_dir=tst.tmpdir, verbose=0) assert_true(isinstance(dataset.labels, _basestring)) assert_true(isinstance(dataset.maps, _basestring)) assert_equal(len(tst.mock_url_request.urls), 1) assert_not_equal(dataset.description, '')
def test_fetch_atlas_msdl(): dataset = atlas.fetch_atlas_msdl(data_dir=tmpdir, verbose=0) assert_true(isinstance(dataset.labels, _basestring)) assert_true(isinstance(dataset.maps, _basestring)) assert_equal(len(mock_url_request.urls), 1)