def test_widget_labels(qtbot):
w = MainWindow()
qtbot.addWidget(w)
assert not w.labels.isEnabled()
w.info.open_dataset(str(gui_file))
labels_orig = w.labels.chan_name
assert w.labels.isEnabled()
w.action['dockwidgets'][1].trigger()
w.labels.table.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'labels_01_table.png'))
w.labels.table.setStyleSheet("")
# change label of second channel
w.labels.table.item(1, 1).setText('newlabel')
w.labels.grab().save(str(GUI_PATH / 'labels_02_newlabel.png'))
w.labels.table.item(3, 1).setText('newlabel')
w.labels.grab().save(str(GUI_PATH / 'labels_03_duplicate.png'))
assert not w.labels.idx_apply.isEnabled()
# fix it
w.labels.table.item(3, 1).setText('correctlabel')
w.labels.grab().save(str(GUI_PATH / 'labels_04_correct.png'))
assert w.labels.idx_apply.isEnabled()
# cancel should reset the list
labels_changed = w.labels._read_labels()
w.labels.idx_cancel.click()
labels_cancel = w.labels._read_labels()
assert labels_cancel == labels_orig
assert labels_cancel != labels_changed
# apply
w.labels.table.item(2, 1).setText('newlabel')
w.labels.idx_apply.click()
assert w.channels.isEnabled()
channel_make_group(w)
find_in_qt(w.channels, QPushButton, 'Apply').click()
w.grab().save(str(GUI_PATH / 'labels_05_traces.png'))
# load data
w.labels.idx_load.setStyleSheet("background-color: red;")
w.labels.grab().save(str(GUI_PATH / 'labels_06_load_button.png'))
w.labels.idx_load.setStyleSheet("")
labels_file = SAMPLE_PATH / 'labels_file.csv'
with labels_file.open('w') as f:
f.write('newchan1, newchan2\tnewchan3;\nnewchan5')
w.labels.load_labels(test_name=str(labels_file))
w.labels.grab().save(str(GUI_PATH / 'labels_07_loaded.png'))
def test_widget_channels_save(qtbot):
w = MainWindow()
qtbot.addWidget(w)
assert not w.channels.action['save_channels'].isEnabled()
w.info.open_dataset(str(gui_file))
assert w.channels.action['save_channels'].isEnabled()
channel_make_group(w)
w.channels.new_group(test_name='eog')
chan_tab_i = w.channels.tabs.currentIndex()
channelsgroup = w.channels.tabs.widget(chan_tab_i)
channelsgroup.idx_l0.item(2).setSelected(True)
channelsgroup.idx_hp.setValue(20)
channelsgroup.idx_lp.setValue(0)
channelsgroup.idx_scale.setValue(10)
w.channels.color_group(test_color=QColor('#ff4a87'))
w.channels.button_apply.click()
toolbar = w.findChild(QToolBar, 'File Management')
button_save = find_in_qt(toolbar, QToolButton, "Save Montage")
button_save.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'channels_12_save_chan.png'))
button_save.setStyleSheet("")
w.channels.save_channels(test_name=str(channel_montage_file))
def test_scroll_data(qtbot):
w = MainWindow()
qtbot.addWidget(w)
w.grab().save(str(GUI_PATH / 'open_01_start.png'))
w.info.idx_filename.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'open_02_open_dataset.png'))
w.info.idx_filename.setStyleSheet("")
w.info.open_dataset(str(gui_file))
new_button = w.channels.layout().itemAt(0).itemAt(0).widget()
new_button.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'open_03_loaded.png'))
new_button.setStyleSheet("")
channel_make_group(w, png=True)
# this shows selected channels and the apply button
button_apply = find_in_qt(w.channels, QPushButton, 'Apply')
button_apply.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'open_05_chan.png'))
button_apply.setStyleSheet("")
button_apply.click()
w.grab().save(str(GUI_PATH / 'open_06_traces.png'))
def test_widget_notes_show_fasst(qtbot):
w = MainWindow()
qtbot.addWidget(w)
w.notes.import_fasst(test_fasst=str(annot_fasst_path),
test_annot=str(annot_fasst_export_file))
w.grab().save(str(GUI_PATH / 'notes_05_show_imported.png'))
def test_widget_notes_load(qtbot):
w = MainWindow()
qtbot.addWidget(w)
toolbar = w.findChild(QToolBar, 'File Management')
button_save = find_in_qt(toolbar, QToolButton, "Load Annotations")
button_save.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'notes_03_load.png'))
button_save.setStyleSheet("")
def test_widget_channels_load(qtbot):
w = MainWindow()
qtbot.addWidget(w)
assert not w.channels.action['load_channels'].isEnabled()
w.info.open_dataset(str(gui_file))
assert w.channels.action['load_channels'].isEnabled()
toolbar = w.findChild(QToolBar, 'File Management')
button_load = find_in_qt(toolbar, QToolButton, "Load Montage")
button_load.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'channels_13_load_chan.png'))
button_load.setStyleSheet("")
w.channels.load_channels(test_name=str(channel_montage_file))
w.grab().save(str(GUI_PATH / 'channels_14_loaded.png'))
def test_widget_notes_toolbar(qtbot):
w = MainWindow()
qtbot.addWidget(w)
w.info.open_dataset(str(gui_file))
channel_make_group(w)
w.channels.button_apply.click()
w.channels.new_group(test_name='eog')
w.notes.update_notes(annot_psg_path)
w.traces.Y_wider()
w.traces.go_to_epoch(test_text_str='23:00')
toolbar = w.findChild(QToolBar, 'Annotations')
button_event = find_in_qt(toolbar, QToolButton, 'Event Mode')
button_event.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'notes_07_event_mode.png'))
button_event.setStyleSheet("")
box_evttype = find_in_qt_by_idx(toolbar, QComboBox, 0)
box_evttype.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'notes_08_evt_type_box.png'))
box_evttype.setStyleSheet("")
box_stage = find_in_qt_by_idx(toolbar, QComboBox, 1)
box_stage.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'notes_09_stage_box.png'))
box_stage.setStyleSheet("")
box_qual = find_in_qt_by_idx(toolbar, QComboBox, 2)
box_qual.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'notes_10_quality_box.png'))
box_qual.setStyleSheet("")
w.close()
def test_widget_channels(qtbot):
w = MainWindow()
qtbot.addWidget(w)
assert not w.channels.isEnabled()
w.info.open_dataset(str(gui_file))
assert w.channels.isEnabled()
assert not w.channels.button_color.isEnabled()
assert not w.channels.button_del.isEnabled()
assert not w.channels.button_apply.isEnabled()
channel_make_group(w)
assert w.channels.button_color.isEnabled()
assert w.channels.button_del.isEnabled()
assert w.channels.button_apply.isEnabled()
w.channels.button_apply.click()
w.channels.grab().save(str(GUI_PATH / 'channels_01_onegroup.png'))
w.channels.new_group(test_name='eog')
chan_tab_i = w.channels.tabs.currentIndex()
channelsgroup = w.channels.tabs.widget(chan_tab_i)
channelsgroup.idx_l0.item(2).setSelected(True)
w.channels.grab().save(str(GUI_PATH / 'channels_02_eog.png'))
chan_tab_i = w.channels.tabs.currentIndex()
channelsgroup = w.channels.tabs.widget(chan_tab_i)
channelsgroup.idx_l0.item(2).setSelected(True)
channelsgroup.idx_hp.setValue(20)
channelsgroup.idx_hp.setStyleSheet("background-color: red;")
w.channels.grab().save(str(GUI_PATH / 'channels_03_hp.png'))
channelsgroup.idx_hp.setStyleSheet("")
channelsgroup.idx_lp.setValue(0)
channelsgroup.idx_lp.setStyleSheet("background-color: red;")
w.channels.grab().save(str(GUI_PATH / 'channels_04_lp.png'))
channelsgroup.idx_lp.setStyleSheet("")
channelsgroup.idx_notch.setValue(60)
channelsgroup.idx_notch.setStyleSheet("background-color: red;")
w.channels.grab().save(str(GUI_PATH / 'channels_05_notch.png'))
channelsgroup.idx_notch.setStyleSheet("")
channelsgroup.idx_scale.setValue(10)
channelsgroup.idx_scale.setStyleSheet("background-color: red;")
w.channels.grab().save(str(GUI_PATH / 'channels_05_scale.png'))
channelsgroup.idx_scale.setStyleSheet("")
channelsgroup.idx_demean.setCheckState(Qt.Checked)
channelsgroup.idx_demean.setStyleSheet("background-color: red;")
w.channels.grab().save(str(GUI_PATH / 'channels_05_demean.png'))
channelsgroup.idx_demean.setStyleSheet("")
channelsgroup.idx_l1.setStyleSheet("background-color: red;")
w.channels.grab().save(str(GUI_PATH / 'channels_06_ref.png'))
channelsgroup.idx_l1.setStyleSheet("")
channelsgroup.idx_reref.setStyleSheet("background-color: red;")
w.channels.grab().save(str(GUI_PATH / 'channels_07_avgref.png'))
channelsgroup.idx_reref.setStyleSheet("")
button_apply = w.channels.button_apply
button_apply.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'channels_08_apply.png'))
button_apply.setStyleSheet("")
button_apply.click()
button_color = w.channels.button_color
button_color.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'channels_09_color.png'))
button_color.setStyleSheet("")
w.channels.color_group(test_color=QColor('#ff4a87'))
w.grab().save(str(GUI_PATH / 'channels_10_colored.png'))
button_delete = w.channels.button_del
button_delete.setStyleSheet("background-color: red;")
w.grab().save(str(GUI_PATH / 'channels_11_delete.png'))
button_delete.setStyleSheet("")
button_delete.click()
# delete also the first channel group
assert w.channels.button_color.isEnabled()
assert w.channels.button_del.isEnabled()
assert w.channels.button_apply.isEnabled()
button_delete.click()
assert not w.channels.button_color.isEnabled()
assert not w.channels.button_del.isEnabled()
assert not w.channels.button_apply.isEnabled()