def test_events():
d = Dataset(ns2_file)
create_empty_annotations(annot_file, d)
annot = Annotations(annot_file)
with raises(IndexError):
annot.add_event_type('spindle')
annot.add_rater('test')
annot.add_event_type('spindle')
annot.add_event_type('spindle')
assert len(annot.event_types) == 1
annot.add_event('slowwave', (1, 2), chan=('FP1', ))
annot.add_event('spindle', (3, 4))
assert len(annot.event_types) == 2
assert len(annot.get_events()) == 2
annot.remove_event_type('spindle')
assert len(annot.event_types) == 1
assert len(annot.get_events()) == 1
annot.remove_event('slowwave')
assert len(annot.event_types) == 1
assert len(annot.get_events()) == 0
def test_read_annot_path():
annot = Annotations(annot_file)
annot.add_rater('test')
annot.add_rater('test')
annot.add_rater('test_2')
annot.dataset
annot.start_time
annot.first_second
annot.last_second
assert len(annot.raters) == 2
annot.remove_rater('test_2')
assert len(annot.raters) == 1
assert annot.current_rater == 'test'
annot.set_stage_for_epoch(510, 'REM')
assert annot.get_epoch_start(517) == 510
assert annot.get_stage_for_epoch(510, 30) == 'REM'
with raises(KeyError):
annot.get_rater('XXX')
annot.export(annot_export_file)
with annot_export_file.open() as f:
assert '18:31:14,510,540,REM' in f.read()
def test_epochs():
d = Dataset(ns2_file)
create_empty_annotations(annot_file, d)
annot = Annotations(annot_file)
annot.add_rater('test')
assert len(annot.get_epochs()) == 50
assert len(annot.get_epochs(time=(1000, 2000))) == 16
def test_bookmarks():
d = Dataset(ns2_file)
create_empty_annotations(annot_file, d)
annot = Annotations(annot_file)
with raises(IndexError):
annot.current_rater
with raises(IndexError):
annot.add_bookmark('bookmark', (1, 2), ('Fpz', ))
annot.add_rater('test')
annot.add_bookmark('bookmark', (1, 2), ('Fpz', ))
assert len(annot.get_bookmarks()) == 1
annot.remove_bookmark('bookmark')
assert len(annot.get_bookmarks()) == 0
def test_get_cycles():
d = Dataset(ns2_file)
create_empty_annotations(annot_file, d)
annot = Annotations(annot_file)
annot.add_rater('test')
annot.set_cycle_mrkr(510)
annot.set_cycle_mrkr(540)
annot.set_cycle_mrkr(570)
annot.set_cycle_mrkr(600, end=True)
cycles = annot.get_cycles()
assert len(cycles) == 3
assert cycles[2] == (570, 600, 3)
annot.remove_cycle_mrkr(510)
annot.clear_cycles()
cycles = annot.get_cycles()
assert cycles is None