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_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
def test_create_annot():
d = Dataset(ns2_file)
create_empty_annotations(annot_file, d)
