mirror of
https://github.com/bec-project/bec_widgets.git
synced 2025-07-14 11:41:49 +02:00
fix: test_config_dialog.py - test_add_new_plot_and_modify qtbot action .click() changed -> function called directly
This commit is contained in:
wyzula-jan
@ -1,200 +1,202 @@
|
|||||||
# import os
|
import os
|
||||||
# import yaml
|
import yaml
|
||||||
#
|
|
||||||
# import pytest
|
import pytest
|
||||||
# from PyQt5.QtWidgets import QTabWidget, QTableWidgetItem, QApplication
|
from PyQt5.QtWidgets import QTabWidget, QTableWidgetItem, QApplication
|
||||||
#
|
|
||||||
# from bec_widgets.widgets import ConfigDialog
|
from bec_widgets.widgets import ConfigDialog
|
||||||
#
|
|
||||||
# current_path = os.path.dirname(__file__)
|
current_path = os.path.dirname(__file__)
|
||||||
#
|
|
||||||
#
|
|
||||||
# def load_config(config_path):
|
def load_config(config_path):
|
||||||
# """Helper function to load config from yaml file."""
|
"""Helper function to load config from yaml file."""
|
||||||
# with open(config_path, "r") as f:
|
with open(config_path, "r") as f:
|
||||||
# config = yaml.safe_load(f)
|
config = yaml.safe_load(f)
|
||||||
# return config
|
return config
|
||||||
#
|
|
||||||
#
|
|
||||||
# # save configs as for test_bec_monitor.py
|
# save configs as for test_bec_monitor.py
|
||||||
# config_device = load_config(os.path.join(current_path, "test_configs/config_device.yaml"))
|
config_device = load_config(os.path.join(current_path, "test_configs/config_device.yaml"))
|
||||||
# config_device_no_entry = load_config(
|
config_device_no_entry = load_config(
|
||||||
# os.path.join(current_path, "test_configs/config_device_no_entry.yaml")
|
os.path.join(current_path, "test_configs/config_device_no_entry.yaml")
|
||||||
# )
|
)
|
||||||
# config_scan = load_config(os.path.join(current_path, "test_configs/config_scan.yaml"))
|
config_scan = load_config(os.path.join(current_path, "test_configs/config_scan.yaml"))
|
||||||
#
|
|
||||||
#
|
|
||||||
# @pytest.fixture(scope="module") # TODO is this needed?
|
@pytest.fixture(scope="module") # TODO is this needed?
|
||||||
# def app():
|
def app():
|
||||||
# app = QApplication([])
|
app = QApplication([])
|
||||||
# yield app
|
yield app
|
||||||
#
|
|
||||||
#
|
|
||||||
# @pytest.fixture
|
@pytest.fixture
|
||||||
# def qtbot(app, qtbot): # TODO is this needed?
|
def qtbot(app, qtbot): # TODO is this needed?
|
||||||
# """A qtbot fixture to ensure that widgets are closed after being used."""
|
"""A qtbot fixture to ensure that widgets are closed after being used."""
|
||||||
# qtbot.old_widgets = set(app.topLevelWidgets())
|
qtbot.old_widgets = set(app.topLevelWidgets())
|
||||||
# yield qtbot
|
yield qtbot
|
||||||
# new_widgets = set(app.topLevelWidgets()) - qtbot.old_widgets
|
new_widgets = set(app.topLevelWidgets()) - qtbot.old_widgets
|
||||||
# for widget in new_widgets:
|
for widget in new_widgets:
|
||||||
# widget.close()
|
widget.close()
|
||||||
#
|
|
||||||
#
|
|
||||||
# def setup_config_dialog(qtbot):
|
def setup_config_dialog(qtbot):
|
||||||
# widget = ConfigDialog()
|
widget = ConfigDialog()
|
||||||
# qtbot.addWidget(widget)
|
qtbot.addWidget(widget)
|
||||||
# qtbot.waitExposed(widget)
|
qtbot.waitExposed(widget)
|
||||||
# return widget
|
return widget
|
||||||
#
|
|
||||||
#
|
|
||||||
# @pytest.mark.parametrize("config", [config_device, config_scan])
|
@pytest.mark.parametrize("config", [config_device, config_scan])
|
||||||
# def test_load_config(qtbot, config):
|
def test_load_config(qtbot, config):
|
||||||
# config_dialog = setup_config_dialog(qtbot)
|
config_dialog = setup_config_dialog(qtbot)
|
||||||
# config_dialog.load_config(config)
|
config_dialog.load_config(config)
|
||||||
#
|
|
||||||
# assert (
|
assert (
|
||||||
# config_dialog.comboBox_appearance.currentText()
|
config_dialog.comboBox_appearance.currentText()
|
||||||
# == config["plot_settings"]["background_color"]
|
== config["plot_settings"]["background_color"]
|
||||||
# )
|
)
|
||||||
# assert config_dialog.spinBox_n_column.value() == config["plot_settings"]["num_columns"]
|
assert config_dialog.spinBox_n_column.value() == config["plot_settings"]["num_columns"]
|
||||||
# assert config_dialog.comboBox_colormap.currentText() == config["plot_settings"]["colormap"]
|
assert config_dialog.comboBox_colormap.currentText() == config["plot_settings"]["colormap"]
|
||||||
#
|
|
||||||
#
|
|
||||||
# @pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
# "config, scan_mode",
|
"config, scan_mode",
|
||||||
# [
|
[
|
||||||
# (config_device, False),
|
(config_device, False),
|
||||||
# (config_scan, True),
|
(config_scan, True),
|
||||||
# (config_device_no_entry, False),
|
(config_device_no_entry, False),
|
||||||
# ],
|
],
|
||||||
# )
|
)
|
||||||
# def test_initialization(qtbot, config, scan_mode):
|
def test_initialization(qtbot, config, scan_mode):
|
||||||
# config_dialog = setup_config_dialog(qtbot)
|
config_dialog = setup_config_dialog(qtbot)
|
||||||
# config_dialog.load_config(config)
|
config_dialog.load_config(config)
|
||||||
#
|
|
||||||
# assert isinstance(config_dialog, ConfigDialog)
|
assert isinstance(config_dialog, ConfigDialog)
|
||||||
# assert (
|
assert (
|
||||||
# config_dialog.comboBox_appearance.currentText()
|
config_dialog.comboBox_appearance.currentText()
|
||||||
# == config["plot_settings"]["background_color"]
|
== config["plot_settings"]["background_color"]
|
||||||
# )
|
)
|
||||||
# assert config_dialog.spinBox_n_column.value() == config["plot_settings"]["num_columns"]
|
assert config_dialog.spinBox_n_column.value() == config["plot_settings"]["num_columns"]
|
||||||
# assert (config_dialog.comboBox_scanTypes.currentText() == "Enabled") == scan_mode
|
assert (config_dialog.comboBox_scanTypes.currentText() == "Enabled") == scan_mode
|
||||||
# assert (
|
assert (
|
||||||
# config_dialog.tabWidget_scan_types.count() > 0
|
config_dialog.tabWidget_scan_types.count() > 0
|
||||||
# ) # Ensures there's at least one tab created
|
) # Ensures there's at least one tab created
|
||||||
#
|
|
||||||
# # If there's a need to check the contents of the first tab (there has to be always at least one tab)
|
# If there's a need to check the contents of the first tab (there has to be always at least one tab)
|
||||||
# first_tab = config_dialog.tabWidget_scan_types.widget(0)
|
first_tab = config_dialog.tabWidget_scan_types.widget(0)
|
||||||
# if scan_mode:
|
if scan_mode:
|
||||||
# assert (
|
assert (
|
||||||
# first_tab.findChild(QTabWidget, "tabWidget_plots") is not None
|
first_tab.findChild(QTabWidget, "tabWidget_plots") is not None
|
||||||
# ) # Ensures plot tab widget exists in scan mode
|
) # Ensures plot tab widget exists in scan mode
|
||||||
# else:
|
else:
|
||||||
# assert (
|
assert (
|
||||||
# first_tab.findChild(QTabWidget) is not None
|
first_tab.findChild(QTabWidget) is not None
|
||||||
# ) # Ensures plot tab widget exists in default mode
|
) # Ensures plot tab widget exists in default mode
|
||||||
#
|
|
||||||
#
|
|
||||||
# def test_edit_and_apply_config(qtbot):
|
def test_edit_and_apply_config(qtbot):
|
||||||
# config_dialog = setup_config_dialog(qtbot)
|
config_dialog = setup_config_dialog(qtbot)
|
||||||
# config_dialog.load_config(config_device)
|
config_dialog.load_config(config_device)
|
||||||
#
|
|
||||||
# config_dialog.comboBox_appearance.setCurrentText("white")
|
config_dialog.comboBox_appearance.setCurrentText("white")
|
||||||
# config_dialog.spinBox_n_column.setValue(2)
|
config_dialog.spinBox_n_column.setValue(2)
|
||||||
# config_dialog.comboBox_colormap.setCurrentText("viridis")
|
config_dialog.comboBox_colormap.setCurrentText("viridis")
|
||||||
#
|
|
||||||
# applied_config = config_dialog.apply_config()
|
applied_config = config_dialog.apply_config()
|
||||||
#
|
|
||||||
# assert applied_config["plot_settings"]["background_color"] == "white"
|
assert applied_config["plot_settings"]["background_color"] == "white"
|
||||||
# assert applied_config["plot_settings"]["num_columns"] == 2
|
assert applied_config["plot_settings"]["num_columns"] == 2
|
||||||
# assert applied_config["plot_settings"]["colormap"] == "viridis"
|
assert applied_config["plot_settings"]["colormap"] == "viridis"
|
||||||
#
|
|
||||||
#
|
|
||||||
# def test_edit_and_apply_config_scan_mode(qtbot):
|
def test_edit_and_apply_config_scan_mode(qtbot):
|
||||||
# config_dialog = setup_config_dialog(qtbot)
|
config_dialog = setup_config_dialog(qtbot)
|
||||||
# config_dialog.load_config(config_scan)
|
config_dialog.load_config(config_scan)
|
||||||
#
|
|
||||||
# config_dialog.comboBox_appearance.setCurrentText("white")
|
config_dialog.comboBox_appearance.setCurrentText("white")
|
||||||
# config_dialog.spinBox_n_column.setValue(2)
|
config_dialog.spinBox_n_column.setValue(2)
|
||||||
# config_dialog.comboBox_colormap.setCurrentText("viridis")
|
config_dialog.comboBox_colormap.setCurrentText("viridis")
|
||||||
# config_dialog.comboBox_scanTypes.setCurrentText("Enabled")
|
config_dialog.comboBox_scanTypes.setCurrentText("Enabled")
|
||||||
#
|
|
||||||
# applied_config = config_dialog.apply_config()
|
applied_config = config_dialog.apply_config()
|
||||||
#
|
|
||||||
# assert applied_config["plot_settings"]["background_color"] == "white"
|
assert applied_config["plot_settings"]["background_color"] == "white"
|
||||||
# assert applied_config["plot_settings"]["num_columns"] == 2
|
assert applied_config["plot_settings"]["num_columns"] == 2
|
||||||
# assert applied_config["plot_settings"]["colormap"] == "viridis"
|
assert applied_config["plot_settings"]["colormap"] == "viridis"
|
||||||
# assert applied_config["plot_settings"]["scan_types"] is True
|
assert applied_config["plot_settings"]["scan_types"] is True
|
||||||
#
|
|
||||||
#
|
|
||||||
# def test_add_new_scan(qtbot):
|
def test_add_new_scan(qtbot):
|
||||||
# config_dialog = setup_config_dialog(qtbot)
|
config_dialog = setup_config_dialog(qtbot)
|
||||||
# # Ensure the tab count is initially 1 (from the default config)
|
# Ensure the tab count is initially 1 (from the default config)
|
||||||
# assert config_dialog.tabWidget_scan_types.count() == 1
|
assert config_dialog.tabWidget_scan_types.count() == 1
|
||||||
#
|
|
||||||
# # Add a new scan tab
|
# Add a new scan tab
|
||||||
# config_dialog.add_new_scan(config_dialog.tabWidget_scan_types, "Test Scan Tab")
|
config_dialog.add_new_scan(config_dialog.tabWidget_scan_types, "Test Scan Tab")
|
||||||
#
|
|
||||||
# # Ensure the tab count is now 2
|
# Ensure the tab count is now 2
|
||||||
# assert config_dialog.tabWidget_scan_types.count() == 2
|
assert config_dialog.tabWidget_scan_types.count() == 2
|
||||||
#
|
|
||||||
# # Ensure the new tab has the correct name
|
# Ensure the new tab has the correct name
|
||||||
# assert config_dialog.tabWidget_scan_types.tabText(1) == "Test Scan Tab"
|
assert config_dialog.tabWidget_scan_types.tabText(1) == "Test Scan Tab"
|
||||||
#
|
|
||||||
#
|
|
||||||
# def test_add_new_plot_and_modify(qtbot):
|
def test_add_new_plot_and_modify(qtbot):
|
||||||
# config_dialog = setup_config_dialog(qtbot)
|
config_dialog = setup_config_dialog(qtbot)
|
||||||
# # Ensure the tab count is initially 1 and it is called "Default"
|
# Ensure the tab count is initially 1 and it is called "Default"
|
||||||
# assert config_dialog.tabWidget_scan_types.count() == 1
|
assert config_dialog.tabWidget_scan_types.count() == 1
|
||||||
# assert config_dialog.tabWidget_scan_types.tabText(0) == "Default"
|
assert config_dialog.tabWidget_scan_types.tabText(0) == "Default"
|
||||||
# # Get the first tab (which should be a scan tab)
|
# Get the first tab (which should be a scan tab)
|
||||||
# scan_tab = config_dialog.tabWidget_scan_types.widget(0)
|
scan_tab = config_dialog.tabWidget_scan_types.widget(0)
|
||||||
#
|
|
||||||
# # Ensure the plot tab count is initially 1 and it is called "Plot 1"
|
# Ensure the plot tab count is initially 1 and it is called "Plot 1"
|
||||||
# tabWidget_plots = scan_tab.findChild(QTabWidget)
|
tabWidget_plots = scan_tab.findChild(QTabWidget)
|
||||||
# assert tabWidget_plots.count() == 1
|
assert tabWidget_plots.count() == 1
|
||||||
# assert tabWidget_plots.tabText(0) == "Plot 1"
|
assert tabWidget_plots.tabText(0) == "Plot 1"
|
||||||
#
|
|
||||||
# # Add a new plot tab
|
# Add a new plot tab
|
||||||
# config_dialog.add_new_plot(scan_tab)
|
config_dialog.add_new_plot(scan_tab)
|
||||||
#
|
|
||||||
# # Ensure the plot tab count is now 2
|
# Ensure the plot tab count is now 2
|
||||||
# assert tabWidget_plots.count() == 2
|
assert tabWidget_plots.count() == 2
|
||||||
#
|
|
||||||
# # Ensure the new tab has the correct name
|
# Ensure the new tab has the correct name
|
||||||
# assert tabWidget_plots.tabText(1) == "Plot 2"
|
assert tabWidget_plots.tabText(1) == "Plot 2"
|
||||||
#
|
|
||||||
# # Access the new plot tab
|
# Access the new plot tab
|
||||||
# new_plot_tab = tabWidget_plots.widget(1)
|
new_plot_tab = tabWidget_plots.widget(1)
|
||||||
#
|
|
||||||
# # Modify the line edits within the new plot tab
|
# Modify the line edits within the new plot tab
|
||||||
# new_plot_tab.ui.lineEdit_plot_title.setText("Modified Plot Title")
|
new_plot_tab.ui.lineEdit_plot_title.setText("Modified Plot Title")
|
||||||
# new_plot_tab.ui.lineEdit_x_label.setText("Modified X Label")
|
new_plot_tab.ui.lineEdit_x_label.setText("Modified X Label")
|
||||||
# new_plot_tab.ui.lineEdit_y_label.setText("Modified Y Label")
|
new_plot_tab.ui.lineEdit_y_label.setText("Modified Y Label")
|
||||||
# new_plot_tab.ui.lineEdit_x_name.setText("Modified X Name")
|
new_plot_tab.ui.lineEdit_x_name.setText("Modified X Name")
|
||||||
# new_plot_tab.ui.lineEdit_x_entry.setText("Modified X Entry")
|
new_plot_tab.ui.lineEdit_x_entry.setText("Modified X Entry")
|
||||||
#
|
|
||||||
# # Modify the table for signals
|
# Modify the table for signals
|
||||||
# new_plot_tab.ui.pushButton_y_new.click() # Press button to add a new row
|
# new_plot_tab.ui.pushButton_y_new.click() # Press button to add a new row
|
||||||
# table = new_plot_tab.ui.tableWidget_y_signals
|
config_dialog.add_new_signal(new_plot_tab.ui.tableWidget_y_signals) # TODO change to click?
|
||||||
# assert table.rowCount() == 1 # Ensure the new row is added
|
|
||||||
#
|
table = new_plot_tab.ui.tableWidget_y_signals
|
||||||
# row_position = table.rowCount() - 1
|
assert table.rowCount() == 1 # Ensure the new row is added
|
||||||
#
|
|
||||||
# # Modify the first row
|
row_position = table.rowCount() - 1
|
||||||
# table.setItem(row_position, 0, QTableWidgetItem("New Signal Name"))
|
|
||||||
# table.setItem(row_position, 1, QTableWidgetItem("New Signal Entry"))
|
# Modify the first row
|
||||||
# # Apply the configuration
|
table.setItem(row_position, 0, QTableWidgetItem("New Signal Name"))
|
||||||
# config = config_dialog.apply_config()
|
table.setItem(row_position, 1, QTableWidgetItem("New Signal Entry"))
|
||||||
#
|
# Apply the configuration
|
||||||
# # Check if the modifications are reflected in the configuration
|
config = config_dialog.apply_config()
|
||||||
# modified_plot_config = config["plot_data"][
|
|
||||||
# 1
|
# Check if the modifications are reflected in the configuration
|
||||||
# ] # Assuming the new plot is the second item in the plot_data list
|
modified_plot_config = config["plot_data"][
|
||||||
# assert modified_plot_config["plot_name"] == "Modified Plot Title"
|
1
|
||||||
# assert modified_plot_config["x"]["label"] == "Modified X Label"
|
] # Assuming the new plot is the second item in the plot_data list
|
||||||
# assert modified_plot_config["y"]["label"] == "Modified Y Label"
|
assert modified_plot_config["plot_name"] == "Modified Plot Title"
|
||||||
# assert modified_plot_config["x"]["signals"][0]["name"] == "Modified X Name"
|
assert modified_plot_config["x"]["label"] == "Modified X Label"
|
||||||
# assert modified_plot_config["x"]["signals"][0]["entry"] == "Modified X Entry"
|
assert modified_plot_config["y"]["label"] == "Modified Y Label"
|
||||||
# assert modified_plot_config["y"]["signals"][0]["name"] == "New Signal Name"
|
assert modified_plot_config["x"]["signals"][0]["name"] == "Modified X Name"
|
||||||
# assert modified_plot_config["y"]["signals"][0]["entry"] == "New Signal Entry"
|
assert modified_plot_config["x"]["signals"][0]["entry"] == "Modified X Entry"
|
||||||
|
assert modified_plot_config["y"]["signals"][0]["name"] == "New Signal Name"
|
||||||
|
assert modified_plot_config["y"]["signals"][0]["entry"] == "New Signal Entry"
|
||||||
|
|||||||
Reference in New Issue
Block a user