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

tests/unit_tests/test_bec_dispatcher.py

@@ -1,119 +0,0 @@
-# pylint: disable = no-name-in-module,missing-class-docstring, missing-module-docstring
-import threading
-import time
-from unittest import mock
-
-import pytest
-import redis
-from bec_lib.connector import MessageObject
-from bec_lib.messages import ScanMessage
-from bec_lib.redis_connector import RedisConnector
-from bec_lib.serialization import MsgpackSerialization
-
-from bec_widgets.utils.bec_dispatcher import QtRedisConnector
-
-
-@pytest.fixture
-def bec_dispatcher_w_connector(bec_dispatcher, topics_msg_list, send_msg_event):
-    def pubsub_msg_generator():
-        send_msg_event.wait()
-        for topic, msg in topics_msg_list:
-            yield {"channel": topic.encode(), "pattern": None, "data": msg}
-        while True:
-            time.sleep(0.2)
-            yield StopIteration
-
-    with mock.patch("redis.Redis"):
-        pubsub = redis.Redis().pubsub()
-        messages = pubsub_msg_generator()
-        pubsub.get_message.side_effect = lambda timeout: next(messages)
-        connector = QtRedisConnector("localhost:1")
-        bec_dispatcher.client.connector = connector
-        yield bec_dispatcher
-
-
-dummy_msg = MsgpackSerialization.dumps(ScanMessage(point_id=0, scan_id="0", data={}))
-
-
-@pytest.fixture
-def send_msg_event():
-    return threading.Event()
-
-
-@pytest.mark.parametrize(
-    "topics_msg_list", [(("topic1", dummy_msg), ("topic2", dummy_msg), ("topic3", dummy_msg))]
-)
-def test_dispatcher_disconnect_all(bec_dispatcher_w_connector, qtbot, send_msg_event):
-    bec_dispatcher = bec_dispatcher_w_connector
-    cb1 = mock.Mock(spec=[])
-    cb2 = mock.Mock(spec=[])
-
-    bec_dispatcher.connect_slot(cb1, "topic1")
-    bec_dispatcher.connect_slot(cb1, "topic2")
-    bec_dispatcher.connect_slot(cb2, "topic2")
-    bec_dispatcher.connect_slot(cb2, "topic3")
-    assert len(bec_dispatcher.client.connector._topics_cb) == 3
-    send_msg_event.set()
-    qtbot.wait(10)
-    assert cb1.call_count == 2
-    assert cb2.call_count == 2
-
-    bec_dispatcher.disconnect_all()
-
-    assert len(bec_dispatcher.client.connector._topics_cb) == 0
-
-
-@pytest.mark.parametrize("topics_msg_list", [(("topic1", dummy_msg), ("topic2", dummy_msg))])
-def test_dispatcher_disconnect_one(bec_dispatcher_w_connector, qtbot, send_msg_event):
-    # test for BEC issue #276
-    bec_dispatcher = bec_dispatcher_w_connector
-    cb1 = mock.Mock(spec=[])
-    cb2 = mock.Mock(spec=[])
-
-    bec_dispatcher.connect_slot(cb1, "topic1")
-    bec_dispatcher.connect_slot(cb2, "topic2")
-    assert len(bec_dispatcher.client.connector._topics_cb) == 2
-    bec_dispatcher.disconnect_slot(cb1, "topic1")
-    assert len(bec_dispatcher.client.connector._topics_cb) == 1
-
-    send_msg_event.set()
-    qtbot.wait(10)
-    assert cb1.call_count == 0
-    cb2.assert_called_once()
-
-
-@pytest.mark.parametrize("topics_msg_list", [(("topic1", dummy_msg),)])
-def test_dispatcher_2_cb_same_topic(bec_dispatcher_w_connector, qtbot, send_msg_event):
-    # test for BEC issue #276
-    bec_dispatcher = bec_dispatcher_w_connector
-    cb1 = mock.Mock(spec=[])
-    cb2 = mock.Mock(spec=[])
-
-    bec_dispatcher.connect_slot(cb1, "topic1")
-    bec_dispatcher.connect_slot(cb2, "topic1")
-    assert len(bec_dispatcher.client.connector._topics_cb) == 1
-    assert len(bec_dispatcher._slots) == 2
-    bec_dispatcher.disconnect_slot(cb1, "topic1")
-    assert len(bec_dispatcher._slots) == 1
-
-    send_msg_event.set()
-    qtbot.wait(10)
-    assert cb1.call_count == 0
-    cb2.assert_called_once()
-
-
-@pytest.mark.parametrize("topics_msg_list", [(("topic1", dummy_msg), ("topic2", dummy_msg))])
-def test_dispatcher_2_topic_same_cb(bec_dispatcher_w_connector, qtbot, send_msg_event):
-    # test for BEC issue #276
-    bec_dispatcher = bec_dispatcher_w_connector
-    cb1 = mock.Mock(spec=[])
-
-    bec_dispatcher.connect_slot(cb1, "topic1")
-    bec_dispatcher.connect_slot(cb1, "topic2")
-    assert len(bec_dispatcher.client.connector._topics_cb) == 2
-    bec_dispatcher.disconnect_slot(cb1, "topic1")
-    assert len(bec_dispatcher.client.connector._topics_cb) == 1
-
-    send_msg_event.set()
-    qtbot.wait(10)
-    cb1.assert_called_once()

