diff --git a/tests/unit_tests/hyperion/experiment_plans/conftest.py b/tests/unit_tests/hyperion/experiment_plans/conftest.py index 2a07c75da..15f7bdaf6 100644 --- a/tests/unit_tests/hyperion/experiment_plans/conftest.py +++ b/tests/unit_tests/hyperion/experiment_plans/conftest.py @@ -235,7 +235,7 @@ def robot_load_composite( sample_shutter, zebra, panda, - panda_fast_grid_scan + panda_fast_grid_scan, ) -> RobotLoadThenCentreComposite: set_mock_value(dcm.energy_in_kev.user_readback, 11.105) aperture_scatterguard = aperture_scatterguard diff --git a/tests/unit_tests/hyperion/experiment_plans/test_load_centre_collect_full_plan.py b/tests/unit_tests/hyperion/experiment_plans/test_load_centre_collect_full_plan.py index b33e6483f..8bc05b9f1 100644 --- a/tests/unit_tests/hyperion/experiment_plans/test_load_centre_collect_full_plan.py +++ b/tests/unit_tests/hyperion/experiment_plans/test_load_centre_collect_full_plan.py @@ -184,7 +184,8 @@ def test_load_centre_collect_full_plan_happy_path_invokes_all_steps( return_value=iter([]), ) @patch( - "mx_bluesky.hyperion.experiment_plans.robot_load_and_change_energy.set_energy_plan", new=MagicMock() + "mx_bluesky.hyperion.experiment_plans.robot_load_and_change_energy.set_energy_plan", + new=MagicMock(), ) def test_load_centre_collect_full_plan_skips_collect_if_pin_tip_not_found( mock_rotation_scan: MagicMock, @@ -212,7 +213,8 @@ def test_load_centre_collect_full_plan_skips_collect_if_pin_tip_not_found( return_value=iter([]), ) @patch( - "mx_bluesky.hyperion.experiment_plans.robot_load_and_change_energy.set_energy_plan", new=MagicMock() + "mx_bluesky.hyperion.experiment_plans.robot_load_and_change_energy.set_energy_plan", + new=MagicMock(), ) def test_load_centre_collect_full_plan_skips_collect_if_no_diffraction( mock_rotation_scan: MagicMock,