bec-widgets 0.99.1__py3-none-any.whl → 0.99.9__py3-none-any.whl

tests/unit_tests/test_waveform_widget.py

@@ -1,462 +0,0 @@
-from unittest.mock import MagicMock, patch
-
-import pyqtgraph as pg
-import pytest
-
-from bec_widgets.qt_utils.settings_dialog import SettingsDialog
-from bec_widgets.widgets.figure.plots.axis_settings import AxisSettings
-from bec_widgets.widgets.waveform.waveform_popups.curve_dialog.curve_dialog import CurveSettings
-from bec_widgets.widgets.waveform.waveform_widget import BECWaveformWidget
-
-from .client_mocks import mocked_client
-
-
-@pytest.fixture
-def waveform_widget(qtbot, mocked_client):
-    widget = BECWaveformWidget(client=mocked_client())
-    qtbot.addWidget(widget)
-    qtbot.waitExposed(widget)
-    yield widget
-
-
-@pytest.fixture
-def mock_waveform(waveform_widget):
-    waveform_mock = MagicMock()
-    waveform_widget.waveform = waveform_mock
-    return waveform_mock
-
-
-def test_waveform_widget_init(waveform_widget):
-    assert waveform_widget is not None
-    assert waveform_widget.client is not None
-    assert isinstance(waveform_widget, BECWaveformWidget)
-    assert waveform_widget.config.widget_class == "BECWaveformWidget"
-
-
-###################################
-# Wrapper methods for Waveform
-###################################
-
-
-def test_waveform_widget_get_curve(waveform_widget, mock_waveform):
-    waveform_widget.get_curve("curve_id")
-    waveform_widget.waveform.get_curve.assert_called_once_with("curve_id")
-
-
-def test_waveform_widget_set_colormap(waveform_widget, mock_waveform):
-    waveform_widget.set_colormap("colormap")
-    waveform_widget.waveform.set_colormap.assert_called_once_with("colormap")
-
-
-def test_waveform_widget_set_x(waveform_widget, mock_waveform):
-    waveform_widget.set_x("samx", "samx")
-    waveform_widget.waveform.set_x.assert_called_once_with("samx", "samx")
-
-
-def test_waveform_plot_data(waveform_widget, mock_waveform):
-    waveform_widget.plot(x=[1, 2, 3], y=[1, 2, 3])
-    waveform_widget.waveform.plot.assert_called_once_with(
-        arg1=None,
-        x=[1, 2, 3],
-        y=[1, 2, 3],
-        x_name=None,
-        y_name=None,
-        z_name=None,
-        x_entry=None,
-        y_entry=None,
-        z_entry=None,
-        color=None,
-        color_map_z="plasma",
-        label=None,
-        validate=True,
-        dap=None,
-    )
-
-
-def test_waveform_plot_scan_curves(waveform_widget, mock_waveform):
-    waveform_widget.plot(x_name="samx", y_name="samy", dap="GaussianModel")
-    waveform_widget.waveform.plot.assert_called_once_with(
-        arg1=None,
-        x=None,
-        y=None,
-        x_name="samx",
-        y_name="samy",
-        z_name=None,
-        x_entry=None,
-        y_entry=None,
-        z_entry=None,
-        color=None,
-        color_map_z="plasma",
-        label=None,
-        validate=True,
-        dap="GaussianModel",
-    )
-
-
-def test_waveform_widget_add_dap(waveform_widget, mock_waveform):
-    waveform_widget.add_dap(x_name="samx", y_name="bpm4i")
-    waveform_widget.waveform.add_dap.assert_called_once_with(
-        x_name="samx",
-        y_name="bpm4i",
-        x_entry=None,
-        y_entry=None,
-        color=None,
-        dap="GaussianModel",
-        validate_bec=True,
-    )
-
-
-def test_waveform_widget_get_dap_params(waveform_widget, mock_waveform):
-    waveform_widget.get_dap_params()
-    waveform_widget.waveform.get_dap_params.assert_called_once()
-
-
-def test_waveform_widget_get_dap_summary(waveform_widget, mock_waveform):
-    waveform_widget.get_dap_summary()
-    waveform_widget.waveform.get_dap_summary.assert_called_once()
-
-
-def test_waveform_widget_remove_curve(waveform_widget, mock_waveform):
-    waveform_widget.remove_curve("curve_id")
-    waveform_widget.waveform.remove_curve.assert_called_once_with("curve_id")
-
-
-def test_waveform_widget_scan_history(waveform_widget, mock_waveform):
-    waveform_widget.scan_history(0)
-    waveform_widget.waveform.scan_history.assert_called_once_with(0, None)
-
-
-def test_waveform_widget_get_all_data(waveform_widget, mock_waveform):
-    waveform_widget.get_all_data()
-    waveform_widget.waveform.get_all_data.assert_called_once()
-
-
-def test_waveform_widget_set_title(waveform_widget, mock_waveform):
-    waveform_widget.set_title("Title")
-    waveform_widget.waveform.set_title.assert_called_once_with("Title")
-
-
-def test_waveform_widget_set_base(waveform_widget, mock_waveform):
-    waveform_widget.set(
-        title="Test Title",
-        x_label="X Label",
-        y_label="Y Label",
-        x_scale="linear",
-        y_scale="log",
-        x_lim=(0, 10),
-        y_lim=(0, 10),
-        legend_label_size=12,
-    )
-    waveform_widget.waveform.set.assert_called_once_with(
-        title="Test Title",
-        x_label="X Label",
-        y_label="Y Label",
-        x_scale="linear",
-        y_scale="log",
-        x_lim=(0, 10),
-        y_lim=(0, 10),
-        legend_label_size=12,
-    )
-
-
-def test_waveform_widget_set_x_label(waveform_widget, mock_waveform):
-    waveform_widget.set_x_label("X Label")
-    waveform_widget.waveform.set_x_label.assert_called_once_with("X Label")
-
-
-def test_waveform_widget_set_y_label(waveform_widget, mock_waveform):
-    waveform_widget.set_y_label("Y Label")
-    waveform_widget.waveform.set_y_label.assert_called_once_with("Y Label")
-
-
-def test_waveform_widget_set_x_scale(waveform_widget, mock_waveform):
-    waveform_widget.set_x_scale("linear")
-    waveform_widget.waveform.set_x_scale.assert_called_once_with("linear")
-
-
-def test_waveform_widget_set_y_scale(waveform_widget, mock_waveform):
-    waveform_widget.set_y_scale("log")
-    waveform_widget.waveform.set_y_scale.assert_called_once_with("log")
-
-
-def test_waveform_widget_set_x_lim(waveform_widget, mock_waveform):
-    waveform_widget.set_x_lim((0, 10))
-    waveform_widget.waveform.set_x_lim.assert_called_once_with((0, 10))
-
-
-def test_waveform_widget_set_y_lim(waveform_widget, mock_waveform):
-    waveform_widget.set_y_lim((0, 10))
-    waveform_widget.waveform.set_y_lim.assert_called_once_with((0, 10))
-
-
-def test_waveform_widget_set_legend_label_size(waveform_widget, mock_waveform):
-    waveform_widget.set_legend_label_size(12)
-    waveform_widget.waveform.set_legend_label_size.assert_called_once_with(12)
-
-
-def test_waveform_widget_set_auto_range(waveform_widget, mock_waveform):
-    waveform_widget.set_auto_range(True, "xy")
-    waveform_widget.waveform.set_auto_range.assert_called_once_with(True, "xy")
-
-
-def test_waveform_widget_set_grid(waveform_widget, mock_waveform):
-    waveform_widget.set_grid(True, False)
-    waveform_widget.waveform.set_grid.assert_called_once_with(True, False)
-
-
-def test_waveform_widget_lock_aspect_ratio(waveform_widget, mock_waveform):
-    waveform_widget.lock_aspect_ratio(True)
-    waveform_widget.waveform.lock_aspect_ratio.assert_called_once_with(True)
-
-
-def test_waveform_widget_export(waveform_widget, mock_waveform):
-    waveform_widget.export()
-    waveform_widget.waveform.export.assert_called_once()
-
-
-###################################
-# ToolBar interactions
-###################################
-
-
-def test_toolbar_drag_mode_action_triggered(waveform_widget, qtbot):
-    action_drag = waveform_widget.toolbar.widgets["drag_mode"].action
-    action_rectangle = waveform_widget.toolbar.widgets["rectangle_mode"].action
-    action_drag.trigger()
-    assert action_drag.isChecked() == True
-    assert action_rectangle.isChecked() == False
-
-
-def test_toolbar_rectangle_mode_action_triggered(waveform_widget, qtbot):
-    action_drag = waveform_widget.toolbar.widgets["drag_mode"].action
-    action_rectangle = waveform_widget.toolbar.widgets["rectangle_mode"].action
-    action_rectangle.trigger()
-    assert action_drag.isChecked() == False
-    assert action_rectangle.isChecked() == True
-
-
-def test_toolbar_auto_range_action_triggered(waveform_widget, mock_waveform, qtbot):
-    action = waveform_widget.toolbar.widgets["auto_range"].action
-    action.trigger()
-    qtbot.wait(200)
-    waveform_widget.waveform.set_auto_range.assert_called_once_with(True, "xy")
-
-
-def test_toolbar_fit_params_action_triggered(qtbot, waveform_widget):
-    action = waveform_widget.toolbar.widgets["fit_params"].action
-    with patch(
-        "bec_widgets.widgets.waveform.waveform_widget.FitSummaryWidget"
-    ) as MockFitSummaryWidget:
-        mock_dialog_instance = MockFitSummaryWidget.return_value
-        action.trigger()
-        mock_dialog_instance.show.assert_called_once()
-
-
-def test_enable_mouse_pan_mode(qtbot, waveform_widget):
-    action_drag = waveform_widget.toolbar.widgets["drag_mode"].action
-    action_rectangle = waveform_widget.toolbar.widgets["rectangle_mode"].action
-
-    mock_view_box = MagicMock()
-    waveform_widget.waveform.plot_item.getViewBox = MagicMock(return_value=mock_view_box)
-
-    waveform_widget.enable_mouse_pan_mode()
-
-    assert action_drag.isChecked() == True
-    assert action_rectangle.isChecked() == False
-    mock_view_box.setMouseMode.assert_called_once_with(pg.ViewBox.PanMode)
-
-
-###################################
-# Curve Dialog Tests
-###################################
-def show_curve_dialog(qtbot, waveform_widget):
-    curve_dialog = SettingsDialog(
-        waveform_widget,
-        settings_widget=CurveSettings(),
-        window_title="Curve Settings",
-        config=waveform_widget.waveform._curves_data,
-    )
-    qtbot.addWidget(curve_dialog)
-    qtbot.waitExposed(curve_dialog)
-    return curve_dialog
-
-
-def test_curve_dialog_scan_curves_interactions(qtbot, waveform_widget):
-    waveform_widget.plot(y_name="bpm4i")
-    waveform_widget.plot(y_name="bpm3a")
-
-    curve_dialog = show_curve_dialog(qtbot, waveform_widget)
-
-    # Check default display of config from waveform widget
-    assert curve_dialog is not None
-    assert curve_dialog.widget.ui.scan_table.rowCount() == 2
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 0).text() == "bpm4i"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 1).text() == "bpm4i"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(1, 0).text() == "bpm3a"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(1, 1).text() == "bpm3a"
-    assert curve_dialog.widget.ui.x_mode.currentText() == "best_effort"
-    assert curve_dialog.widget.ui.x_name.isEnabled() == False
-    assert curve_dialog.widget.ui.x_entry.isEnabled() == False
-
-    # Add a new curve
-    curve_dialog.widget.ui.add_curve.click()
-    qtbot.wait(200)
-    assert curve_dialog.widget.ui.scan_table.rowCount() == 3
-
-    # Set device to new curve
-    curve_dialog.widget.ui.scan_table.cellWidget(2, 0).setText("bpm3i")
-
-    # Change the x mode to device
-    curve_dialog.widget.ui.x_mode.setCurrentText("device")
-    qtbot.wait(200)
-    assert curve_dialog.widget.ui.x_name.isEnabled() == True
-    assert curve_dialog.widget.ui.x_entry.isEnabled() == True
-
-    # Set the x device
-    curve_dialog.widget.ui.x_name.setText("samx")
-
-    # Delete first curve ('bpm4i')
-    curve_dialog.widget.ui.scan_table.cellWidget(0, 6).click()
-    qtbot.wait(200)
-    assert curve_dialog.widget.ui.scan_table.rowCount() == 2
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 0).text() == "bpm3a"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 1).text() == "bpm3a"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(1, 0).text() == "bpm3i"
-
-    # Close the dialog
-    curve_dialog.accept()
-    qtbot.wait(200)
-
-    # Check the curve data in the target widget
-    assert list(waveform_widget.waveform._curves_data["scan_segment"].keys()) == [
-        "bpm3a-bpm3a",
-        "bpm3i-bpm3i",
-    ]
-    assert len(waveform_widget.curves) == 2
-
-
-def test_curve_dialog_async(qtbot, waveform_widget):
-    waveform_widget.plot(y_name="bpm4i")
-    waveform_widget.plot(y_name="async_device")
-
-    curve_dialog = show_curve_dialog(qtbot, waveform_widget)
-
-    assert curve_dialog is not None
-    assert curve_dialog.widget.ui.scan_table.rowCount() == 2
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 0).text() == "bpm4i"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 1).text() == "bpm4i"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(1, 0).text() == "async_device"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(1, 1).text() == "async_device"
-
-
-def test_curve_dialog_dap(qtbot, waveform_widget):
-    waveform_widget.plot(x_name="samx", y_name="bpm4i", dap="GaussianModel")
-
-    curve_dialog = show_curve_dialog(qtbot, waveform_widget)
-
-    assert curve_dialog is not None
-    assert curve_dialog.widget.ui.scan_table.rowCount() == 1
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 0).text() == "bpm4i"
-    assert curve_dialog.widget.ui.scan_table.cellWidget(0, 1).text() == "bpm4i"
-    assert curve_dialog.widget.ui.dap_table.isEnabled() == True
-    assert curve_dialog.widget.ui.dap_table.rowCount() == 1
-    assert curve_dialog.widget.ui.dap_table.cellWidget(0, 0).text() == "bpm4i"
-    assert curve_dialog.widget.ui.dap_table.cellWidget(0, 1).text() == "bpm4i"
-    assert curve_dialog.widget.ui.x_mode.currentText() == "device"
-    assert curve_dialog.widget.ui.x_name.isEnabled() == True
-    assert curve_dialog.widget.ui.x_entry.isEnabled() == True
-    assert curve_dialog.widget.ui.x_name.text() == "samx"
-    assert curve_dialog.widget.ui.x_entry.text() == "samx"
-
-    curve_dialog.accept()
-    qtbot.wait(200)
-
-    assert list(waveform_widget.waveform._curves_data["scan_segment"].keys()) == ["bpm4i-bpm4i"]
-    assert len(waveform_widget.curves) == 2
-
-
-###################################
-# Axis Dialog Tests
-###################################
-
-
-def show_axis_dialog(qtbot, waveform_widget):
-    axis_dialog = SettingsDialog(
-        waveform_widget,
-        settings_widget=AxisSettings(),
-        window_title="Axis Settings",
-        config=waveform_widget._config_dict["axis"],
-    )
-    qtbot.addWidget(axis_dialog)
-    qtbot.waitExposed(axis_dialog)
-    return axis_dialog
-
-
-def test_axis_dialog_with_axis_limits(qtbot, waveform_widget):
-    waveform_widget.set(
-        title="Test Title",
-        x_label="X Label",
-        y_label="Y Label",
-        x_scale="linear",
-        y_scale="log",
-        x_lim=(0, 10),
-        y_lim=(0, 10),
-    )
-
-    axis_dialog = show_axis_dialog(qtbot, waveform_widget)
-
-    assert axis_dialog is not None
-    assert axis_dialog.widget.ui.plot_title.text() == "Test Title"
-    assert axis_dialog.widget.ui.x_label.text() == "X Label"
-    assert axis_dialog.widget.ui.y_label.text() == "Y Label"
-    assert axis_dialog.widget.ui.x_scale.currentText() == "linear"
-    assert axis_dialog.widget.ui.y_scale.currentText() == "log"
-    assert axis_dialog.widget.ui.x_min.value() == 0
-    assert axis_dialog.widget.ui.x_max.value() == 10
-    assert axis_dialog.widget.ui.y_min.value() == 0
-    assert axis_dialog.widget.ui.y_max.value() == 10
-
-
-def test_axis_dialog_without_axis_limits(qtbot, waveform_widget):
-    waveform_widget.set(
-        title="Test Title", x_label="X Label", y_label="Y Label", x_scale="linear", y_scale="log"
-    )
-    x_range = waveform_widget.fig.widget_list[0].plot_item.viewRange()[0]
-    y_range = waveform_widget.fig.widget_list[0].plot_item.viewRange()[1]
-
-    axis_dialog = show_axis_dialog(qtbot, waveform_widget)
-
-    assert axis_dialog is not None
-    assert axis_dialog.widget.ui.plot_title.text() == "Test Title"
-    assert axis_dialog.widget.ui.x_label.text() == "X Label"
-    assert axis_dialog.widget.ui.y_label.text() == "Y Label"
-    assert axis_dialog.widget.ui.x_scale.currentText() == "linear"
-    assert axis_dialog.widget.ui.y_scale.currentText() == "log"
-    assert axis_dialog.widget.ui.x_min.value() == x_range[0]
-    assert axis_dialog.widget.ui.x_max.value() == x_range[1]
-    assert axis_dialog.widget.ui.y_min.value() == y_range[0]
-    assert axis_dialog.widget.ui.y_max.value() == y_range[1]
-
-
-def test_axis_dialog_set_properties(qtbot, waveform_widget):
-    axis_dialog = show_axis_dialog(qtbot, waveform_widget)
-
-    axis_dialog.widget.ui.plot_title.setText("New Title")
-    axis_dialog.widget.ui.x_label.setText("New X Label")
-    axis_dialog.widget.ui.y_label.setText("New Y Label")
-    axis_dialog.widget.ui.x_scale.setCurrentText("log")
-    axis_dialog.widget.ui.y_scale.setCurrentText("linear")
-    axis_dialog.widget.ui.x_min.setValue(5)
-    axis_dialog.widget.ui.x_max.setValue(15)
-    axis_dialog.widget.ui.y_min.setValue(5)
-    axis_dialog.widget.ui.y_max.setValue(15)
-
-    axis_dialog.accept()
-
-    assert waveform_widget._config_dict["axis"]["title"] == "New Title"
-    assert waveform_widget._config_dict["axis"]["x_label"] == "New X Label"
-    assert waveform_widget._config_dict["axis"]["y_label"] == "New Y Label"
-    assert waveform_widget._config_dict["axis"]["x_scale"] == "log"
-    assert waveform_widget._config_dict["axis"]["y_scale"] == "linear"
-    assert waveform_widget._config_dict["axis"]["x_lim"] == (5, 15)
-    assert waveform_widget._config_dict["axis"]["y_lim"] == (5, 15)

