Refactor test

DiamondLightSource · Sep 18, 2024 · 92401d8 · 92401d8
1 parent 16679ff
commit 92401d8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit_tests/hyperion/experiment_plans/test_rotation_scan_plan.py b/tests/unit_tests/hyperion/experiment_plans/test_rotation_scan_plan.py
@@ -238,9 +238,9 @@ async def test_rotation_plan_moves_aperture_correctly(
     aperture_scatterguard: ApertureScatterguard = (
         run_full_rotation_plan.aperture_scatterguard
     )
-    assert (
-        await aperture_scatterguard.selected_aperture.get_value() == ApertureValue.SMALL
-    )
+    reading = await aperture_scatterguard.read()
+    aperture_state = reading[aperture_scatterguard.selected_aperture.name]["value"]
+    assert aperture_state == ApertureValue.SMALL
 
 
 async def test_rotation_plan_smargon_doesnt_move_xyz_if_not_given_in_params(