tests/unit_tests/test_bec_dock.py

@@ -1,155 +0,0 @@
-# pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
-from unittest.mock import MagicMock, patch
-
-import pytest
-
-from bec_widgets.widgets.dock import BECDock, BECDockArea
-
-from .client_mocks import mocked_client
-
-
-@pytest.fixture
-def bec_dock_area(qtbot, mocked_client):
-    widget = BECDockArea(client=mocked_client)
-    qtbot.addWidget(widget)
-    qtbot.waitExposed(widget)
-    yield widget
-
-
-def test_bec_dock_area_init(bec_dock_area):
-    assert bec_dock_area is not None
-    assert bec_dock_area.client is not None
-    assert isinstance(bec_dock_area, BECDockArea)
-    assert bec_dock_area.config.widget_class == "BECDockArea"
-
-
-def test_bec_dock_area_add_remove_dock(bec_dock_area, qtbot):
-    initial_count = len(bec_dock_area.dock_area.docks)
-
-    # Adding 3 docks
-    d0 = bec_dock_area.add_dock()
-    d1 = bec_dock_area.add_dock()
-    d2 = bec_dock_area.add_dock()
-
-    # Check if the docks were added
-    assert len(bec_dock_area.dock_area.docks) == initial_count + 3
-    assert d0.name() in dict(bec_dock_area.dock_area.docks)
-    assert d1.name() in dict(bec_dock_area.dock_area.docks)
-    assert d2.name() in dict(bec_dock_area.dock_area.docks)
-    assert bec_dock_area.dock_area.docks[d0.name()].config.widget_class == "BECDock"
-    assert bec_dock_area.dock_area.docks[d1.name()].config.widget_class == "BECDock"
-    assert bec_dock_area.dock_area.docks[d2.name()].config.widget_class == "BECDock"
-
-    # Check panels API for getting docks to CLI
-    assert bec_dock_area.panels == dict(bec_dock_area.dock_area.docks)
-
-    # Remove docks
-    d0_name = d0.name()
-    bec_dock_area.remove_dock(d0_name)
-    qtbot.wait(200)
-    d1.remove()
-    qtbot.wait(200)
-
-    assert len(bec_dock_area.dock_area.docks) == initial_count + 1
-    assert d0.name() not in dict(bec_dock_area.dock_area.docks)
-    assert d1.name() not in dict(bec_dock_area.dock_area.docks)
-    assert d2.name() in dict(bec_dock_area.dock_area.docks)
-
-
-def test_add_remove_bec_figure_to_dock(bec_dock_area):
-    d0 = bec_dock_area.add_dock()
-    fig = d0.add_widget("BECFigure")
-    plt = fig.plot(x_name="samx", y_name="bpm4i")
-    im = fig.image("eiger")
-    mm = fig.motor_map("samx", "samy")
-
-    assert len(bec_dock_area.dock_area.docks) == 1
-    assert len(d0.widgets) == 1
-    assert len(d0.widget_list) == 1
-    assert len(fig.widgets) == 3
-
-    assert fig.config.widget_class == "BECFigure"
-    assert plt.config.widget_class == "BECWaveform"
-    assert im.config.widget_class == "BECImageShow"
-    assert mm.config.widget_class == "BECMotorMap"
-
-
-def test_close_docks(bec_dock_area, qtbot):
-    d0 = bec_dock_area.add_dock(name="dock_0")
-    d1 = bec_dock_area.add_dock(name="dock_1")
-    d2 = bec_dock_area.add_dock(name="dock_2")
-
-    bec_dock_area.clear_all()
-    qtbot.wait(200)
-    assert len(bec_dock_area.dock_area.docks) == 0
-
-
-def test_undock_and_dock_docks(bec_dock_area, qtbot):
-    d0 = bec_dock_area.add_dock(name="dock_0")
-    d1 = bec_dock_area.add_dock(name="dock_1")
-    d2 = bec_dock_area.add_dock(name="dock_4")
-    d3 = bec_dock_area.add_dock(name="dock_3")
-
-    d0.detach()
-    bec_dock_area.detach_dock("dock_1")
-    d2.detach()
-
-    assert len(bec_dock_area.dock_area.docks) == 4
-    assert len(bec_dock_area.dock_area.tempAreas) == 3
-
-    d0.attach()
-    assert len(bec_dock_area.dock_area.docks) == 4
-    assert len(bec_dock_area.dock_area.tempAreas) == 2
-
-    bec_dock_area.attach_all()
-    assert len(bec_dock_area.dock_area.docks) == 4
-    assert len(bec_dock_area.dock_area.tempAreas) == 0
-
-
-###################################
-# Toolbar Actions
-###################################
-def test_toolbar_add_plot_waveform(bec_dock_area):
-    bec_dock_area.toolbar.widgets["menu_plots"].widgets["waveform"].trigger()
-    assert "waveform_1" in bec_dock_area.panels
-    assert bec_dock_area.panels["waveform_1"].widgets[0].config.widget_class == "BECWaveformWidget"
-
-
-def test_toolbar_add_plot_image(bec_dock_area):
-    bec_dock_area.toolbar.widgets["menu_plots"].widgets["image"].trigger()
-    assert "image_1" in bec_dock_area.panels
-    assert bec_dock_area.panels["image_1"].widgets[0].config.widget_class == "BECImageWidget"
-
-
-def test_toolbar_add_plot_motor_map(bec_dock_area):
-    bec_dock_area.toolbar.widgets["menu_plots"].widgets["motor_map"].trigger()
-    assert "motor_map_1" in bec_dock_area.panels
-    assert bec_dock_area.panels["motor_map_1"].widgets[0].config.widget_class == "BECMotorMapWidget"
-
-
-def test_toolbar_add_device_positioner_box(bec_dock_area):
-    bec_dock_area.toolbar.widgets["menu_devices"].widgets["positioner_box"].trigger()
-    assert "positioner_box_1" in bec_dock_area.panels
-    assert (
-        bec_dock_area.panels["positioner_box_1"].widgets[0].config.widget_class == "PositionerBox"
-    )
-
-
-def test_toolbar_add_utils_queue(bec_dock_area):
-    bec_dock_area.toolbar.widgets["menu_utils"].widgets["queue"].trigger()
-    assert "queue_1" in bec_dock_area.panels
-    assert bec_dock_area.panels["queue_1"].widgets[0].config.widget_class == "BECQueue"
-
-
-def test_toolbar_add_utils_status(bec_dock_area):
-    bec_dock_area.toolbar.widgets["menu_utils"].widgets["status"].trigger()
-    assert "status_1" in bec_dock_area.panels
-    assert bec_dock_area.panels["status_1"].widgets[0].config.widget_class == "BECStatusBox"
-
-
-def test_toolbar_add_utils_progress_bar(bec_dock_area):
-    bec_dock_area.toolbar.widgets["menu_utils"].widgets["progress_bar"].trigger()
-    assert "progress_bar_1" in bec_dock_area.panels
-    assert (
-        bec_dock_area.panels["progress_bar_1"].widgets[0].config.widget_class == "RingProgressBar"
-    )

