bec-widgets 0.81.1__py3-none-any.whl → 0.82.1__py3-none-any.whl

tests/unit_tests/test_motor_map_widget.py

@@ -0,0 +1,196 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from bec_widgets.widgets.motor_map.motor_map_dialog.motor_map_settings import MotorMapSettings
+from bec_widgets.widgets.motor_map.motor_map_widget import BECMotorMapWidget
+
+from .client_mocks import mocked_client
+
+
+@pytest.fixture
+def motor_map_widget(qtbot, mocked_client):
+    widget = BECMotorMapWidget(client=mocked_client())
+    widget.toolbar.widgets["motor_x"].device_combobox.set_device_filter("FakePositioner")
+    widget.toolbar.widgets["motor_y"].device_combobox.set_device_filter("FakePositioner")
+    qtbot.addWidget(widget)
+    qtbot.waitExposed(widget)
+    yield widget
+    widget.close()
+
+
+@pytest.fixture
+def mock_motor_map(motor_map_widget):
+    motor_map_mock = MagicMock()
+    motor_map_widget.map = motor_map_mock
+    return motor_map_mock
+
+
+def test_motor_map_widget_init(motor_map_widget):
+    assert motor_map_widget is not None
+    assert motor_map_widget.client is not None
+    assert isinstance(motor_map_widget, BECMotorMapWidget)
+    assert motor_map_widget.config.widget_class == "BECMotorMapWidget"
+
+    # check initial state of toolbar actions
+    assert motor_map_widget.toolbar.widgets["connect"].action.isEnabled() == True
+    assert motor_map_widget.toolbar.widgets["config"].action.isEnabled() == False
+    assert motor_map_widget.toolbar.widgets["history"].action.isEnabled() == False
+    assert (
+        motor_map_widget.toolbar.widgets["motor_x"].device_combobox.config.device_filter
+        == "FakePositioner"
+    )
+    assert (
+        motor_map_widget.toolbar.widgets["motor_y"].device_combobox.config.device_filter
+        == "FakePositioner"
+    )
+    assert motor_map_widget.map.motor_x is None
+    assert motor_map_widget.map.motor_y is None
+
+
+###################################
+# Toolbar Actions
+###################################
+
+
+def test_motor_map_widget_change_motors_enable_toolbar(motor_map_widget):
+    motor_map_widget.change_motors("samx", "samy")
+    assert motor_map_widget.map.motor_x == "samx"
+    assert motor_map_widget.map.motor_y == "samy"
+    assert motor_map_widget.toolbar.widgets["motor_x"].device_combobox.currentText() == "samx"
+    assert motor_map_widget.toolbar.widgets["motor_y"].device_combobox.currentText() == "samy"
+    assert motor_map_widget.toolbar.widgets["config"].action.isEnabled() == True
+    assert motor_map_widget.toolbar.widgets["history"].action.isEnabled() == True
+
+
+###################################
+# Wrapper methods for MotorMap
+###################################
+
+
+def test_change_motors(motor_map_widget, mock_motor_map):
+    motor_map_widget.change_motors("motor_x", "motor_y", "motor_x_entry", "motor_y_entry", True)
+    mock_motor_map.change_motors.assert_called_once_with(
+        "motor_x", "motor_y", "motor_x_entry", "motor_y_entry", True
+    )
+
+
+def test_get_data(motor_map_widget, mock_motor_map):
+    motor_map_widget.get_data()
+    mock_motor_map.get_data.assert_called_once()
+
+
+def test_reset_history(motor_map_widget, mock_motor_map):
+    motor_map_widget.reset_history()
+    mock_motor_map.reset_history.assert_called_once()
+
+
+def test_set_color(motor_map_widget, mock_motor_map):
+    motor_map_widget.set_color("blue")
+    mock_motor_map.set_color.assert_called_once_with("blue")
+
+
+def test_set_max_points(motor_map_widget, mock_motor_map):
+    motor_map_widget.set_max_points(100)
+    mock_motor_map.set_max_points.assert_called_once_with(100)
+
+
+def test_set_precision(motor_map_widget, mock_motor_map):
+    motor_map_widget.set_precision(2)
+    mock_motor_map.set_precision.assert_called_once_with(2)
+
+
+def test_set_num_dim_points(motor_map_widget, mock_motor_map):
+    motor_map_widget.set_num_dim_points(50)
+    mock_motor_map.set_num_dim_points.assert_called_once_with(50)
+
+
+def test_set_background_value(motor_map_widget, mock_motor_map):
+    motor_map_widget.set_background_value(128)
+    mock_motor_map.set_background_value.assert_called_once_with(128)
+
+
+def test_set_scatter_size(motor_map_widget, mock_motor_map):
+    motor_map_widget.set_scatter_size(10)
+    mock_motor_map.set_scatter_size.assert_called_once_with(10)
+
+
+###################################
+# MotorMap Dialog
+###################################
+
+
+def test_motor_map_widget_clicked(motor_map_widget, qtbot):
+    motor_map_widget.toolbar.widgets["motor_x"].device_combobox.setCurrentText("samx")
+    motor_map_widget.toolbar.widgets["motor_y"].device_combobox.setCurrentText("samy")
+    motor_map_widget.toolbar.widgets["connect"].action.trigger()
+
+    qtbot.wait(200)
+
+    assert motor_map_widget.map.motor_x == "samx"
+    assert motor_map_widget.map.motor_y == "samy"
+    assert motor_map_widget.toolbar.widgets["config"].action.isEnabled() == True
+    assert motor_map_widget.toolbar.widgets["history"].action.isEnabled() == True
+
+
+@pytest.fixture
+def motor_map_settings(qtbot):
+    widget = MotorMapSettings()
+    qtbot.addWidget(widget)
+    qtbot.waitExposed(widget)
+    yield widget
+    widget.close()
+
+
+def test_display_current_settings(motor_map_settings):
+    config = {
+        "max_points": 100,
+        "num_dim_points": 50,
+        "precision": 2,
+        "scatter_size": 10,
+        "background_value": 128,
+        "color": (255, 0, 0, 255),
+    }
+
+    with patch("bec_widgets.utils.widget_io.WidgetIO.set_value") as mock_set_value:
+        with patch.object(motor_map_settings.ui.color, "setColor") as mock_set_color:
+            motor_map_settings.display_current_settings(config)
+            mock_set_value.assert_any_call(motor_map_settings.ui.max_points, config["max_points"])
+            mock_set_value.assert_any_call(
+                motor_map_settings.ui.trace_dim, config["num_dim_points"]
+            )
+            mock_set_value.assert_any_call(motor_map_settings.ui.precision, config["precision"])
+            mock_set_value.assert_any_call(
+                motor_map_settings.ui.scatter_size, config["scatter_size"]
+            )
+            mock_set_value.assert_any_call(
+                motor_map_settings.ui.background_value, 50
+            )  # 128/255*100 = 50
+            mock_set_color.assert_called_once_with(config["color"])
+
+
+def test_accept_changes(motor_map_settings):
+    with patch(
+        "bec_widgets.utils.widget_io.WidgetIO.get_value", side_effect=[100, 50, 2, 10, 50]
+    ) as mock_get_value:
+        with patch.object(
+            motor_map_settings.ui.color, "get_color", return_value=(255, 0, 0, 255)
+        ) as mock_get_color:
+            mock_target_widget = MagicMock()
+            motor_map_settings.set_target_widget(mock_target_widget)
+
+            motor_map_settings.accept_changes()
+
+            mock_get_value.assert_any_call(motor_map_settings.ui.max_points)
+            mock_get_value.assert_any_call(motor_map_settings.ui.trace_dim)
+            mock_get_value.assert_any_call(motor_map_settings.ui.precision)
+            mock_get_value.assert_any_call(motor_map_settings.ui.scatter_size)
+            mock_get_value.assert_any_call(motor_map_settings.ui.background_value)
+            mock_get_color.assert_called_once()
+
+            mock_target_widget.set_max_points.assert_called_once_with(100)
+            mock_target_widget.set_num_dim_points.assert_called_once_with(50)
+            mock_target_widget.set_precision.assert_called_once_with(2)
+            mock_target_widget.set_scatter_size.assert_called_once_with(10)
+            mock_target_widget.set_background_value.assert_called_once_with(127)
+            mock_target_widget.set_color.assert_called_once_with((255, 0, 0, 255))

