def test_raw_data_load_data_in_memory(sample_ids, tiny_files_structure):
raw = RawData(sample_ids)
raw.load_data_in_memory()
assert not raw.correlations
assert raw.y.all().all()
assert not raw.icn
assert not raw.fmri_maps
def load_data(self, ids: np.ndarray, set_id: str,
file_structure: Structure) -> RawData:
raw = RawData(ids, set_id)
raw.load_data_in_memory(fmri_path=file_structure.raw.fmri_map)
return raw
def build_dataset(self, raw: RawData, out_path: Path) -> dd.DataFrame:
raw.load_data_in_memory(fmri_path=structure.raw.fmri_map)
ddf = self.make_fmri_features(raw.fmri_maps)
return ddf
def load_data(self, ids: np.ndarray, set_id: str,
file_structure: Structure) -> RawData:
raw = RawData(ids, set_id)
raw.load_data_in_memory(file_structure.raw.correlations)
return raw
def test_raw_load_icn(sample_ids):
raw = RawData(sample_ids)
raw.load_icn(structure.raw.icn)
assert raw.icn.all()
def test_raw_load_loading_data(sample_ids, tiny_files_structure):
raw = RawData(sample_ids)
raw.load_loading_data(tiny_files_structure.raw.loading)
assert raw.loadings.compute().any().all()
def test_raw_load_fmri_map(sample_ids):
raw = RawData(sample_ids)
raw.load_fmri(structure.raw.fmri_map)
assert raw.fmri_maps
assert raw.fmri_maps[0].compute().any()
def test_raw_load_correlations(sample_ids, tiny_files_structure):
raw = RawData(sample_ids)
raw.load_correlations(tiny_files_structure.raw.correlations)
assert raw.correlations.compute().any().all()
def test_raw_load_y(sample_ids, tiny_files_structure):
raw = RawData(sample_ids)
raw.load_y(tiny_files_structure.raw.y_train)
assert raw.y.all().all()
def test_raw_data_init(sample_ids):
raw = RawData(sample_ids)
assert raw.ids
def raw_data_sample():
raw = RawData([10001, 10002])
return raw