bec/ophyd_devices
0
0
Fork 0
mirror of https://github.com/bec-project/ophyd_devices.git synced 2026-02-06 07:08:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve device mocking for tests

Browse Source
This commit is contained in:
perl_d
2025-11-17 13:02:59 +01:00
committed by David Perl
parent 011b68f3dc
commit eceab997b8
5 changed files with 47 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tests/test_psi_motors.py
View File

@@ -3,42 +3,27 @@ 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 MockPV, patch_dual_pvs
from ophyd_devices.tests.utils import patched_device
@pytest.fixture(scope="function")
def mock_epics_motor():
"""Fixture to create a mock EpicsMotor instance."""
name = "test_motor"
prefix = "SIM:MOTOR"
with mock.patch.object(ophyd, "cl") as mock_cl:
mock_cl.get_pv = MockPV
mock_cl.thread_class = threading.Thread
motor = EpicsMotor(name=name, prefix=prefix)
motor.wait_for_connection()
patch_dual_pvs(motor)
with patched_device(EpicsMotor, name="test_motor", prefix="SIM:MOTOR") as motor:
yield motor
@pytest.fixture(scope="function")
def mock_epics_motor_ec():
"""Fixture to create a mock EpicsMotorEC instance."""
name = "test_motor_ec"
prefix = "SIM:MOTOR:EC"
with mock.patch.object(ophyd, "cl") as mock_cl:
mock_cl.get_pv = MockPV
mock_cl.thread_class = threading.Thread
motor_ec = EpicsMotorEC(name=name, prefix=prefix)
motor_ec.wait_for_connection()
patch_dual_pvs(motor_ec)
yield motor_ec
with patched_device(EpicsMotorEC, name="test_motor_ec", prefix="SIM:MOTOR:EC") as motor:
yield motor
def test_epics_motor_limits_raise(mock_epics_motor):