fix the tests imports

DiamondLightSource · Jan 16, 2025 · 734f958 · 734f958
1 parent bc3d535
commit 734f958
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/devices/i18/test_kb_mirror.py b/tests/devices/i18/test_kb_mirror.py
@@ -1,7 +1,8 @@
 from unittest.mock import ANY
 
 import pytest
-from ophyd_async.core import DeviceCollector, set_mock_value
+from ophyd_async.core import DeviceCollector
+from ophyd_async.testing import set_mock_value
 
 from dodal.devices.i18.KBMirror import KBMirror, XYPosition
 

diff --git a/tests/devices/i18/test_table.py b/tests/devices/i18/test_table.py
@@ -1,7 +1,8 @@
 from unittest.mock import ANY
 
 import pytest
-from ophyd_async.core import DeviceCollector, set_mock_value
+from ophyd_async.core import DeviceCollector
+from ophyd_async.testing import set_mock_value
 
 from dodal.devices.i18.table import Four_D_Position, Table