tests/unit_tests/test_website_widget.py

@@ -1,25 +0,0 @@
-import pytest
-from qtpy.QtCore import QUrl
-
-from bec_widgets.widgets.website.website import WebsiteWidget
-
-from .client_mocks import mocked_client
-
-
-@pytest.fixture
-def website_widget(qtbot, mocked_client):
-    widget = WebsiteWidget(client=mocked_client)
-    qtbot.addWidget(widget)
-    qtbot.waitExposed(widget)
-    yield widget
-
-
-def test_website_widget_set_url(website_widget):
-    website_widget.set_url("https://scilog.psi.ch")
-    assert website_widget.url() == QUrl("https://scilog.psi.ch")
-
-    website_widget.set_url(None)
-    assert website_widget.url() == QUrl("https://scilog.psi.ch")
-
-    website_widget.set_url("https://google.com")
-    assert website_widget.get_url() == "https://google.com"

tests/unit_tests/test_widget_io.py

@@ -1,90 +0,0 @@
-# pylint: disable = no-name-in-module,missing-class-docstring, missing-module-docstring
-import pytest
-from qtpy.QtWidgets import QComboBox, QLineEdit, QSpinBox, QTableWidget, QVBoxLayout, QWidget
-
-from bec_widgets.utils.widget_io import WidgetHierarchy
-
-
-@pytest.fixture(scope="function")
-def example_widget(qtbot):
-    # Create a widget with a few child widgets
-    main_widget = QWidget()
-    layout = QVBoxLayout(main_widget)
-    line_edit = QLineEdit(main_widget)
-    combo_box = QComboBox(main_widget)
-    table_widget = QTableWidget(2, 2, main_widget)
-    spin_box = QSpinBox(main_widget)
-    layout.addWidget(line_edit)
-    layout.addWidget(combo_box)
-    layout.addWidget(table_widget)
-    layout.addWidget(spin_box)
-
-    # Add text items to the combo box
-    combo_box.addItems(["Option 1", "Option 2", "Option 3"])
-
-    qtbot.addWidget(main_widget)
-    qtbot.waitExposed(main_widget)
-    yield main_widget
-
-
-def test_export_import_config(example_widget):
-    initial_config = {
-        "QWidget ()": {
-            "QLineEdit ()": {"value": "New Text"},
-            "QComboBox ()": {"value": 1},
-            "QTableWidget ()": {"value": [["a", "b"], ["c", "d"]]},
-            "QSpinBox ()": {"value": 10},
-        }
-    }
-    WidgetHierarchy.import_config_from_dict(example_widget, initial_config, set_values=True)
-
-    exported_config_full = WidgetHierarchy.export_config_to_dict(example_widget, grab_values=True)
-    exported_config_reduced = WidgetHierarchy.export_config_to_dict(
-        example_widget, grab_values=True, save_all=False
-    )
-
-    expected_full = {
-        "QWidget ()": {
-            "QVBoxLayout ()": {},
-            "QLineEdit ()": {"value": "New Text", "QObject ()": {}},
-            "QComboBox ()": {"value": 1, "QStandardItemModel ()": {}},
-            "QTableWidget ()": {
-                "value": [["a", "b"], ["c", "d"]],
-                "QWidget (qt_scrollarea_viewport)": {},
-                "QStyledItemDelegate ()": {},
-                "QHeaderView ()": {
-                    "QWidget (qt_scrollarea_viewport)": {},
-                    "QWidget (qt_scrollarea_hcontainer)": {
-                        "QScrollBar ()": {},
-                        "QBoxLayout ()": {},
-                    },
-                    "QWidget (qt_scrollarea_vcontainer)": {
-                        "QScrollBar ()": {},
-                        "QBoxLayout ()": {},
-                    },
-                    "QItemSelectionModel ()": {},
-                },
-                "QAbstractButton ()": {},
-                "QAbstractTableModel ()": {},
-                "QItemSelectionModel ()": {},
-                "QWidget (qt_scrollarea_hcontainer)": {"QScrollBar ()": {}, "QBoxLayout ()": {}},
-                "QWidget (qt_scrollarea_vcontainer)": {"QScrollBar ()": {}, "QBoxLayout ()": {}},
-            },
-            "QSpinBox ()": {
-                "value": 10,
-                "QLineEdit (qt_spinbox_lineedit)": {"value": "10", "QObject ()": {}},
-                "QValidator (qt_spinboxvalidator)": {},
-            },
-        }
-    }
-    expected_reduced = {
-        "QWidget ()": {
-            "QLineEdit ()": {"value": "New Text"},
-            "QComboBox ()": {"value": 1},
-            "QTableWidget ()": {"value": [["a", "b"], ["c", "d"]]},
-            "QSpinBox ()": {"value": 10, "QLineEdit (qt_spinbox_lineedit)": {"value": "10"}},
-        }
-    }
-
-    assert exported_config_full == expected_full
-    assert exported_config_reduced == expected_reduced