0
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 11:41:49 +02:00

test(image_widget): tests added

This commit is contained in:
2024-07-20 19:21:52 +02:00
committed by wyzula_j
parent 43711680ba
commit 70fb276fdf
5 changed files with 232 additions and 3 deletions

View File

@ -89,12 +89,19 @@ class FakePositioner(FakeDevice):
class Positioner(FakePositioner):
"""just placeholder for testing embeded isinstance check in DeviceCombobox"""
"""just placeholder for testing embedded isinstance check in DeviceCombobox"""
def __init__(self, name="test", limits=None, read_value=1.0):
super().__init__(name, limits, read_value)
class Device(FakeDevice):
"""just placeholder for testing embedded isinstance check in DeviceCombobox"""
def __init__(self, name, enabled=True):
super().__init__(name, enabled)
class DMMock:
def __init__(self):
self.devices = DeviceContainer()
@ -120,6 +127,7 @@ DEVICES = [
FakeDevice("eiger"),
FakeDevice("async_device", readout_priority=ReadoutPriority.ASYNC),
Positioner("test", limits=[-10, 10], read_value=2.0),
Device("test_device"),
]

View File

@ -0,0 +1,219 @@
from unittest.mock import MagicMock, patch
import pyqtgraph as pg
import pytest
from bec_widgets.widgets.image.image_widget import BECImageWidget
from .client_mocks import mocked_client
@pytest.fixture
def image_widget(qtbot, mocked_client):
widget = BECImageWidget(client=mocked_client())
widget.toolbar.widgets["monitor"].device_combobox.set_device_filter("FakeDevice")
qtbot.addWidget(widget)
qtbot.waitExposed(widget)
yield widget
widget.close()
@pytest.fixture
def mock_image(image_widget):
image_mock = MagicMock()
image_widget._image = image_mock
return image_mock
def test_image_widget_init(image_widget):
assert image_widget is not None
assert image_widget.client is not None
assert isinstance(image_widget, BECImageWidget)
assert image_widget.config.widget_class == "BECImageWidget"
assert image_widget._image is not None
assert (
image_widget.toolbar.widgets["monitor"].device_combobox.config.device_filter == "FakeDevice"
)
assert image_widget.toolbar.widgets["drag_mode"].action.isChecked() == True
assert image_widget.toolbar.widgets["rectangle_mode"].action.isChecked() == False
assert image_widget.toolbar.widgets["auto_range"].action.isChecked() == False
assert image_widget.toolbar.widgets["auto_range_image"].action.isChecked() == True
assert image_widget.toolbar.widgets["FFT"].action.isChecked() == False
assert image_widget.toolbar.widgets["transpose"].action.isChecked() == False
assert image_widget.toolbar.widgets["log"].action.isChecked() == False
###################################
# Toolbar Actions
###################################
def test_toolbar_connect_action(image_widget, mock_image, qtbot):
combo = image_widget.toolbar.widgets["monitor"].device_combobox
combo.setCurrentText("eiger")
qtbot.wait(200)
assert combo.currentText() == "eiger"
action = image_widget.toolbar.widgets["connect"].action
action.trigger()
image_widget._image.image.assert_called_once_with(
monitor="eiger",
color_map="magma",
color_bar="full",
downsample=True,
opacity=1.0,
vrange=None,
)
def test_image_toolbar_drag_mode_action_triggered(image_widget, qtbot):
action_drag = image_widget.toolbar.widgets["drag_mode"].action
action_rectangle = image_widget.toolbar.widgets["rectangle_mode"].action
action_drag.trigger()
assert action_drag.isChecked() == True
assert action_rectangle.isChecked() == False
def test_image_toolbar_rectangle_mode_action_triggered(image_widget, qtbot):
action_drag = image_widget.toolbar.widgets["drag_mode"].action
action_rectangle = image_widget.toolbar.widgets["rectangle_mode"].action
action_rectangle.trigger()
assert action_drag.isChecked() == False
assert action_rectangle.isChecked() == True
def test_image_toolbar_auto_range(image_widget, mock_image):
action = image_widget.toolbar.widgets["auto_range"].action
action.trigger()
assert action.isChecked() == False
image_widget._image.set_auto_range.assert_called_once_with(False, "xy")
def test_image_toolbar_enable_mouse_pan_mode(qtbot, image_widget):
action_drag = image_widget.toolbar.widgets["drag_mode"].action
action_rectangle = image_widget.toolbar.widgets["rectangle_mode"].action
mock_view_box = MagicMock()
image_widget._image.plot_item.getViewBox = MagicMock(return_value=mock_view_box)
image_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)
def test_image_toolbar_auto_range_image(image_widget, mock_image):
action = image_widget.toolbar.widgets["auto_range_image"].action
action.trigger()
assert action.isChecked() == False
image_widget._image.set_autorange.assert_called_once_with(False)
def test_image_toolbar_FFT(image_widget, mock_image):
action = image_widget.toolbar.widgets["FFT"].action
action.trigger()
assert action.isChecked() == True
image_widget._image.set_fft.assert_called_once_with(True, None)
def test_image_toolbar_log(image_widget, mock_image):
action = image_widget.toolbar.widgets["log"].action
action.trigger()
assert action.isChecked() == True
image_widget._image.set_log.assert_called_once_with(True, None)
def test_image_toggle_transpose(image_widget, mock_image):
action = image_widget.toolbar.widgets["transpose"].action
action.trigger()
assert action.isChecked() == True
image_widget._image.set_transpose.assert_called_once_with(True, None)
def test_image_toolbar_rotation(image_widget, mock_image):
action_left = image_widget.toolbar.widgets["rotate_left"].action
action_right = image_widget.toolbar.widgets["rotate_right"].action
action_left.trigger()
image_widget._image.set_rotation(1, None)
action_right.trigger()
image_widget._image.set_rotation(2, None)
action_right.trigger()
image_widget._image.set_rotation(1, None)
###################################
# Wrapper methods for ImageShow
###################################
def test_image_set_image(image_widget, mock_image):
image_widget.image(monitor="image")
image_widget._image.image.assert_called_once_with(
monitor="image",
color_map="magma",
color_bar="full",
downsample=True,
opacity=1.0,
vrange=None,
)
def test_image_vrange(image_widget, mock_image):
image_widget.set_vrange(0, 1)
image_widget._image.set_vrange.assert_called_once_with(0, 1, None)
def test_image_set_color_map(image_widget, mock_image):
image_widget.set_color_map("viridis")
image_widget._image.set_color_map.assert_called_once_with("viridis", None)
def test_image_widget_set_title(image_widget, mock_image):
image_widget.set_title("Title Label")
image_widget._image.set_title.assert_called_once_with("Title Label")
def test_image_widget_set_x_label(image_widget, mock_image):
image_widget.set_x_label("X Label")
image_widget._image.set_x_label.assert_called_once_with("X Label")
def test_image_widget_set_y_label(image_widget, mock_image):
image_widget.set_y_label("Y Label")
image_widget._image.set_y_label.assert_called_once_with("Y Label")
def test_image_widget_set_x_scale(image_widget, mock_image):
image_widget.set_x_scale("linear")
image_widget._image.set_x_scale.assert_called_once_with("linear")
def test_image_widget_set_y_scale(image_widget, mock_image):
image_widget.set_y_scale("log")
image_widget._image.set_y_scale.assert_called_once_with("log")
def test_image_widget_set_x_lim(image_widget, mock_image):
image_widget.set_x_lim((0, 10))
image_widget._image.set_x_lim.assert_called_once_with((0, 10))
def test_image_widget_set_y_lim(image_widget, mock_image):
image_widget.set_y_lim((0, 10))
image_widget._image.set_y_lim.assert_called_once_with((0, 10))
def test_image_widget_set_grid(image_widget, mock_image):
image_widget.set_grid(True, False)
image_widget._image.set_grid.assert_called_once_with(True, False)
def test_image_widget_lock_aspect_ratio(image_widget, mock_image):
image_widget.lock_aspect_ratio(True)
image_widget._image.lock_aspect_ratio.assert_called_once_with(True)
def test_image_widget_export(image_widget, mock_image):
image_widget.export()
image_widget._image.export.assert_called_once()

View File

@ -69,4 +69,4 @@ def test_device_input_base_get_device_list(device_input_base):
def test_device_input_base_get_filters(device_input_base):
filters = device_input_base.get_available_filters()
assert filters == {"FakePositioner", "FakeDevice", "Positioner"}
assert filters == {"FakePositioner", "FakeDevice", "Positioner", "Device"}

View File

@ -69,6 +69,7 @@ def test_device_input_combobox_init(device_input_combobox):
"eiger",
"async_device",
"test",
"test_device",
]
@ -157,6 +158,7 @@ def test_device_input_line_edit_init(device_input_line_edit):
"eiger",
"async_device",
"test",
"test_device",
]

View File

@ -24,7 +24,7 @@ def mock_waveform(waveform_widget):
return waveform_mock
def test_waveform_widget_inti(waveform_widget):
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)