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

test: fixed client mock

Browse Source
This commit is contained in:
wakonig_k
2023-07-19 10:06:21 +02:00
parent d37fbf5c4f
commit 9883caa1b3
2 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
bec_widgets/line_plot.py
View File

@ -1,14 +1,13 @@
from typing import Any import os
import warnings import warnings
from typing import Any
import numpy as np import numpy as np
import pyqtgraph as pg import pyqtgraph as pg
from pyqtgraph import mkPen, mkBrush, mkColor from bec_lib import BECClient
from pyqtgraph import mkBrush, mkColor, mkPen
from pyqtgraph.Qt import QtCore, QtWidgets, uic from pyqtgraph.Qt import QtCore, QtWidgets, uic
from pyqtgraph.Qt.QtCore import pyqtSignal from pyqtgraph.Qt.QtCore import pyqtSignal
from bec_lib import BECClient
import os
class BasicPlot(QtWidgets.QWidget): class BasicPlot(QtWidgets.QWidget):
@ -169,7 +168,7 @@ class BasicPlot(QtWidgets.QWidget):
remove_y_value_index = [ remove_y_value_index = [
index index
for index, y_value in enumerate(self.y_value_list) for index, y_value in enumerate(self.y_value_list)
if y_value not in client.device_manager.devices.keys() if y_value not in client.device_manager.devices
] ]
if remove_y_value_index: if remove_y_value_index:
for ii in sorted(remove_y_value_index, reverse=True): for ii in sorted(remove_y_value_index, reverse=True):
@ -252,6 +251,7 @@ class BasicPlot(QtWidgets.QWidget):
if __name__ == "__main__": if __name__ == "__main__":
import argparse import argparse
from bec_widgets import ctrl_c from bec_widgets import ctrl_c
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()

12
tests/test_line_plot.py
View File

@ -1,7 +1,9 @@
from pytestqt import qtbot
from unittest import mock from unittest import mock
from bec_widgets import line_plot
import numpy as np import numpy as np
from pytestqt import qtbot
from bec_widgets import line_plot
def test_line_plot_emits_no_signal(qtbot): def test_line_plot_emits_no_signal(qtbot):
@ -63,12 +65,14 @@ def test_line_plot_raise_warning_wrong_signal_request(qtbot):
metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]} metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
with mock.patch("bec_widgets.line_plot.BECClient") as mock_client: with mock.patch("bec_widgets.line_plot.BECClient") as mock_client:
# TODO fix mock_client # TODO fix mock_client
mock_client.device_manager.devices.keys.return_value = ["y1"] mock_dict = {"y1": [1, 2]}
mock_client().device_manager.devices.__contains__.side_effect = mock_dict.__contains__
# = {"y1": [1, 2]} # = {"y1": [1, 2]}
with mock.patch.object(plot, "update_signal") as mock_update_signal: with mock.patch.object(plot, "update_signal") as mock_update_signal:
mock_update_signal.emit() mock_update_signal.emit()
plot(data=data, metadata=metadata) plot(data=data, metadata=metadata)
assert all(plot.y_value_list == ["y1"]) assert plot.y_value_list == ["y1"]
def test_line_plot_update(qtbot): def test_line_plot_update(qtbot):