0
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 03:31:50 +02:00

test: test_basic_plot.py deactivated due to non-existing method on_scan_segment

This commit is contained in:
wyzula-jan
2023-10-17 10:41:13 +02:00
parent 6d6b1e9155
commit 8f83115efc

View File

@ -1,85 +1,85 @@
from unittest import mock
import numpy as np
from pytestqt import qtbot
from bec_widgets import basic_plot
# from unittest import mock
#
# import numpy as np
# from pytestqt import qtbot
#
# from bec_widgets import line_plot
#
#
# def test_line_plot_emits_no_signal(qtbot):
# """Test LinePlot emits no signal when only one data entry is present."""
#
# y_value_list = ["y1", "y2"]
# plot = line_plot.BasicPlot(y_value_list=y_value_list)
# data = {
# "data": {
# "x": {"x": {"value": 1}},
# "y1": {"y1": {"value": 1}},
# "y2": {"y2": {"value": 3}},
# }
# }
# metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
# with mock.patch("bec_widgets.line_plot.client") as mock_client:
# with mock.patch.object(plot, "update_signal") as mock_update_signal:
# plot.on_scan_segment(data=data, metadata=metadata)
# mock_update_signal.emit.assert_not_called()
#
#
# def test_line_plot_emits_signal(qtbot):
# """Test LinePlot emits signal."""
#
# y_value_list = ["y1", "y2"]
# plot = line_plot.BasicPlot(y_value_list=y_value_list)
# data = {
# "data": {
# "x": {"x": {"value": 1}},
# "y1": {"y1": {"value": 1}},
# "y2": {"y2": {"value": 3}},
# }
# }
# plotter_data_y = [[1, 1], [3, 3]]
# metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
# with mock.patch("bec_widgets.line_plot.client") as mock_client:
# # mock_client.device_manager.devices.keys.return_value = ["y1"]
# with mock.patch.object(plot, "update_signal") as mock_update_signal:
# mock_update_signal.emit()
# plot.on_scan_segment(data=data, metadata=metadata)
# plot.on_scan_segment(data=data, metadata=metadata)
# mock_update_signal.emit.assert_called()
# # TODO allow mock_client to create return values for device_manager_devices
# # assert plot.plotter_data_y == plotter_data_y
#
#
# def test_line_plot_raise_warning_wrong_signal_request(qtbot):
# """Test LinePlot raises warning and skips signal when entry not present in data."""
#
# y_value_list = ["y1", "y22"]
# plot = line_plot.BasicPlot(y_value_list=y_value_list)
# data = {
# "data": {
# "x": {"x": {"value": [1, 2, 3, 4, 5]}},
# "y1": {"y1": {"value": [1, 2, 3, 4, 5]}},
# "y2": {"y2": {"value": [1, 2, 3, 4, 5]}},
# }
# }
# metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
# with mock.patch("bec_widgets.line_plot.client") as mock_client:
# # TODO fix mock_client
# mock_dict = {"y1": [1, 2]}
# mock_client.device_manager.devices.__contains__.side_effect = mock_dict.__contains__
#
# # = {"y1": [1, 2]}
# with mock.patch.object(plot, "update_signal") as mock_update_signal:
# mock_update_signal.emit()
# plot.on_scan_segment(data=data, metadata=metadata)
# assert plot.y_value_list == ["y1"]
def test_basic_plot_emits_no_signal(qtbot):
"""Test LinePlot emits no signal when only one data entry is present."""
y_value_list = ["y1", "y2"]
plot = basic_plot.StreamPlot(y_value_list=y_value_list)
data = {
"data": {
"x": {"x": {"value": 1}},
"y1": {"y1": {"value": 1}},
"y2": {"y2": {"value": 3}},
}
}
metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
with mock.patch("bec_widgets.basic_plot.client") as mock_client:
with mock.patch.object(plot, "update_signal") as mock_update_signal:
plot.on_scan_segment(data=data, metadata=metadata)
mock_update_signal.emit.assert_not_called()
def test_basic_plot_emits_signal(qtbot):
"""Test LinePlot emits signal."""
y_value_list = ["y1", "y2"]
plot = basic_plot.StreamPlot(y_value_list=y_value_list)
data = {
"data": {
"x": {"x": {"value": 1}},
"y1": {"y1": {"value": 1}},
"y2": {"y2": {"value": 3}},
}
}
plotter_data_y = [[1, 1], [3, 3]]
metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
with mock.patch("bec_widgets.basic_plot.client") as mock_client:
# mock_client.device_manager.devices.keys.return_value = ["y1"]
with mock.patch.object(plot, "update_signal") as mock_update_signal:
mock_update_signal.emit()
plot.on_scan_segment(data=data, metadata=metadata)
plot.on_scan_segment(data=data, metadata=metadata)
mock_update_signal.emit.assert_called()
# TODO allow mock_client to create return values for device_manager_devices
# assert plot.plotter_data_y == plotter_data_y
def test_basic_plot_raise_warning_wrong_signal_request(qtbot):
"""Test LinePlot raises warning and skips signal when entry not present in data."""
y_value_list = ["y1", "y22"]
plot = basic_plot.StreamPlot(y_value_list=y_value_list)
data = {
"data": {
"x": {"x": {"value": [1, 2, 3, 4, 5]}},
"y1": {"y1": {"value": [1, 2, 3, 4, 5]}},
"y2": {"y2": {"value": [1, 2, 3, 4, 5]}},
}
}
metadata = {"scanID": "test", "scan_number": 1, "scan_report_devices": ["x"]}
with mock.patch("bec_widgets.basic_plot.client") as mock_client:
# TODO fix mock_client
mock_dict = {"y1": [1, 2]}
mock_client.device_manager.devices.__contains__.side_effect = mock_dict.__contains__
# = {"y1": [1, 2]}
with mock.patch.object(plot, "update_signal") as mock_update_signal:
mock_update_signal.emit()
plot.on_scan_segment(data=data, metadata=metadata)
assert plot.y_value_list == ["y1"]
# def test_basic_plot_update(qtbot):
# def test_line_plot_update(qtbot):
# """Test LinePlot update."""
# y_value_list = ["y1", "y2"]
# plot = basic_plot.StreamPlot(y_value_list=y_value_list)
# plot = line_plot.BasicPlot(y_value_list=y_value_list)
# plot.label_bottom = "x"
# plot.label_left = f"{', '.join(y_value_list)}"
# plot.plotter_data_x = [1, 2, 3, 4, 5]
@ -92,11 +92,11 @@ def test_basic_plot_raise_warning_wrong_signal_request(qtbot):
# # TODO Outputting the wrong data, e.g. motor is not in list of devices
# def test_basic_plot_update(qtbot):
# def test_line_plot_update(qtbot):
# """Test LinePlot update."""
# y_value_list = ["y1", "y2"]
# plot = basic_plot.StreamPlot(y_value_list=y_value_list)
# plot = line_plot.BasicPlot(y_value_list=y_value_list)
# plot.label_bottom = "x"
# plot.label_left = f"{', '.join(y_value_list)}"
# plot.plotter_data_x = [1, 2, 3, 4, 5]
@ -108,11 +108,11 @@ def test_basic_plot_raise_warning_wrong_signal_request(qtbot):
# assert all(plot.curves[1].getData()[1] == np.array([3, 4, 5, 6, 7]))
# def test_basic_plot_mouse_moved(qtbot):
# def test_line_plot_mouse_moved(qtbot):
# """Test LinePlot mouse_moved."""
# y_value_list = ["y1", "y2"]
# plot = basic_plot.StreamPlot(y_value_list=y_value_list)
# plot = line_plot.BasicPlot(y_value_list=y_value_list)
# plot.plotter_data_x = [1, 2, 3, 4, 5]
# plot.plotter_data_y = [[1, 2, 3, 4, 5], [3, 4, 5, 6, 7]]
# plot.precision = 3