def test_determine_localization_or_montage(self, subject, experiment, session, localization, montage): with patched_cmlreader(): reader = CMLReader(subject=subject, experiment=experiment, session=session) assert reader.montage == montage assert reader.localization == localization
def test_load(self, file_type): with patched_cmlreader(datafile(file_type)): data_type = os.path.splitext(file_type)[0] reader = CMLReader(subject="R1405E", localization=0, experiment="FR5", session=1) reader.load(data_type=data_type)
def test_load_unimplemented(self): with patched_cmlreader(): reader = CMLReader(subject='R1405E', localization=0, experiment='FR1', session=0, montage=0) with pytest.raises(NotImplementedError): reader.load("fake_data")
def test_get_reader(self, file_type): with patched_cmlreader(): reader = CMLReader(subject='R1405E', localization=0, experiment='FR1', session=0, montage=0) reader_obj = reader.get_reader(file_type) assert type(reader_obj) == reader.readers[file_type]
def test_read_categories(self, kind, read_categories): from cmlreaders.path_finder import PathFinder from cmlreaders.readers.electrodes import ( ElectrodeCategoriesReader, MontageReader ) cpath = datafile("R1111M_electrode_categories.txt") categories = ElectrodeCategoriesReader.fromfile(cpath) mpath = datafile("R1111M_{}.json".format(kind)) with ExitStack() as stack: stack.enter_context(patched_cmlreader()) stack.enter_context(patch.object(PathFinder, "find", return_value="")) stack.enter_context(patch.object(ElectrodeCategoriesReader, "load", return_value=categories)) stack.enter_context(patch.object(MontageReader, "_file_path", mpath)) reader = CMLReader("R1111M", "FR1", 0, 0, 0) df = reader.load(kind, read_categories=read_categories) self.assert_categories_correct(df, categories, read_categories)
def events(): with patched_cmlreader(): cml_reader = CMLReader("R1389J") path = resource_filename('cmlreaders.test.data', 'all_events.json') reader = cml_reader.get_reader('events', file_path=path) return reader.as_dataframe()