tests/unit_tests/test_setting_dialog.py

@@ -0,0 +1,97 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+from qtpy.QtWidgets import QWidget
+
+from bec_widgets.qt_utils.settings_dialog import SettingsDialog, SettingWidget
+
+###################################
+# SettingWidget base class tests
+###################################
+
+
+@pytest.fixture
+def setting_widget(qtbot):
+    widget = SettingWidget()
+    qtbot.addWidget(widget)
+    qtbot.waitExposed(widget)
+    yield widget
+    widget.close()
+
+
+def test_setting_widget_initialization(setting_widget):
+    assert setting_widget.target_widget is None
+
+
+def test_setting_widget_set_target_widget(setting_widget):
+    mock_target = MagicMock(spec=QWidget)
+    setting_widget.set_target_widget(mock_target)
+    assert setting_widget.target_widget == mock_target
+
+
+def test_setting_widget_accept_changes(setting_widget):
+    with patch.object(setting_widget, "accept_changes") as mock_accept_changes:
+        setting_widget.accept_changes()
+        mock_accept_changes.assert_called_once()
+
+
+def test_setting_widget_display_current_settings(setting_widget):
+    config_dict = {"setting1": "value1", "setting2": "value2"}
+    with patch.object(setting_widget, "display_current_settings") as mock_display_current_settings:
+        setting_widget.display_current_settings(config_dict)
+        mock_display_current_settings.assert_called_once_with(config_dict)
+
+
+###################################
+# SettingsDialog tests
+###################################
+@pytest.fixture
+def settings_dialog(qtbot):
+    parent_widget = QWidget()
+    settings_widget = SettingWidget()
+    settings_widget.set_target_widget = MagicMock()
+    settings_widget.display_current_settings = MagicMock()
+    settings_widget.accept_changes = MagicMock()
+
+    dialog = SettingsDialog(
+        parent=parent_widget,
+        settings_widget=settings_widget,
+        window_title="Test Settings",
+        config={"setting1": "value1", "setting2": "value2"},
+    )
+    qtbot.addWidget(dialog)
+    qtbot.waitExposed(dialog)
+    yield dialog, parent_widget, settings_widget
+    dialog.close()
+
+
+def test_settings_dialog_initialization(settings_dialog):
+    dialog, parent_widget, settings_widget = settings_dialog
+
+    assert dialog.windowTitle() == "Test Settings"
+    settings_widget.set_target_widget.assert_called_once_with(parent_widget)
+    settings_widget.display_current_settings.assert_called_once_with(
+        {"setting1": "value1", "setting2": "value2"}
+    )
+
+
+def test_settings_dialog_accept(settings_dialog, qtbot):
+    dialog, _, settings_widget = settings_dialog
+
+    dialog.button_box.buttons()[0].click()  # OK Button
+    settings_widget.accept_changes.assert_called_once()
+
+
+def test_settings_dialog_reject(settings_dialog, qtbot):
+    dialog, _, _ = settings_dialog
+
+    with patch.object(dialog, "reject", wraps=dialog.reject) as mock_reject:
+        dialog.button_box.buttons()[1].click()  # Cancel Button
+        mock_reject.assert_called_once()
+
+
+def test_settings_dialog_apply_changes(settings_dialog, qtbot):
+    dialog, _, settings_widget = settings_dialog
+
+    dialog.apply_button.click()
+    settings_widget.accept_changes.assert_called_once()

