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(