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