usov_i-org/bec
1
0
Fork 0
mirror of https://github.com/ivan-usov-org/bec.git synced 2025-04-22 02:20:02 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

tests: updated test_get_devices_from_instruction

Browse Source
This commit is contained in:
stalbe_j 2023-02-03 14:10:32 +01:00
parent 888f1a5432
commit c6029e89e4
1 changed files with 43 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
scan_server/tests/test_scan_worker.py
View File

@ -37,13 +37,53 @@ def get_scan_worker() -> ScanWorker:
),
["samx", "samy"],
),
(
BECMessage.DeviceInstructionMessage(
device="",
action="wait",
parameter={"type": "move", "group": "scan_motor", "wait_group": "scan_motor"},
metadata={"stream": "primary", "DIID": 3},
),
["samx", "samy"],
),
(
BECMessage.DeviceInstructionMessage(
device="",
action="wait",
parameter={"type": "move", "group": "primary", "wait_group": "scan_motor"},
metadata={"stream": "primary", "DIID": 3},
),
["samx", "samy"],
),
(
BECMessage.DeviceInstructionMessage(
device="",
action="wait",
parameter={"type": "move", "group": "nogroup", "wait_group": "scan_motor"},
metadata={"stream": "primary", "DIID": 3},
),
["samx", "samy"],
),
],
)
def test_get_devices_from_instruction(instruction, devices):
worker = get_scan_worker()
assert worker._get_devices_from_instruction(instruction) == [
worker.device_manager.devices[dev] for dev in devices
]
worker.scan_motors = devices
returned_devices = worker._get_devices_from_instruction(instruction)
if not instruction.content.get("device"):
group = instruction.content["parameter"].get("group")
if group == "primary":
assert returned_devices == worker.device_manager.devices.primary_devices(
worker.scan_motors
)
elif group == "scan_motor":
assert returned_devices == devices
else:
assert returned_devices == []
else:
assert returned_devices == [worker.device_manager.devices[dev] for dev in devices]
@pytest.mark.parametrize(