def test_import_compumedics(): annot = Annotations(annot_file) record_start = datetime(2000, 1, 1, 0, 0, 0) annot.import_staging(str(annot_compumedics_path), 'compumedics', 'compumedics', record_start, staging_start=record_start) assert annot.time_in_stage('REM') == 5970
def test_fetch(): dset = Dataset(str(gui_file)) annot = Annotations(str(annot_psg_path)) seg = fetch(dset, annot, stage=['NREM2', 'NREM3'], epoch='locked', reject_epoch=True, reject_artf=True) assert len(seg) == 356 seg = fetch(dset, annot, cat=(0, 1, 0, 0), stage=['NREM2', 'NREM3'], reject_epoch=True, reject_artf=True) assert len(seg) == 31 assert seg[14]['times'][0] == (34380, 34410) seg = fetch(dset, annot, cat=(0, 0, 1, 0), stage=['NREM1']) seg.read_data(['EEG Fpz-Cz'], ref_chan=['EEG Pz-Oz']) assert seg[0]['data']()[0][0].shape == (297000, ) assert approx(seg[0]['data']()[0][0][100]) == -4.3201466
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_get_times(): annot = Annotations(str(annot_psg_path)) bundles = get_times(annot, stage=['NREM2'], exclude=True) assert len(bundles[0]['times']) == 229 assert bundles[0]['stage'] == 'NREM2'
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_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_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_read_annot_str(): Annotations(annot_file)
def test_export_sleepstats(): annot = Annotations(annot_psg_path) assert annot.export_sleep_stats(annot_sleepstats_path, 0, 29000) == (338, 36, 133)
def test_import_sandman(): annot = Annotations(annot_file) record_start = datetime(2010, 10, 10, 21, 3, 36) annot.import_staging(str(annot_sandman_path), 'sandman', 'sandman', record_start) assert annot.time_in_stage('REM') == 5610
def test_import_remlogic(): annot = Annotations(annot_file) record_start = datetime(2016, 8, 9, 22, 21, 30) annot.import_staging(str(annot_remlogic_path), 'remlogic', 'remlogic', record_start) assert annot.time_in_stage('REM') == 3420
def test_import_domino(): annot = Annotations(annot_file) record_start = datetime(2015, 9, 21, 21, 40, 30) annot.import_staging(str(annot_domino_path), 'domino', 'domino', record_start) assert annot.time_in_stage('REM') == 2460
def test_import_alice(): annot = Annotations(annot_file) record_start = datetime(2000, 1, 1, 22, 55, 6) annot.import_staging(str(annot_alice_path), 'alice', 'alice', record_start) assert annot.time_in_stage('REM') == 5160
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