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

tests/unit_tests/test_plot_base.py

@@ -1,95 +0,0 @@
-# pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
-from unittest import mock
-
-import pytest
-from qtpy.QtGui import QFontInfo
-
-from bec_widgets.widgets.figure import BECFigure
-
-from .client_mocks import mocked_client
-from .conftest import create_widget
-
-
-def test_init_plot_base(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    plot_base = bec_figure.add_widget(widget_type="BECPlotBase", widget_id="test_plot")
-    assert plot_base is not None
-    assert plot_base.config.widget_class == "BECPlotBase"
-    assert plot_base.config.gui_id == "test_plot"
-
-
-def test_plot_base_axes_by_separate_methods(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    plot_base = bec_figure.add_widget(widget_type="BECPlotBase", widget_id="test_plot")
-
-    plot_base.set_title("Test Title")
-    plot_base.set_x_label("Test x Label")
-    plot_base.set_y_label("Test y Label")
-    plot_base.set_x_lim(1, 100)
-    plot_base.set_y_lim(5, 500)
-    plot_base.set_grid(True, True)
-    plot_base.set_x_scale("log")
-    plot_base.set_y_scale("log")
-
-    assert plot_base.plot_item.titleLabel.text == "Test Title"
-    assert plot_base.config.axis.title == "Test Title"
-    assert plot_base.plot_item.getAxis("bottom").labelText == "Test x Label"
-    assert plot_base.config.axis.x_label == "Test x Label"
-    assert plot_base.plot_item.getAxis("left").labelText == "Test y Label"
-    assert plot_base.config.axis.y_label == "Test y Label"
-    assert plot_base.config.axis.x_lim == (1, 100)
-    assert plot_base.config.axis.y_lim == (5, 500)
-    assert plot_base.plot_item.ctrl.xGridCheck.isChecked() == True
-    assert plot_base.plot_item.ctrl.yGridCheck.isChecked() == True
-    assert plot_base.plot_item.ctrl.logXCheck.isChecked() == True
-    assert plot_base.plot_item.ctrl.logYCheck.isChecked() == True
-
-    # Check the font size by mocking the set functions
-    # I struggled retrieving it from the QFont object directly
-    # thus I mocked the set functions to check internally the functionality
-    with (
-        mock.patch.object(plot_base.plot_item, "setLabel") as mock_set_label,
-        mock.patch.object(plot_base.plot_item, "setTitle") as mock_set_title,
-    ):
-        plot_base.set_x_label("Test x Label", 20)
-        plot_base.set_y_label("Test y Label", 16)
-        assert mock_set_label.call_count == 2
-        assert plot_base.config.axis.x_label_size == 20
-        assert plot_base.config.axis.y_label_size == 16
-        col = plot_base.get_text_color()
-        calls = []
-        style = {"color": col, "font-size": "20pt"}
-        calls.append(mock.call("bottom", "Test x Label", **style))
-        style = {"color": col, "font-size": "16pt"}
-        calls.append(mock.call("left", "Test y Label", **style))
-        assert mock_set_label.call_args_list == calls
-        plot_base.set_title("Test Title", 16)
-        style = {"color": col, "size": "16pt"}
-        call = mock.call("Test Title", **style)
-        assert mock_set_title.call_args == call
-
-
-def test_plot_base_axes_added_by_kwargs(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    plot_base = bec_figure.add_widget(widget_type="BECPlotBase", widget_id="test_plot")
-
-    plot_base.set(
-        title="Test Title",
-        x_label="Test x Label",
-        y_label="Test y Label",
-        x_lim=(1, 100),
-        y_lim=(5, 500),
-        x_scale="log",
-        y_scale="log",
-    )
-
-    assert plot_base.plot_item.titleLabel.text == "Test Title"
-    assert plot_base.config.axis.title == "Test Title"
-    assert plot_base.plot_item.getAxis("bottom").labelText == "Test x Label"
-    assert plot_base.config.axis.x_label == "Test x Label"
-    assert plot_base.plot_item.getAxis("left").labelText == "Test y Label"
-    assert plot_base.config.axis.y_label == "Test y Label"
-    assert plot_base.config.axis.x_lim == (1, 100)
-    assert plot_base.config.axis.y_lim == (5, 500)
-    assert plot_base.plot_item.ctrl.logXCheck.isChecked() == True
-    assert plot_base.plot_item.ctrl.logYCheck.isChecked() == True

tests/unit_tests/test_plugin_utils.py

@@ -1,13 +0,0 @@
-from bec_widgets.utils.plugin_utils import get_rpc_classes
-
-
-def test_client_generator_classes():
-    out = get_rpc_classes("bec_widgets")
-    connector_cls_names = [cls.__name__ for cls in out.connector_classes]
-    top_level_cls_names = [cls.__name__ for cls in out.top_level_classes]
-
-    assert "BECFigure" in connector_cls_names
-    assert "BECWaveform" in connector_cls_names
-    assert "BECDockArea" in top_level_cls_names
-    assert "BECFigure" in top_level_cls_names
-    assert "BECWaveform" not in top_level_cls_names

tests/unit_tests/test_positioner_box.py

@@ -1,130 +0,0 @@
-from unittest import mock
-
-import pytest
-from bec_lib.device import Positioner
-from bec_lib.endpoints import MessageEndpoints
-from bec_lib.messages import ScanQueueMessage
-from qtpy.QtGui import QValidator
-
-from bec_widgets.widgets.positioner_box.positioner_box import PositionerBox
-from bec_widgets.widgets.positioner_box.positioner_control_line import PositionerControlLine
-
-from .client_mocks import mocked_client
-
-
-@pytest.fixture
-def positioner_box(qtbot, mocked_client):
-    """Fixture for PositionerBox widget"""
-    with mock.patch("bec_widgets.widgets.positioner_box.positioner_box.uuid.uuid4") as mock_uuid:
-        mock_uuid.return_value = "fake_uuid"
-        with mock.patch(
-            "bec_widgets.widgets.positioner_box.positioner_box.PositionerBox._check_device_is_valid",
-            return_value=True,
-        ):
-            db = PositionerBox(device="samx", client=mocked_client)
-            qtbot.addWidget(db)
-            yield db
-
-
-def test_positioner_box(positioner_box):
-    """Test init of positioner box"""
-    assert positioner_box.device == "samx"
-    data = positioner_box.dev["samx"].read()
-    # Avoid check for Positioner class from BEC in _init_device
-
-    setpoint_text = positioner_box.ui.setpoint.text()
-    # check that the setpoint is taken correctly after init
-    assert float(setpoint_text) == data["samx_setpoint"]["value"]
-
-    # check that the precision is taken correctly after isnit
-    precision = positioner_box.dev["samx"].precision
-    assert setpoint_text == f"{data['samx_setpoint']['value']:.{precision}f}"
-
-    # check that the step size is set according to the device precision
-    assert positioner_box.ui.step_size.value() == 10**-precision * 10
-
-
-def test_positioner_box_update_limits(positioner_box):
-    """Test update of limits"""
-    positioner_box._limits = None
-    positioner_box.update_limits([0, 10])
-    assert positioner_box._limits == [0, 10]
-    assert positioner_box.setpoint_validator.bottom() == 0
-    assert positioner_box.setpoint_validator.top() == 10
-    assert positioner_box.setpoint_validator.validate("100", 0) == (
-        QValidator.State.Intermediate,
-        "100",
-        0,
-    )
-
-    positioner_box.update_limits(None)
-    assert positioner_box._limits is None
-    assert positioner_box.setpoint_validator.validate("100", 0) == (
-        QValidator.State.Acceptable,
-        "100",
-        0,
-    )
-
-
-def test_positioner_box_on_stop(positioner_box):
-    """Test on stop button"""
-    with mock.patch.object(positioner_box.client.connector, "send") as mock_send:
-        positioner_box.on_stop()
-        params = {"device": "samx", "rpc_id": "fake_uuid", "func": "stop", "args": [], "kwargs": {}}
-        msg = ScanQueueMessage(
-            scan_type="device_rpc",
-            parameter=params,
-            queue="emergency",
-            metadata={"RID": "fake_uuid", "response": False},
-        )
-        mock_send.assert_called_once_with(MessageEndpoints.scan_queue_request(), msg)
-
-
-def test_positioner_box_setpoint_change(positioner_box):
-    """Test positioner box setpoint change"""
-    with mock.patch.object(positioner_box.dev["samx"], "move") as mock_move:
-        positioner_box.ui.setpoint.setText("100")
-        positioner_box.on_setpoint_change()
-        mock_move.assert_called_once_with(100, relative=False)
-
-
-def test_positioner_box_on_tweak_right(positioner_box):
-    """Test tweak right button"""
-    with mock.patch.object(positioner_box.dev["samx"], "move") as mock_move:
-        positioner_box.ui.step_size.setValue(0.1)
-        positioner_box.on_tweak_right()
-        mock_move.assert_called_once_with(0.1, relative=True)
-
-
-def test_positioner_box_on_tweak_left(positioner_box):
-    """Test tweak left button"""
-    with mock.patch.object(positioner_box.dev["samx"], "move") as mock_move:
-        positioner_box.ui.step_size.setValue(0.1)
-        positioner_box.on_tweak_left()
-        mock_move.assert_called_once_with(-0.1, relative=True)
-
-
-def test_positioner_box_setpoint_out_of_range(positioner_box):
-    """Test setpoint out of range"""
-    positioner_box.update_limits([0, 10])
-    positioner_box.ui.setpoint.setText("100")
-    positioner_box.on_setpoint_change()
-    assert positioner_box.ui.setpoint.text() == "100"
-    assert positioner_box.ui.setpoint.hasAcceptableInput() == False
-
-
-def test_positioner_control_line(qtbot, mocked_client):
-    """Test PositionerControlLine.
-    Inherits from PositionerBox, but the layout is changed. Check dimensions only
-    """
-    with mock.patch("bec_widgets.widgets.positioner_box.positioner_box.uuid.uuid4") as mock_uuid:
-        mock_uuid.return_value = "fake_uuid"
-        with mock.patch(
-            "bec_widgets.widgets.positioner_box.positioner_box.PositionerBox._check_device_is_valid",
-            return_value=True,
-        ):
-            db = PositionerControlLine(device="samx", client=mocked_client)
-            qtbot.addWidget(db)
-
-            assert db.ui.device_box.height() == 60
-            assert db.ui.device_box.width() == 600

tests/unit_tests/test_ring_progress_bar.py

@@ -1,337 +0,0 @@
-# pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
-
-import pytest
-from bec_lib.endpoints import MessageEndpoints
-from pydantic import ValidationError
-
-from bec_widgets.utils import Colors
-from bec_widgets.widgets.ring_progress_bar import RingProgressBar
-from bec_widgets.widgets.ring_progress_bar.ring import ProgressbarConnections, RingConfig
-from bec_widgets.widgets.ring_progress_bar.ring_progress_bar import RingProgressBarConfig
-
-from .client_mocks import mocked_client
-
-
-@pytest.fixture
-def ring_progress_bar(qtbot, mocked_client):
-    widget = RingProgressBar(client=mocked_client)
-    qtbot.addWidget(widget)
-    qtbot.waitExposed(widget)
-    yield widget
-
-
-def test_bar_init(ring_progress_bar):
-    assert ring_progress_bar is not None
-    assert ring_progress_bar.client is not None
-    assert isinstance(ring_progress_bar, RingProgressBar)
-    assert ring_progress_bar.config.widget_class == "RingProgressBar"
-    assert ring_progress_bar.config.gui_id is not None
-    assert ring_progress_bar.gui_id == ring_progress_bar.config.gui_id
-
-
-def test_config_validation_num_of_bars():
-    config = RingProgressBarConfig(num_bars=100, min_num_bars=1, max_num_bars=10)
-
-    assert config.num_bars == 10
-
-
-def test_config_validation_num_of_ring_error():
-    ring_config_0 = RingConfig(index=0)
-    ring_config_1 = RingConfig(index=1)
-
-    with pytest.raises(ValidationError) as excinfo:
-        RingProgressBarConfig(rings=[ring_config_0, ring_config_1], num_bars=1)
-    errors = excinfo.value.errors()
-    assert len(errors) == 1
-    assert errors[0]["type"] == "different number of configs"
-    assert "Length of rings configuration (2) does not match the number of bars (1)." in str(
-        excinfo.value
-    )
-
-
-def test_config_validation_ring_indices_wrong_order():
-    ring_config_0 = RingConfig(index=2)
-    ring_config_1 = RingConfig(index=5)
-
-    with pytest.raises(ValidationError) as excinfo:
-        RingProgressBarConfig(rings=[ring_config_0, ring_config_1], num_bars=2)
-    errors = excinfo.value.errors()
-    assert len(errors) == 1
-    assert errors[0]["type"] == "wrong indices"
-    assert (
-        "Indices of ring configurations must be unique and in order from 0 to num_bars 2."
-        in str(excinfo.value)
-    )
-
-
-def test_config_validation_ring_same_indices():
-    ring_config_0 = RingConfig(index=0)
-    ring_config_1 = RingConfig(index=0)
-
-    with pytest.raises(ValidationError) as excinfo:
-        RingProgressBarConfig(rings=[ring_config_0, ring_config_1], num_bars=2)
-    errors = excinfo.value.errors()
-    assert len(errors) == 1
-    assert errors[0]["type"] == "wrong indices"
-    assert (
-        "Indices of ring configurations must be unique and in order from 0 to num_bars 2."
-        in str(excinfo.value)
-    )
-
-
-def test_config_validation_invalid_colormap():
-    with pytest.raises(ValueError) as excinfo:
-        RingProgressBarConfig(color_map="crazy_colors")
-    errors = excinfo.value.errors()
-    assert len(errors) == 1
-    assert errors[0]["type"] == "unsupported colormap"
-    assert "Colormap 'crazy_colors' not found in the current installation of pyqtgraph" in str(
-        excinfo.value
-    )
-
-
-def test_ring_connection_endpoint_validation():
-    with pytest.raises(ValueError) as excinfo:
-        ProgressbarConnections(slot="on_scan_progress", endpoint="non_existing")
-    errors = excinfo.value.errors()
-    assert len(errors) == 1
-    assert errors[0]["type"] == "unsupported endpoint"
-    assert (
-        "For slot 'on_scan_progress', endpoint must be MessageEndpoint.scan_progress or 'scans/scan_progress'."
-        in str(excinfo.value)
-    )
-
-    with pytest.raises(ValueError) as excinfo:
-        ProgressbarConnections(slot="on_device_readback", endpoint="non_existing")
-    errors = excinfo.value.errors()
-    assert len(errors) == 1
-    assert errors[0]["type"] == "unsupported endpoint"
-    assert (
-        "For slot 'on_device_readback', endpoint must be MessageEndpoint.device_readback(device) or 'internal/devices/readback/{device}'."
-        in str(excinfo.value)
-    )
-
-
-def test_bar_add_number_of_bars(ring_progress_bar):
-    assert ring_progress_bar.config.num_bars == 1
-
-    ring_progress_bar.set_number_of_bars(5)
-    assert ring_progress_bar.config.num_bars == 5
-
-    ring_progress_bar.set_number_of_bars(2)
-    assert ring_progress_bar.config.num_bars == 2
-
-
-def test_add_remove_bars_individually(ring_progress_bar):
-    ring_progress_bar.add_ring()
-    ring_progress_bar.add_ring()
-
-    assert ring_progress_bar.config.num_bars == 3
-    assert len(ring_progress_bar.config.rings) == 3
-
-    ring_progress_bar.remove_ring(1)
-    assert ring_progress_bar.config.num_bars == 2
-    assert len(ring_progress_bar.config.rings) == 2
-    assert ring_progress_bar.rings[0].config.index == 0
-    assert ring_progress_bar.rings[1].config.index == 1
-
-
-def test_bar_set_value(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(5)
-
-    assert ring_progress_bar.config.num_bars == 5
-    assert len(ring_progress_bar.config.rings) == 5
-    assert len(ring_progress_bar.rings) == 5
-
-    ring_progress_bar.set_value([10, 20, 30, 40, 50])
-    ring_values = [ring.config.value for ring in ring_progress_bar.rings]
-    assert ring_values == [10, 20, 30, 40, 50]
-
-    # update just one bar
-    ring_progress_bar.set_value(90, 1)
-    ring_values = [ring.config.value for ring in ring_progress_bar.rings]
-    assert ring_values == [10, 90, 30, 40, 50]
-
-
-def test_bar_set_precision(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(3)
-
-    assert ring_progress_bar.config.num_bars == 3
-    assert len(ring_progress_bar.config.rings) == 3
-    assert len(ring_progress_bar.rings) == 3
-
-    ring_progress_bar.set_precision(2)
-    ring_precision = [ring.config.precision for ring in ring_progress_bar.rings]
-    assert ring_precision == [2, 2, 2]
-
-    ring_progress_bar.set_value([10.1234, 20.1234, 30.1234])
-    ring_values = [ring.config.value for ring in ring_progress_bar.rings]
-    assert ring_values == [10.12, 20.12, 30.12]
-
-    ring_progress_bar.set_precision(4, 1)
-    ring_precision = [ring.config.precision for ring in ring_progress_bar.rings]
-    assert ring_precision == [2, 4, 2]
-
-    ring_progress_bar.set_value([10.1234, 20.1234, 30.1234])
-    ring_values = [ring.config.value for ring in ring_progress_bar.rings]
-    assert ring_values == [10.12, 20.1234, 30.12]
-
-
-def test_set_min_max_value(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(2)
-
-    ring_progress_bar.set_min_max_values(0, 10)
-    ring_min_values = [ring.config.min_value for ring in ring_progress_bar.rings]
-    ring_max_values = [ring.config.max_value for ring in ring_progress_bar.rings]
-
-    assert ring_min_values == [0, 0]
-    assert ring_max_values == [10, 10]
-
-    ring_progress_bar.set_value([5, 15])
-    ring_values = [ring.config.value for ring in ring_progress_bar.rings]
-    assert ring_values == [5, 10]
-
-
-def test_setup_colors_from_colormap(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(5)
-    ring_progress_bar.set_colors_from_map("viridis", "RGB")
-
-    expected_colors = Colors.golden_angle_color("viridis", 5, "RGB")
-    converted_colors = [ring.color.getRgb() for ring in ring_progress_bar.rings]
-    ring_config_colors = [ring.config.color for ring in ring_progress_bar.rings]
-
-    assert expected_colors == converted_colors
-    assert ring_config_colors == expected_colors
-
-
-def get_colors_from_rings(rings):
-    converted_colors = [ring.color.getRgb() for ring in rings]
-    ring_config_colors = [ring.config.color for ring in rings]
-    return converted_colors, ring_config_colors
-
-
-def test_set_colors_from_colormap_and_change_num_of_bars(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(2)
-    ring_progress_bar.set_colors_from_map("viridis", "RGB")
-
-    expected_colors = Colors.golden_angle_color("viridis", 2, "RGB")
-    converted_colors, ring_config_colors = get_colors_from_rings(ring_progress_bar.rings)
-
-    assert expected_colors == converted_colors
-    assert ring_config_colors == expected_colors
-
-    # increase the number of bars to 6
-    ring_progress_bar.set_number_of_bars(6)
-    expected_colors = Colors.golden_angle_color("viridis", 6, "RGB")
-    converted_colors, ring_config_colors = get_colors_from_rings(ring_progress_bar.rings)
-
-    assert expected_colors == converted_colors
-    assert ring_config_colors == expected_colors
-
-    # decrease the number of bars to 3
-    ring_progress_bar.set_number_of_bars(3)
-    expected_colors = Colors.golden_angle_color("viridis", 3, "RGB")
-    converted_colors, ring_config_colors = get_colors_from_rings(ring_progress_bar.rings)
-
-    assert expected_colors == converted_colors
-    assert ring_config_colors == expected_colors
-
-
-def test_set_colors_directly(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(3)
-
-    # setting as a list of rgb tuples
-    colors = [(255, 0, 0, 255), (0, 255, 0, 255), (0, 0, 255, 255)]
-    ring_progress_bar.set_colors_directly(colors)
-    converted_colors = get_colors_from_rings(ring_progress_bar.rings)[0]
-
-    assert colors == converted_colors
-
-    ring_progress_bar.set_colors_directly((255, 0, 0, 255), 1)
-    converted_colors = get_colors_from_rings(ring_progress_bar.rings)[0]
-
-    assert converted_colors == [(255, 0, 0, 255), (255, 0, 0, 255), (0, 0, 255, 255)]
-
-
-def test_set_line_width(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(3)
-
-    ring_progress_bar.set_line_widths(5)
-    line_widths = [ring.config.line_width for ring in ring_progress_bar.rings]
-
-    assert line_widths == [5, 5, 5]
-
-    ring_progress_bar.set_line_widths([10, 20, 30])
-    line_widths = [ring.config.line_width for ring in ring_progress_bar.rings]
-
-    assert line_widths == [10, 20, 30]
-
-    ring_progress_bar.set_line_widths(15, 1)
-    line_widths = [ring.config.line_width for ring in ring_progress_bar.rings]
-
-    assert line_widths == [10, 15, 30]
-
-
-def test_set_gap(ring_progress_bar):
-    ring_progress_bar.set_number_of_bars(3)
-    ring_progress_bar.set_gap(20)
-
-    assert ring_progress_bar.config.gap == 20
-
-
-def test_auto_update(ring_progress_bar):
-    ring_progress_bar.enable_auto_updates(True)
-
-    scan_queue_status_scan_progress = {
-        "queue": {
-            "primary": {
-                "info": [{"active_request_block": {"report_instructions": [{"scan_progress": 10}]}}]
-            }
-        }
-    }
-    meta = {}
-
-    ring_progress_bar.on_scan_queue_status(scan_queue_status_scan_progress, meta)
-
-    assert ring_progress_bar._auto_updates is True
-    assert len(ring_progress_bar._rings) == 1
-    assert ring_progress_bar._rings[0].config.connections == ProgressbarConnections(
-        slot="on_scan_progress", endpoint=MessageEndpoints.scan_progress()
-    )
-
-    scan_queue_status_device_readback = {
-        "queue": {
-            "primary": {
-                "info": [
-                    {
-                        "active_request_block": {
-                            "report_instructions": [
-                                {
-                                    "readback": {
-                                        "devices": ["samx", "samy"],
-                                        "start": [1, 2],
-                                        "end": [10, 20],
-                                    }
-                                }
-                            ]
-                        }
-                    }
-                ]
-            }
-        }
-    }
-    ring_progress_bar.on_scan_queue_status(scan_queue_status_device_readback, meta)
-
-    assert ring_progress_bar._auto_updates is True
-    assert len(ring_progress_bar._rings) == 2
-    assert ring_progress_bar._rings[0].config.connections == ProgressbarConnections(
-        slot="on_device_readback", endpoint=MessageEndpoints.device_readback("samx")
-    )
-    assert ring_progress_bar._rings[1].config.connections == ProgressbarConnections(
-        slot="on_device_readback", endpoint=MessageEndpoints.device_readback("samy")
-    )
-
-    assert ring_progress_bar._rings[0].config.min_value == 1
-    assert ring_progress_bar._rings[0].config.max_value == 10
-    assert ring_progress_bar._rings[1].config.min_value == 2
-    assert ring_progress_bar._rings[1].config.max_value == 20

tests/unit_tests/test_rpc_register.py

@@ -1,52 +0,0 @@
-from bec_widgets.cli.rpc_register import RPCRegister
-
-
-class FakeObject:
-    def __init__(self, gui_id):
-        self.gui_id = gui_id
-
-
-def test_add_connection(rpc_register):
-    obj1 = FakeObject("id1")
-    obj2 = FakeObject("id2")
-
-    rpc_register.add_rpc(obj1)
-    rpc_register.add_rpc(obj2)
-
-    all_connections = rpc_register.list_all_connections()
-
-    assert len(all_connections) == 2
-    assert all_connections["id1"] == obj1
-    assert all_connections["id2"] == obj2
-
-
-def test_remove_connection(rpc_register):
-
-    obj1 = FakeObject("id1")
-    obj2 = FakeObject("id2")
-
-    rpc_register.add_rpc(obj1)
-    rpc_register.add_rpc(obj2)
-
-    rpc_register.remove_rpc(obj1)
-
-    all_connections = rpc_register.list_all_connections()
-
-    assert len(all_connections) == 1
-    assert all_connections["id2"] == obj2
-
-
-def test_reset_singleton(rpc_register):
-    obj1 = FakeObject("id1")
-    obj2 = FakeObject("id2")
-
-    rpc_register.add_rpc(obj1)
-    rpc_register.add_rpc(obj2)
-
-    rpc_register.reset_singleton()
-    rpc_register = RPCRegister()
-
-    all_connections = rpc_register.list_all_connections()
-
-    assert len(all_connections) == 0
-    assert all_connections == {}

tests/unit_tests/test_rpc_server.py

@@ -1,42 +0,0 @@
-from unittest import mock
-
-import pytest
-from bec_lib.service_config import ServiceConfig
-
-from bec_widgets.cli.server import _start_server
-from bec_widgets.widgets.figure import BECFigure
-
-
-@pytest.fixture
-def mocked_cli_server():
-    with mock.patch("bec_widgets.cli.server.BECWidgetsCLIServer") as mock_server:
-        with mock.patch("bec_widgets.cli.server.ServiceConfig") as mock_config:
-            with mock.patch("bec_lib.logger.bec_logger.configure") as mock_logger:
-                yield mock_server, mock_config, mock_logger
-
-
-def test_rpc_server_start_server_without_service_config(mocked_cli_server):
-    """
-    Test that the server is started with the correct arguments.
-    """
-    mock_server, mock_config, _ = mocked_cli_server
-
-    _start_server("gui_id", BECFigure, None)
-    mock_server.assert_called_once_with(gui_id="gui_id", config=mock_config(), gui_class=BECFigure)
-
-
-@pytest.mark.parametrize(
-    "config, call_config",
-    [
-        ("/path/to/config.yml", {"config_path": "/path/to/config.yml"}),
-        ({"key": "value"}, {"config": {"key": "value"}}),
-    ],
-)
-def test_rpc_server_start_server_with_service_config(mocked_cli_server, config, call_config):
-    """
-    Test that the server is started with the correct arguments.
-    """
-    mock_server, mock_config, _ = mocked_cli_server
-    config = mock_config(**call_config)
-    _start_server("gui_id", BECFigure, config)
-    mock_server.assert_called_once_with(gui_id="gui_id", config=config, gui_class=BECFigure)

tests/unit_tests/test_rpc_widget_handler.py

@@ -1,7 +0,0 @@
-from bec_widgets.cli.rpc_wigdet_handler import RPCWidgetHandler
-
-
-def test_rpc_widget_handler():
-    handler = RPCWidgetHandler()
-    assert "BECFigure" in handler.widget_classes
-    assert "RingProgressBar" in handler.widget_classes