bec/ophyd_devices
0
0
Fork 0
mirror of https://github.com/bec-project/ophyd_devices.git synced 2026-01-30 20:54:55 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix(psi-motor): Add new user motor imlementation that checks if the IOC is enabled.

Browse Source
This commit is contained in:
appel_c
2025-12-02 11:11:38 +01:00
committed by Christian Appel
parent fa51ffe33a
commit 71a9b3c103
4 changed files with 74 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_psi_motors.py
View File

@@ -3,13 +3,15 @@ PSI motor integration from the ophyd_devices.devices.psi_motor module."""
from __future__ import annotations
import threading
from unittest import mock
import ophyd
import pytest
from ophyd_devices.devices.psi_motor import EpicsMotor, EpicsMotorEC, SpmgStates
from ophyd_devices.tests.utils import patched_device
from ophyd_devices.devices.epics_motor_ex import EpicsMotorEx
from ophyd_devices.devices.psi_motor import EpicsMotor, EpicsMotorEC, EpicsUserMotors, SpmgStates
from ophyd_devices.tests.utils import MockPV, patched_device
@pytest.fixture(scope="function")
@@ -160,3 +162,17 @@ def test_epics_motor_high_limit_switch_raises(mock_epics_motor):
motor.high_limit_switch._read_pv.mock_data = 1 # Simulate high limit switch active
with pytest.raises(ophyd.utils.LimitError):
motor.move(15)
def test_epics_vme_user_motor():
"""Test extended VME based user motor implementation EpicsUserMotors."""
with mock.patch.object(ophyd, "cl") as mock_cl:
mock_cl.get_pv = MockPV
mock_cl.thread_class = threading.Thread
device = EpicsUserMotors(name="test_motor_ex", prefix="SIM:MOTOR:EX")
# Should raise
with pytest.raises(RuntimeError):
device.wait_for_connection(all_signals=True)
# Should not raise
device._ioc_enable._read_pv.mock_data = "Enable"
device.wait_for_connection(all_signals=True)