|
3 | 3 | from dodal.common.beamlines.beamline_utils import device_factory |
4 | 4 | from dodal.utils import OphydV1Device, OphydV2Device |
5 | 5 | from ophyd_async.core import DEFAULT_TIMEOUT, LazyMock, StandardReadable |
6 | | -from ophyd_async.epics.motor import Motor |
| 6 | +from ophyd_async.sim import SimMotor |
7 | 7 |
|
8 | 8 |
|
9 | 9 | def fake_motor_bundle_b( |
10 | | - fake_motor_x: Motor, |
11 | | - fake_motor_y: Motor, |
12 | | -) -> Motor: |
13 | | - return Motor("BAR:", "motor_bundle_b") |
| 10 | + fake_motor_x: SimMotor, |
| 11 | + fake_motor_y: SimMotor, |
| 12 | +) -> SimMotor: |
| 13 | + return SimMotor("motor_bundle_b") |
14 | 14 |
|
15 | 15 |
|
16 | | -def fake_motor_x() -> Motor: |
17 | | - return Motor("FOO:", "motor_x") |
| 16 | +def fake_motor_x() -> SimMotor: |
| 17 | + return SimMotor("motor_x") |
18 | 18 |
|
19 | 19 |
|
20 | 20 | class DeviceA(StandardReadable): |
21 | 21 | def __init__(self, name: str = "") -> None: |
22 | 22 | with self.add_children_as_readables(): |
23 | | - self.motor = Motor("X:SIZE") |
| 23 | + self.motor = SimMotor("X:SIZE") |
24 | 24 | super().__init__(name) |
25 | 25 |
|
26 | 26 |
|
@@ -56,15 +56,15 @@ def ophyd_async_device() -> UnconnectableOphydAsyncDevice: |
56 | 56 | return UnconnectableOphydAsyncDevice(name="ophyd_async_device") |
57 | 57 |
|
58 | 58 |
|
59 | | -def fake_motor_y() -> Motor: |
60 | | - return Motor("BAZ:", "motor_y") |
| 59 | +def fake_motor_y() -> SimMotor: |
| 60 | + return SimMotor("motor_y") |
61 | 61 |
|
62 | 62 |
|
63 | 63 | def fake_motor_bundle_a( |
64 | | - fake_motor_x: Motor, |
65 | | - fake_motor_y: Motor, |
66 | | -) -> Motor: |
67 | | - return Motor("QUX:", "motor_bundle_a") |
| 64 | + fake_motor_x: SimMotor, |
| 65 | + fake_motor_y: SimMotor, |
| 66 | +) -> SimMotor: |
| 67 | + return SimMotor("motor_bundle_a") |
68 | 68 |
|
69 | 69 |
|
70 | 70 | def wrong_return_type() -> int: |
|
0 commit comments