bec/debye_bec
0
0
Fork 0
Code Issues 9 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

test: fix ScanStatusMessage

Browse Source
This commit is contained in:
appel_c
2025-01-20 13:46:24 +01:00
parent d50519f3d3
commit a55f8bf705
1 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tests/tests_devices/test_mo1_bragg.py
View File

@@ -149,6 +149,7 @@ def test_set_xas_settings(mock_bragg):
assert dev.scan_settings.s_scan_angle_hi.get() == 20
assert dev.scan_settings.s_scan_scantime.get() == 1
def test_set_xrd_settings(mock_bragg):
dev = mock_bragg
dev.set_xrd_settings(
@@ -170,6 +171,7 @@ def test_set_xrd_settings(mock_bragg):
assert dev.scan_settings.xrd_every_n_lo.get() == 1
assert dev.scan_settings.xrd_every_n_hi.get() == 5
def test_set_control_settings(mock_bragg):
dev = mock_bragg
dev.set_scan_control_settings(mode=ScanControlMode.SIMPLE, scan_duration=10)
@@ -281,7 +283,13 @@ def test_unstage(mock_bragg):
scan_type="monitor_scan",
parameter={
"args": {},
"kwargs": {"device": "mo1_bragg", "start": 0, "stop": 10, "relative": True},
"kwargs": {
"device": "mo1_bragg",
"start": 0,
"stop": 10,
"relative": True,
"system_config": {"file_suffix": None, "file_directory": None},
},
"num_points": 100,
},
queue="primary",
@@ -297,6 +305,7 @@ def test_unstage(mock_bragg):
"stop": 10,
"scan_time": 1,
"scan_duration": 10,
"system_config": {"file_suffix": None, "file_directory": None},
},
"num_points": 100,
},
@@ -321,6 +330,7 @@ def test_unstage(mock_bragg):
"exp_time_high": 3,
"cycle_low": 1,
"cycle_high": 5,
"system_config": {"file_suffix": None, "file_directory": None},
},
"num_points": 10,
},
@@ -339,6 +349,7 @@ def test_unstage(mock_bragg):
"scan_duration": 10,
"p_kink": 50,
"e_kink": 8500,
"system_config": {"file_suffix": None, "file_directory": None},
},
"num_points": 100,
},
@@ -365,6 +376,7 @@ def test_unstage(mock_bragg):
"exp_time_high": 3,
"cycle_low": 1,
"cycle_high": 5,
"system_config": {"file_suffix": None, "file_directory": None},
},
"num_points": 10,
},
@@ -409,7 +421,12 @@ def test_stage(mock_bragg, scan_worker_mock, msg):
):
scan_name = scan_status_msg.content["info"].get("scan_name", "")
# Chek the not implemented fly scan first, should raise Mo1BraggError
if scan_name not in ["xas_simple_scan", "xas_simple_scan_with_xrd", "xas_advanced_scan", "xas_advanced_scan_with_xrd"]:
if scan_name not in [
"xas_simple_scan",
"xas_simple_scan_with_xrd",
"xas_advanced_scan",
"xas_advanced_scan_with_xrd",
]:
with pytest.raises(Mo1BraggError):
mock_bragg.stage()
assert mock_check_scan_msg.call_count == 1
@@ -417,7 +434,9 @@ def test_stage(mock_bragg, scan_worker_mock, msg):
else:
with (
mock.patch.object(mock_bragg, "set_xas_settings") as mock_xas_settings,
mock.patch.object(mock_bragg, "set_advanced_xas_settings") as mock_advanced_xas_settings,
mock.patch.object(
mock_bragg, "set_advanced_xas_settings"
) as mock_advanced_xas_settings,
mock.patch.object(mock_bragg, "set_xrd_settings") as mock_xrd_settings,
mock.patch.object(
mock_bragg, "set_scan_control_settings"
@@ -538,4 +557,4 @@ def test_stage(mock_bragg, scan_worker_mock, msg):
scan_duration=scan_status_msg.content["info"]["kwargs"][
"scan_duration"
],
)
)