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()