tests/unit_tests/test_bec_figure.py

@@ -1,270 +0,0 @@
-# pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
-
-import numpy as np
-import pytest
-
-from bec_widgets.widgets.figure import BECFigure
-from bec_widgets.widgets.figure.plots.image.image import BECImageShow
-from bec_widgets.widgets.figure.plots.motor_map.motor_map import BECMotorMap
-from bec_widgets.widgets.figure.plots.waveform.waveform import BECWaveform
-
-from .client_mocks import mocked_client
-from .conftest import create_widget
-
-
-def test_bec_figure_init(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    assert bec_figure is not None
-    assert bec_figure.client is not None
-    assert isinstance(bec_figure, BECFigure)
-    assert bec_figure.config.widget_class == "BECFigure"
-
-
-def test_bec_figure_init_with_config(mocked_client):
-    config = {"widget_class": "BECFigure", "gui_id": "test_gui_id", "theme": "dark"}
-    widget = BECFigure(client=mocked_client, config=config)
-    assert widget.config.gui_id == "test_gui_id"
-    assert widget.config.theme == "dark"
-
-
-def test_bec_figure_add_remove_plot(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    initial_count = len(bec_figure._widgets)
-
-    # Adding 3 widgets -  2 WaveformBase and 1 PlotBase
-    w0 = bec_figure.plot(new=True)
-    w1 = bec_figure.plot(new=True)
-    w2 = bec_figure.add_widget(widget_type="BECPlotBase")
-
-    # Check if the widgets were added
-    assert len(bec_figure._widgets) == initial_count + 3
-    assert w0.gui_id in bec_figure._widgets
-    assert w1.gui_id in bec_figure._widgets
-    assert w2.gui_id in bec_figure._widgets
-    assert bec_figure._widgets[w0.gui_id].config.widget_class == "BECWaveform"
-    assert bec_figure._widgets[w1.gui_id].config.widget_class == "BECWaveform"
-    assert bec_figure._widgets[w2.gui_id].config.widget_class == "BECPlotBase"
-
-    # Check accessing positions by the grid in figure
-    assert bec_figure[0, 0] == w0
-    assert bec_figure[1, 0] == w1
-    assert bec_figure[2, 0] == w2
-
-    # Removing 1 widget
-    bec_figure.remove(widget_id=w0.gui_id)
-    assert len(bec_figure._widgets) == initial_count + 2
-    assert w0.gui_id not in bec_figure._widgets
-    assert w2.gui_id in bec_figure._widgets
-    assert bec_figure._widgets[w1.gui_id].config.widget_class == "BECWaveform"
-
-
-def test_add_different_types_of_widgets(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    plt = bec_figure.plot(x_name="samx", y_name="bpm4i")
-    im = bec_figure.image("eiger")
-    motor_map = bec_figure.motor_map("samx", "samy")
-
-    assert plt.__class__ == BECWaveform
-    assert im.__class__ == BECImageShow
-    assert motor_map.__class__ == BECMotorMap
-
-
-def test_access_widgets_access_errors(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    bec_figure.plot(row=0, col=0)
-
-    # access widget by non-existent coordinates
-    with pytest.raises(ValueError) as excinfo:
-        bec_figure[0, 2]
-        assert "No widget at coordinates (0, 2)" in str(excinfo.value)
-
-    # access widget by non-existent widget_id
-    with pytest.raises(KeyError) as excinfo:
-        bec_figure["non_existent_widget"]
-        assert "Widget with id 'non_existent_widget' not found" in str(excinfo.value)
-
-    # access widget by wrong type
-    with pytest.raises(TypeError) as excinfo:
-        bec_figure[1.2]
-        assert (
-            "Key must be a string (widget id) or a tuple of two integers (grid coordinates)"
-            in str(excinfo.value)
-        )
-
-
-def test_add_plot_to_occupied_position(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    bec_figure.plot(row=0, col=0)
-
-    with pytest.raises(ValueError) as excinfo:
-        bec_figure.plot(row=0, col=0, new=True)
-        assert "Position at row 0 and column 0 is already occupied." in str(excinfo.value)
-
-
-def test_remove_plots(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    w1 = bec_figure.plot(row=0, col=0)
-    w2 = bec_figure.plot(row=0, col=1)
-    w3 = bec_figure.plot(row=1, col=0)
-    w4 = bec_figure.plot(row=1, col=1)
-
-    assert bec_figure[0, 0] == w1
-    assert bec_figure[0, 1] == w2
-    assert bec_figure[1, 0] == w3
-    assert bec_figure[1, 1] == w4
-
-    # remove by coordinates
-    bec_figure[0, 0].remove()
-    assert w1.gui_id not in bec_figure._widgets
-
-    # remove by widget_id
-    bec_figure.remove(widget_id=w2.gui_id)
-    assert w2.gui_id not in bec_figure._widgets
-
-    # remove by widget object
-    w3.remove()
-    assert w3.gui_id not in bec_figure._widgets
-
-    # check the remaining widget 4
-    assert bec_figure[0, 0] == w4
-    assert bec_figure[w4.gui_id] == w4
-    assert w4.gui_id in bec_figure._widgets
-    assert len(bec_figure._widgets) == 1
-
-
-def test_remove_plots_by_coordinates_ints(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    w1 = bec_figure.plot(row=0, col=0)
-    w2 = bec_figure.plot(row=0, col=1)
-
-    bec_figure.remove(row=0, col=0)
-    assert w1.gui_id not in bec_figure._widgets
-    assert w2.gui_id in bec_figure._widgets
-    assert bec_figure[0, 0] == w2
-    assert len(bec_figure._widgets) == 1
-
-
-def test_remove_plots_by_coordinates_tuple(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    w1 = bec_figure.plot(row=0, col=0)
-    w2 = bec_figure.plot(row=0, col=1)
-
-    bec_figure.remove(coordinates=(0, 0))
-    assert w1.gui_id not in bec_figure._widgets
-    assert w2.gui_id in bec_figure._widgets
-    assert bec_figure[0, 0] == w2
-    assert len(bec_figure._widgets) == 1
-
-
-def test_remove_plot_by_id_error(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    bec_figure.plot()
-
-    with pytest.raises(ValueError) as excinfo:
-        bec_figure.remove(widget_id="non_existent_widget")
-        assert "Widget with ID 'non_existent_widget' does not exist." in str(excinfo.value)
-
-
-def test_remove_plot_by_coordinates_error(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    bec_figure.plot(row=0, col=0)
-
-    with pytest.raises(ValueError) as excinfo:
-        bec_figure.remove(0, 1)
-        assert "No widget at coordinates (0, 1)" in str(excinfo.value)
-
-
-def test_remove_plot_by_providing_nothing(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    bec_figure.plot(row=0, col=0)
-
-    with pytest.raises(ValueError) as excinfo:
-        bec_figure.remove()
-        assert "Must provide either widget_id or coordinates for removal." in str(excinfo.value)
-
-
-# def test_change_theme(bec_figure): #TODO do no work at python 3.12
-#     bec_figure.change_theme("dark")
-#     assert bec_figure.config.theme == "dark"
-#     assert bec_figure.backgroundBrush().color().name() == "#000000"
-#     bec_figure.change_theme("light")
-#     assert bec_figure.config.theme == "light"
-#     assert bec_figure.backgroundBrush().color().name() == "#ffffff"
-#     bec_figure.change_theme("dark")
-#     assert bec_figure.config.theme == "dark"
-#     assert bec_figure.backgroundBrush().color().name() == "#000000"
-
-
-def test_change_layout(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    w1 = bec_figure.plot(row=0, col=0)
-    w2 = bec_figure.plot(row=0, col=1)
-    w3 = bec_figure.plot(row=1, col=0)
-    w4 = bec_figure.plot(row=1, col=1)
-
-    bec_figure.change_layout(max_columns=1)
-
-    assert np.shape(bec_figure.grid) == (4, 1)
-    assert bec_figure[0, 0] == w1
-    assert bec_figure[1, 0] == w2
-    assert bec_figure[2, 0] == w3
-    assert bec_figure[3, 0] == w4
-
-    bec_figure.change_layout(max_rows=1)
-
-    assert np.shape(bec_figure.grid) == (1, 4)
-    assert bec_figure[0, 0] == w1
-    assert bec_figure[0, 1] == w2
-    assert bec_figure[0, 2] == w3
-    assert bec_figure[0, 3] == w4
-
-
-def test_clear_all(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    bec_figure.plot(row=0, col=0)
-    bec_figure.plot(row=0, col=1)
-    bec_figure.plot(row=1, col=0)
-    bec_figure.plot(row=1, col=1)
-
-    bec_figure.clear_all()
-
-    assert len(bec_figure._widgets) == 0
-    assert np.shape(bec_figure.grid) == (0,)
-
-
-def test_shortcuts(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    plt = bec_figure.plot(x_name="samx", y_name="bpm4i")
-    im = bec_figure.image("eiger")
-    motor_map = bec_figure.motor_map("samx", "samy")
-
-    assert plt.config.widget_class == "BECWaveform"
-    assert plt.__class__ == BECWaveform
-    assert im.config.widget_class == "BECImageShow"
-    assert im.__class__ == BECImageShow
-    assert motor_map.config.widget_class == "BECMotorMap"
-    assert motor_map.__class__ == BECMotorMap
-
-
-def test_plot_access_factory(qtbot, mocked_client):
-    bec_figure = create_widget(qtbot, BECFigure, client=mocked_client)
-    plt_00 = bec_figure.plot(x_name="samx", y_name="bpm4i")
-    plt_01 = bec_figure.plot(x_name="samx", y_name="bpm4i", row=0, col=1)
-    plt_10 = bec_figure.plot(new=True)
-
-    assert bec_figure.widget_list[0] == plt_00
-    assert bec_figure.widget_list[1] == plt_01
-    assert bec_figure.widget_list[2] == plt_10
-    assert bec_figure.axes(row=0, col=0) == plt_00
-    assert bec_figure.axes(row=0, col=1) == plt_01
-    assert bec_figure.axes(row=1, col=0) == plt_10
-
-    assert len(plt_00.curves) == 1
-    assert len(plt_01.curves) == 1
-    assert len(plt_10.curves) == 0
-
-    # update plt_00
-    bec_figure.plot(x_name="samx", y_name="bpm3a")
-    bec_figure.plot(x=[1, 2, 3], y=[1, 2, 3], row=0, col=0)
-
-    assert len(plt_00.curves) == 3

tests/unit_tests/test_bec_image.py

@@ -1,63 +0,0 @@
-# pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
-from unittest import mock
-
-import numpy as np
-import pytest
-from bec_lib import messages
-from qtpy.QtGui import QFontInfo
-
-from bec_widgets.widgets.figure import BECFigure
-
-from .client_mocks import mocked_client
-from .conftest import create_widget
-
-
-@pytest.fixture
-def bec_image_show(bec_figure):
-    yield bec_figure.image("eiger")
-
-
-def test_on_image_update(qtbot, mocked_client):
-    bec_image_show = create_widget(qtbot, BECFigure, client=mocked_client).image("eiger")
-    data = np.random.rand(100, 100)
-    msg = messages.DeviceMonitor2DMessage(device="eiger", data=data, metadata={"scan_id": "12345"})
-    bec_image_show.on_image_update(msg.content, msg.metadata)
-    img = bec_image_show.images[0]
-    assert np.array_equal(img.get_data(), data)
-
-
-def test_autorange_on_image_update(qtbot, mocked_client):
-    bec_image_show = create_widget(qtbot, BECFigure, client=mocked_client).image("eiger")
-    # Check if autorange mode "mean" works, should be default
-    data = np.random.rand(100, 100)
-    msg = messages.DeviceMonitor2DMessage(device="eiger", data=data, metadata={"scan_id": "12345"})
-    bec_image_show.on_image_update(msg.content, msg.metadata)
-    img = bec_image_show.images[0]
-    assert np.array_equal(img.get_data(), data)
-    vmin = max(np.mean(data) - 2 * np.std(data), 0)
-    vmax = np.mean(data) + 2 * np.std(data)
-    assert np.isclose(img.color_bar.getLevels(), (vmin, vmax), rtol=(1e-5, 1e-5)).all()
-    # Test general update with autorange True, mode "max"
-    bec_image_show.set_autorange_mode("max")
-    bec_image_show.on_image_update(msg.content, msg.metadata)
-    img = bec_image_show.images[0]
-    vmin = np.min(data)
-    vmax = np.max(data)
-    assert np.array_equal(img.get_data(), data)
-    assert np.isclose(img.color_bar.getLevels(), (vmin, vmax), rtol=(1e-5, 1e-5)).all()
-    # Change the input data, and switch to autorange False, colormap levels should stay untouched
-    data *= 100
-    msg = messages.DeviceMonitor2DMessage(device="eiger", data=data, metadata={"scan_id": "12345"})
-    bec_image_show.set_autorange(False)
-    bec_image_show.on_image_update(msg.content, msg.metadata)
-    img = bec_image_show.images[0]
-    assert np.array_equal(img.get_data(), data)
-    assert np.isclose(img.color_bar.getLevels(), (vmin, vmax), rtol=(1e-3, 1e-3)).all()
-    # Reactivate autorange, should now scale the new data
-    bec_image_show.set_autorange(True)
-    bec_image_show.set_autorange_mode("mean")
-    bec_image_show.on_image_update(msg.content, msg.metadata)
-    img = bec_image_show.images[0]
-    vmin = max(np.mean(data) - 2 * np.std(data), 0)
-    vmax = np.mean(data) + 2 * np.std(data)
-    assert np.isclose(img.color_bar.getLevels(), (vmin, vmax), rtol=(1e-5, 1e-5)).all()