mirror of
https://github.com/ivan-usov-org/bec.git
synced 2025-04-21 02:10:01 +02:00
141 lines
9.5 KiB
Python
141 lines
9.5 KiB
Python
from unittest import mock
|
|
|
|
import pytest
|
|
from bec_lib.core.tests.utils import bec_client
|
|
from bec_lib.scans import ScanObject
|
|
|
|
|
|
def test_scan_object_raises(bec_client):
|
|
scan_info = {
|
|
"class": "FermatSpiralScan",
|
|
"arg_input": ["device", "float", "float"],
|
|
"required_kwargs": ["step", "relative"],
|
|
"arg_bundle_size": 3,
|
|
"scan_report_hint": "table",
|
|
"doc": "\n A scan following Fermat's spiral.\n\n Args:\n *args: pairs of device / start position / end position / steps arguments\n relative: Start from an absolute or relative position\n burst: number of acquisition per point\n optim_trajectory: routine used for the trajectory optimization, e.g. 'corridor'. Default: None\n\n Returns:\n\n Examples:\n >>> scans.fermat_scan(dev.motor1, -5, 5, dev.motor2, -5, 5, step=0.5, exp_time=0.1, relative=True, optim_trajectory=\"corridor\")\n\n ",
|
|
}
|
|
scan_name = "fermat_scan"
|
|
obj = ScanObject(scan_name, scan_info, bec_client)
|
|
with mock.patch.object(bec_client, "alarm_handler"):
|
|
with pytest.raises(TypeError):
|
|
obj._run()
|
|
|
|
|
|
def test_scan_object(bec_client):
|
|
scan_info = {
|
|
"class": "FermatSpiralScan",
|
|
"arg_input": ["device", "float", "float"],
|
|
"required_kwargs": ["step", "relative"],
|
|
"arg_bundle_size": 3,
|
|
"scan_report_hint": "table",
|
|
"doc": "\n A scan following Fermat's spiral.\n\n Args:\n *args: pairs of device / start position / end position / steps arguments\n relative: Start from an absolute or relative position\n burst: number of acquisition per point\n optim_trajectory: routine used for the trajectory optimization, e.g. 'corridor'. Default: None\n\n Returns:\n\n Examples:\n >>> scans.fermat_scan(dev.motor1, -5, 5, dev.motor2, -5, 5, step=0.5, exp_time=0.1, relative=True, optim_trajectory=\"corridor\")\n\n ",
|
|
}
|
|
scan_name = "fermat_scan"
|
|
obj = ScanObject(scan_name, scan_info, bec_client)
|
|
dev = bec_client.device_manager.devices
|
|
with mock.patch.object(bec_client, "alarm_handler"):
|
|
with mock.patch("bec_lib.scan_manager.ScanReport.from_request") as report:
|
|
obj._run(dev.samx, -5, 5, dev.samy, -5, 5, step=0.5, exp_time=0.1, relative=False)
|
|
report().wait.assert_not_called()
|
|
|
|
|
|
def test_scan_object_wo_live_updates(bec_client):
|
|
scan_info = {
|
|
"class": "FermatSpiralScan",
|
|
"arg_input": ["device", "float", "float"],
|
|
"required_kwargs": ["step", "relative"],
|
|
"arg_bundle_size": 3,
|
|
"scan_report_hint": "table",
|
|
"doc": "\n A scan following Fermat's spiral.\n\n Args:\n *args: pairs of device / start position / end position / steps arguments\n relative: Start from an absolute or relative position\n burst: number of acquisition per point\n optim_trajectory: routine used for the trajectory optimization, e.g. 'corridor'. Default: None\n\n Returns:\n\n Examples:\n >>> scans.fermat_scan(dev.motor1, -5, 5, dev.motor2, -5, 5, step=0.5, exp_time=0.1, relative=True, optim_trajectory=\"corridor\")\n\n ",
|
|
}
|
|
scan_name = "fermat_scan"
|
|
obj = ScanObject(scan_name, scan_info, bec_client)
|
|
dev = bec_client.device_manager.devices
|
|
bec_client.live_updates = None
|
|
with mock.patch.object(bec_client, "alarm_handler"):
|
|
with mock.patch("bec_lib.scan_manager.ScanReport.from_request") as report:
|
|
obj._run(dev.samx, -5, 5, dev.samy, -5, 5, step=0.5, exp_time=0.1, relative=False)
|
|
report().wait.assert_not_called()
|
|
|
|
|
|
def test_scan_object_receives_sample_name(bec_client):
|
|
scan_info = {
|
|
"class": "FermatSpiralScan",
|
|
"arg_input": ["device", "float", "float"],
|
|
"required_kwargs": ["step", "relative"],
|
|
"arg_bundle_size": 3,
|
|
"scan_report_hint": "table",
|
|
"doc": "\n A scan following Fermat's spiral.\n\n Args:\n *args: pairs of device / start position / end position / steps arguments\n relative: Start from an absolute or relative position\n burst: number of acquisition per point\n optim_trajectory: routine used for the trajectory optimization, e.g. 'corridor'. Default: None\n\n Returns:\n\n Examples:\n >>> scans.fermat_scan(dev.motor1, -5, 5, dev.motor2, -5, 5, step=0.5, exp_time=0.1, relative=True, optim_trajectory=\"corridor\")\n\n ",
|
|
}
|
|
scan_name = "fermat_scan"
|
|
obj = ScanObject(scan_name, scan_info, bec_client)
|
|
dev = bec_client.device_manager.devices
|
|
with mock.patch.object(bec_client, "alarm_handler"):
|
|
with mock.patch("bec_lib.scan_manager.ScanReport.from_request") as scan_report:
|
|
with mock.patch.object(bec_client, "get_global_var", return_value="test_sample"):
|
|
obj._run(dev.samx, -5, 5, dev.samy, -5, 5, step=0.5, exp_time=0.1, relative=False)
|
|
assert scan_report.call_args.args[0].metadata["sample_name"] == "test_sample"
|
|
|
|
|
|
def test_scan_object_receives_scan_group(bec_client):
|
|
scan_info = {
|
|
"class": "FermatSpiralScan",
|
|
"arg_input": ["device", "float", "float"],
|
|
"required_kwargs": ["step", "relative"],
|
|
"arg_bundle_size": 3,
|
|
"scan_report_hint": "table",
|
|
"doc": "\n A scan following Fermat's spiral.\n\n Args:\n *args: pairs of device / start position / end position / steps arguments\n relative: Start from an absolute or relative position\n burst: number of acquisition per point\n optim_trajectory: routine used for the trajectory optimization, e.g. 'corridor'. Default: None\n\n Returns:\n\n Examples:\n >>> scans.fermat_scan(dev.motor1, -5, 5, dev.motor2, -5, 5, step=0.5, exp_time=0.1, relative=True, optim_trajectory=\"corridor\")\n\n ",
|
|
}
|
|
scan_name = "fermat_scan"
|
|
obj = ScanObject(scan_name, scan_info, bec_client)
|
|
bec_client.scans._scan_group = "group_id"
|
|
dev = bec_client.device_manager.devices
|
|
with mock.patch.object(bec_client, "alarm_handler"):
|
|
with mock.patch("bec_lib.scan_manager.ScanReport.from_request") as scan_report:
|
|
with mock.patch.object(bec_client, "get_global_var", return_value="test_sample"):
|
|
obj._run(dev.samx, -5, 5, dev.samy, -5, 5, step=0.5, exp_time=0.1, relative=False)
|
|
assert scan_report.call_args.args[0].metadata["queue_group"] == "group_id"
|
|
|
|
|
|
def test_scan_object_receives_scan_def_id(bec_client):
|
|
scan_info = {
|
|
"class": "FermatSpiralScan",
|
|
"arg_input": ["device", "float", "float"],
|
|
"required_kwargs": ["step", "relative"],
|
|
"arg_bundle_size": 3,
|
|
"scan_report_hint": "table",
|
|
"doc": "\n A scan following Fermat's spiral.\n\n Args:\n *args: pairs of device / start position / end position / steps arguments\n relative: Start from an absolute or relative position\n burst: number of acquisition per point\n optim_trajectory: routine used for the trajectory optimization, e.g. 'corridor'. Default: None\n\n Returns:\n\n Examples:\n >>> scans.fermat_scan(dev.motor1, -5, 5, dev.motor2, -5, 5, step=0.5, exp_time=0.1, relative=True, optim_trajectory=\"corridor\")\n\n ",
|
|
}
|
|
scan_name = "fermat_scan"
|
|
obj = ScanObject(scan_name, scan_info, bec_client)
|
|
bec_client.scans._scan_def_id = "scan_def_id"
|
|
dev = bec_client.device_manager.devices
|
|
with mock.patch.object(bec_client, "alarm_handler"):
|
|
with mock.patch("bec_lib.scan_manager.ScanReport.from_request") as scan_report:
|
|
with mock.patch.object(bec_client, "get_global_var", return_value="test_sample"):
|
|
obj._run(dev.samx, -5, 5, dev.samy, -5, 5, step=0.5, exp_time=0.1, relative=False)
|
|
assert scan_report.call_args.args[0].metadata["scan_def_id"] == "scan_def_id"
|
|
|
|
|
|
def test_scan_object_receives_dataset_id_on_hold(bec_client):
|
|
scan_info = {
|
|
"class": "FermatSpiralScan",
|
|
"arg_input": ["device", "float", "float"],
|
|
"required_kwargs": ["step", "relative"],
|
|
"arg_bundle_size": 3,
|
|
"scan_report_hint": "table",
|
|
"doc": "\n A scan following Fermat's spiral.\n\n Args:\n *args: pairs of device / start position / end position / steps arguments\n relative: Start from an absolute or relative position\n burst: number of acquisition per point\n optim_trajectory: routine used for the trajectory optimization, e.g. 'corridor'. Default: None\n\n Returns:\n\n Examples:\n >>> scans.fermat_scan(dev.motor1, -5, 5, dev.motor2, -5, 5, step=0.5, exp_time=0.1, relative=True, optim_trajectory=\"corridor\")\n\n ",
|
|
}
|
|
scan_name = "fermat_scan"
|
|
obj = ScanObject(scan_name, scan_info, bec_client)
|
|
bec_client.scans._dataset_id_on_hold = "dataset_id_on_hold"
|
|
dev = bec_client.device_manager.devices
|
|
with mock.patch.object(bec_client, "alarm_handler"):
|
|
with mock.patch("bec_lib.scan_manager.ScanReport.from_request") as scan_report:
|
|
with mock.patch.object(bec_client, "get_global_var", return_value="test_sample"):
|
|
obj._run(dev.samx, -5, 5, dev.samy, -5, 5, step=0.5, exp_time=0.1, relative=False)
|
|
assert (
|
|
scan_report.call_args.args[0].metadata["dataset_id_on_hold"]
|
|
== "dataset_id_on_hold"
|
|
)
|