tests/unit_tests/test_stop_button.py

@@ -2,7 +2,7 @@
 
 import pytest
 
-from bec_widgets.widgets.buttons import StopButton
+from bec_widgets.widgets.stop_button.stop_button import StopButton
 
 from .client_mocks import mocked_client
 
@@ -18,7 +18,10 @@ def stop_button(qtbot, mocked_client):
 
 def test_stop_button(stop_button):
     assert stop_button.text() == "Stop"
-    assert stop_button.styleSheet() == "background-color:  #cc181e; color: white"
+    assert (
+        stop_button.styleSheet()
+        == "background-color:  #cc181e; color: white; font-weight: bold; font-size: 12px;"
+    )
     stop_button.click()
     assert stop_button.queue.request_scan_abortion.called
     assert stop_button.queue.request_queue_reset.called

tests/unit_tests/test_toggle.py

@@ -0,0 +1,38 @@
+import pytest
+from qtpy.QtCore import QPointF, Qt
+
+from bec_widgets.widgets.toggle.toggle import ToggleSwitch
+
+
+@pytest.fixture
+def toggle(qtbot):
+    widget = ToggleSwitch()
+    qtbot.addWidget(widget)
+    qtbot.waitExposed(widget)
+    yield widget
+
+
+def test_toggle(toggle):
+    toggle.checked = False
+    assert toggle.checked is False
+
+    assert toggle.thumb_pos == QPointF(3, 2)
+
+    toggle.checked = True
+    assert toggle.checked is True
+
+    assert toggle.thumb_pos == QPointF(22, 2)
+
+
+def test_toggle_click(qtbot, toggle):
+    init_state = toggle.checked
+
+    qtbot.mouseClick(toggle, Qt.LeftButton)
+    toggle.paintEvent(None)
+    assert toggle.checked is not init_state
+
+    init_state = toggle.checked
+
+    qtbot.mouseClick(toggle, Qt.LeftButton)
+    toggle.paintEvent(None)
+    assert toggle.checked is not init_state

