bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-13 19:21:50 +02:00
Code Activity

fix: add is_log checks and functionality to plot_indicator_items

Browse Source
This commit is contained in:
appel_c
2024-09-11 22:55:59 +02:00
parent 63c24f97a3
commit 0f9953e8fd
2 changed files with 80 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
tests/unit_tests/test_utils_plot_indicators.py
View File

@ -1,43 +1,55 @@
import pyqtgraph as pg
import pytest
from bec_qthemes._main import AccentColors
from bec_widgets.utils.plot_indicator_items import BECArrowItem, BECTickItem
from bec_widgets.widgets.waveform.waveform_widget import BECWaveformWidget
from .client_mocks import mocked_client
@pytest.fixture(scope="function")
def arrow_item():
"""Fixture for the BECArrowItem class"""
item = BECArrowItem(plot_item=pg.PlotItem())
yield item
@pytest.fixture
def plot_widget_with_arrow_item(qtbot, mocked_client):
widget = BECWaveformWidget(client=mocked_client())
qtbot.addWidget(widget)
qtbot.waitExposed(widget)
yield widget.waveform.arrow_item, widget.waveform.plot_item
def test_arrow_item_add_to_plot(arrow_item):
@pytest.fixture
def plot_widget_with_tick_item(qtbot, mocked_client):
widget = BECWaveformWidget(client=mocked_client())
qtbot.addWidget(widget)
qtbot.waitExposed(widget)
yield widget.waveform.motor_pos_tick, widget.waveform.plot_item
def test_arrow_item_add_to_plot(plot_widget_with_arrow_item):
"""Test the add_to_plot method"""
arrow_item, plot_item = plot_widget_with_arrow_item
assert arrow_item.plot_item is not None
assert arrow_item.plot_item.items == []
arrow_item.accent_colors = AccentColors(theme="dark")
arrow_item.add_to_plot()
assert arrow_item.plot_item.items == [arrow_item.arrow_item]
def test_arrow_item_remove_to_plot(arrow_item):
def test_arrow_item_remove_to_plot(plot_widget_with_arrow_item):
"""Test the remove_from_plot method"""
arrow_item.accent_colors = AccentColors(theme="dark")
arrow_item, plot_item = plot_widget_with_arrow_item
arrow_item.add_to_plot()
assert arrow_item.plot_item.items == [arrow_item.arrow_item]
arrow_item.remove_from_plot()
assert arrow_item.plot_item.items == []
def test_arrow_item_set_position(arrow_item):
def test_arrow_item_set_position(plot_widget_with_arrow_item):
"""Test the set_position method"""
arrow_item, plot_item = plot_widget_with_arrow_item
container = []
def signal_callback(tup: tuple):
container.append(tup)
arrow_item.accent_colors = AccentColors(theme="dark")
arrow_item.add_to_plot()
arrow_item.position_changed.connect(signal_callback)
arrow_item.set_position(pos=(1, 1))
@ -47,44 +59,37 @@ def test_arrow_item_set_position(arrow_item):
assert container == [(1, 1), (2, 2)]
@pytest.fixture(scope="function")
def tick_item():
"""Fixture for the BECArrowItem class"""
item = BECTickItem(plot_item=pg.PlotItem())
yield item
def test_tick_item_add_to_plot(tick_item):
def test_tick_item_add_to_plot(plot_widget_with_tick_item):
"""Test the add_to_plot method"""
tick_item, plot_item = plot_widget_with_tick_item
assert tick_item.plot_item is not None
assert tick_item.plot_item.items == []
tick_item.accent_colors = AccentColors(theme="dark")
tick_item.add_to_plot()
assert tick_item.plot_item.layout.itemAt(4, 1) == tick_item.tick_item
def test_tick_item_remove_to_plot(tick_item):
def test_tick_item_remove_to_plot(plot_widget_with_tick_item):
"""Test the remove_from_plot method"""
tick_item.accent_colors = AccentColors(theme="dark")
tick_item, plot_item = plot_widget_with_tick_item
tick_item.add_to_plot()
assert tick_item.plot_item.layout.itemAt(4, 1) == tick_item.tick_item
tick_item.remove_from_plot()
assert tick_item.plot_item.layout.itemAt(4, 1) is None
def test_tick_item_set_position(tick_item):
def test_tick_item_set_position(plot_widget_with_tick_item):
"""Test the set_position method"""
tick_item, plot_item = plot_widget_with_tick_item
container = []
def signal_callback(val: float):
container.append(val)
tick_item.accent_colors = AccentColors(theme="dark")
tick_item.add_to_plot()
tick_item.position_changed.connect(signal_callback)
tick_item.set_position(pos=1)
assert tick_item._tick_pos == 1
assert tick_item._pos == 1
tick_item.set_position(pos=2)
assert tick_item._tick_pos == 2
assert tick_item._pos == 2
assert container == [1.0, 2.0]