mirror of
https://github.com/bec-project/bec_widgets.git
synced 2025-07-14 03:31:50 +02:00
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
import pytest
|
|
|
|
from bec_widgets.widgets.dap.dap_combo_box.dap_combo_box import DapComboBox
|
|
|
|
from .client_mocks import mocked_client
|
|
from .conftest import create_widget
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def dap_combobox(qtbot, mocked_client):
|
|
"""DapComboBox fixture."""
|
|
models = ["GaussianModel", "LorentzModel", "SineModel"]
|
|
mocked_client.dap._available_dap_plugins.keys.return_value = models
|
|
widget = create_widget(qtbot, DapComboBox, client=mocked_client)
|
|
return widget
|
|
|
|
|
|
def test_dap_combobox_init(dap_combobox):
|
|
"""Test DapComboBox init."""
|
|
assert dap_combobox.fit_model_combobox.currentText() == "GaussianModel"
|
|
assert dap_combobox.available_models == ["GaussianModel", "LorentzModel", "SineModel"]
|
|
assert dap_combobox._validate_dap_model("GaussianModel") is True
|
|
assert dap_combobox._validate_dap_model("somemodel") is False
|
|
assert dap_combobox._validate_dap_model(None) is False
|
|
|
|
|
|
def test_dap_combobox_set_axis(dap_combobox):
|
|
"""Test DapComboBox set axis."""
|
|
# Container to store the messages
|
|
container = []
|
|
|
|
def my_callback(msg: str):
|
|
"""Calback function to store the messages."""
|
|
container.append(msg)
|
|
|
|
dap_combobox.x_axis_updated.connect(my_callback)
|
|
dap_combobox.y_axis_updated.connect(my_callback)
|
|
dap_combobox.select_x_axis("x_axis")
|
|
assert dap_combobox.x_axis == "x_axis"
|
|
dap_combobox.select_y_axis("y_axis")
|
|
assert dap_combobox.y_axis == "y_axis"
|
|
assert container[0] == "x_axis"
|
|
assert container[1] == "y_axis"
|
|
|
|
|
|
def test_dap_combobox_select_fit(dap_combobox):
|
|
"""Test DapComboBox select fit."""
|
|
# Container to store the messages
|
|
container = []
|
|
|
|
def my_callback(msg: str):
|
|
"""Calback function to store the messages."""
|
|
container.append(msg)
|
|
|
|
dap_combobox.fit_model_updated.connect(my_callback)
|
|
dap_combobox.select_fit_model("LorentzModel")
|
|
assert dap_combobox.fit_model_combobox.currentText() == "LorentzModel"
|
|
assert container[0] == "LorentzModel"
|
|
|
|
|
|
def test_dap_combobox_currentTextchanged(dap_combobox):
|
|
"""Test DapComboBox currentTextChanged."""
|
|
# Container to store the messages
|
|
container = []
|
|
|
|
def my_callback(msg: str):
|
|
"""Calback function to store the messages."""
|
|
container.append(msg)
|
|
|
|
assert dap_combobox.fit_model_combobox.currentText() == "GaussianModel"
|
|
dap_combobox.fit_model_updated.connect(my_callback)
|
|
dap_combobox.fit_model_combobox.setCurrentText("SineModel")
|
|
assert container[0] == "SineModel"
|