bec_widgets/widgets/buttons/__init__.py

@@ -1 +0,0 @@
-from .stop_button.stop_button import StopButton

bec_widgets/widgets/device_inputs/__init__.py

@@ -1,2 +0,0 @@
-from .device_combobox.device_combobox import DeviceComboBox
-from .device_line_edit.device_line_edit import DeviceLineEdit

bec_widgets/widgets/device_inputs/device_combobox/device_combobox.pyproject

@@ -1,4 +0,0 @@
-{
-    "files": ["device_combobox.py", "launch_device_combobox.py",
-    ]
-}

bec_widgets/widgets/device_inputs/device_combobox/launch_device_combobox.py

@@ -1,11 +0,0 @@
-from bec_widgets.widgets.device_inputs import DeviceComboBox
-
-if __name__ == "__main__":  # pragma: no cover
-    import sys
-
-    from qtpy.QtWidgets import QApplication
-
-    app = QApplication(sys.argv)
-    w = DeviceComboBox()
-    w.show()
-    sys.exit(app.exec_())

bec_widgets/widgets/device_inputs/device_line_edit/device_line_edit.pyproject

@@ -1,4 +0,0 @@
-{
-    "files": ["device_line_edit.py", "launch_device_line_edit.py",
-    ]
-}

bec_widgets/widgets/device_inputs/device_line_edit/launch_device_line_edit.py

@@ -1,11 +0,0 @@
-from bec_widgets.widgets.device_inputs import DeviceLineEdit
-
-if __name__ == "__main__":  # pragma: no cover
-    import sys
-
-    from qtpy.QtWidgets import QApplication
-
-    app = QApplication(sys.argv)
-    w = DeviceLineEdit()
-    w.show()
-    sys.exit(app.exec_())

bec_widgets/widgets/toolbar/__init__.py

@@ -1 +0,0 @@
-from .toolbar import ModularToolBar