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(waveform1d): dap e2e test added

Browse Source
This commit is contained in:
wyzula_j
2024-06-25 15:51:36 +02:00
parent 1866ba66c8
commit 7271b422f9
2 changed files with 36 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/end-2-end/test_bec_dock_rpc_e2e.py
View File

@ -53,6 +53,7 @@ def test_rpc_add_dock_with_figure_e2e(bec_client_lib, rpc_server_dock):
assert im.__class__ == BECImageShow assert im.__class__ == BECImageShow
assert mm.config_dict["signals"] == { assert mm.config_dict["signals"] == {
"dap": None,
"source": "device_readback", "source": "device_readback",
"x": { "x": {
"name": "samx", "name": "samx",
@ -71,6 +72,7 @@ def test_rpc_add_dock_with_figure_e2e(bec_client_lib, rpc_server_dock):
"z": None, "z": None,
} }
assert plt.config_dict["curves"]["bpm4i-bpm4i"]["signals"] == { assert plt.config_dict["curves"]["bpm4i-bpm4i"]["signals"] == {
"dap": None,
"source": "scan_segment", "source": "scan_segment",
"x": {"name": "samx", "entry": "samx", "unit": None, "modifier": None, "limits": None}, "x": {"name": "samx", "entry": "samx", "unit": None, "modifier": None, "limits": None},
"y": {"name": "bpm4i", "entry": "bpm4i", "unit": None, "modifier": None, "limits": None}, "y": {"name": "bpm4i", "entry": "bpm4i", "unit": None, "modifier": None, "limits": None},

34
tests/end-2-end/test_bec_figure_rpc_e2e.py
View File

@ -1,3 +1,5 @@
import time
import numpy as np import numpy as np
import pytest import pytest
from bec_lib.endpoints import MessageEndpoints from bec_lib.endpoints import MessageEndpoints
@ -38,6 +40,7 @@ def test_rpc_plotting_shortcuts_init_configs(rpc_server_figure, qtbot):
# check if the correct devices are set # check if the correct devices are set
# plot # plot
assert plt.config_dict["curves"]["bpm4i-bpm4i"]["signals"] == { assert plt.config_dict["curves"]["bpm4i-bpm4i"]["signals"] == {
"dap": None,
"source": "scan_segment", "source": "scan_segment",
"x": {"name": "samx", "entry": "samx", "unit": None, "modifier": None, "limits": None}, "x": {"name": "samx", "entry": "samx", "unit": None, "modifier": None, "limits": None},
"y": {"name": "bpm4i", "entry": "bpm4i", "unit": None, "modifier": None, "limits": None}, "y": {"name": "bpm4i", "entry": "bpm4i", "unit": None, "modifier": None, "limits": None},
@ -47,6 +50,7 @@ def test_rpc_plotting_shortcuts_init_configs(rpc_server_figure, qtbot):
assert im.config_dict["images"]["eiger"]["monitor"] == "eiger" assert im.config_dict["images"]["eiger"]["monitor"] == "eiger"
# motor map # motor map
assert motor_map.config_dict["signals"] == { assert motor_map.config_dict["signals"] == {
"dap": None,
"source": "device_readback", "source": "device_readback",
"x": { "x": {
"name": "samx", "name": "samx",
@ -66,6 +70,7 @@ def test_rpc_plotting_shortcuts_init_configs(rpc_server_figure, qtbot):
} }
# plot with z scatter # plot with z scatter
assert plt_z.config_dict["curves"]["bpm4i-bpm4i"]["signals"] == { assert plt_z.config_dict["curves"]["bpm4i-bpm4i"]["signals"] == {
"dap": None,
"source": "scan_segment", "source": "scan_segment",
"x": {"name": "samx", "entry": "samx", "unit": None, "modifier": None, "limits": None}, "x": {"name": "samx", "entry": "samx", "unit": None, "modifier": None, "limits": None},
"y": {"name": "samy", "entry": "samy", "unit": None, "modifier": None, "limits": None}, "y": {"name": "samy", "entry": "samy", "unit": None, "modifier": None, "limits": None},
@ -151,3 +156,32 @@ def test_rpc_motor_map(rpc_server_figure, bec_client_lib):
np.testing.assert_equal( np.testing.assert_equal(
[motor_map_data["x"][-1], motor_map_data["y"][-1]], [final_pos_x, final_pos_y] [motor_map_data["x"][-1], motor_map_data["y"][-1]], [final_pos_x, final_pos_y]
) )
def test_dap_rpc(rpc_server_figure, bec_client_lib):
fig = BECFigure(rpc_server_figure)
plt = fig.plot(x_name="samx", y_name="bpm4i", dap="GaussianModel")
client = bec_client_lib
dev = client.device_manager.devices
scans = client.scans
dev.bpm4i.sim.sim_select_model("GaussianModel")
params = dev.bpm4i.sim.sim_params
params.update(
{"noise": "uniform", "noise_multiplier": 10, "center": 5, "sigma": 1, "amplitude": 200}
)
dev.bpm4i.sim.sim_params = params
time.sleep(1)
res = scans.line_scan(dev.samx, 0, 8, steps=50, relative=False)
res.wait()
time.sleep(2)
dap_curve = plt.get_curve("bpm4i-bpm4i-GaussianModel")
fit_params = dap_curve.dap_params
print(fit_params)
assert np.isclose(fit_params["center"], 5, atol=0.5)