From c47918d6e7ff41721aa4fa67043ff6cd1aeee2c7 Mon Sep 17 00:00:00 2001 From: appel_c Date: Wed, 28 Aug 2024 14:43:40 +0200 Subject: [PATCH] feat: add dual patch pvs to ophyd_devices --- ophyd_devices/tests/utils.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/ophyd_devices/tests/utils.py b/ophyd_devices/tests/utils.py index 1e53850..1a9e24f 100644 --- a/ophyd_devices/tests/utils.py +++ b/ophyd_devices/tests/utils.py @@ -1,6 +1,17 @@ from unittest import mock +def patch_dual_pvs(device): + device.wait_for_connection(all_signals=True) + for walk in device.walk_signals(): + if not hasattr(walk.item, "_read_pv"): + continue + if not hasattr(walk.item, "_write_pv"): + continue + if walk.item._read_pv.pvname.endswith("_RBV"): + walk.item._read_pv = walk.item._write_pv + + class SocketMock: """Socket Mock. Used for testing"""