(#601) Fix failing dual_backlight unit test, mypy errors

DiamondLightSource · Jul 2, 2024 · cc16c11 · cc16c11
1 parent 6ede9af
commit cc16c11
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tests/devices/unit_tests/i24/test_dual_backlight.py b/tests/devices/unit_tests/i24/test_dual_backlight.py
@@ -52,7 +52,7 @@ async def test_when_backlight_out_it_switches_off(
 async def test_when_backlight_not_out_it_switches_on(
     fake_backlight: DualBacklight, RE: RunEngine
 ):
-    RE(bps.abs_set(fake_backlight, "OAV2"))
+    RE(bps.abs_set(fake_backlight, "OAV2", wait=True))
     assert await fake_backlight.backlight_state.get_value() == "ON"
 
 

diff --git a/tests/devices/unit_tests/test_utils.py b/tests/devices/unit_tests/test_utils.py
@@ -5,7 +5,6 @@
 import pytest
 from bluesky.protocols import Status
 from ophyd.sim import NullStatus
-from ophyd.status import Status
 from ophyd.utils.errors import StatusTimeoutError, WaitTimeoutError
 from ophyd_async.core import AsyncStatus, get_mock_put, set_mock_value