From 734f9580b2e11d899f764f656cfcac173b283c9e Mon Sep 17 00:00:00 2001 From: Stanislaw Malinowski Date: Tue, 14 Jan 2025 17:02:44 +0000 Subject: [PATCH] fix the tests imports --- tests/devices/i18/test_kb_mirror.py | 3 ++- tests/devices/i18/test_table.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/devices/i18/test_kb_mirror.py b/tests/devices/i18/test_kb_mirror.py index 1eff890564..c66486cf40 100644 --- 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 index 900339894a..80f816312c 100644 --- 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