Fix deprecated locations from ophyd_async

Author: EmsArnold

tests/unit_tests/test_plans.py

@@ -7,8 +7,9 @@
 import pytest
 from bluesky import RunEngine
 from dodal.devices.motors import XYZStage
+from ophyd_async.core import callback_on_mock_put, set_mock_value
 from ophyd_async.epics.adaravis import AravisDetector
-from ophyd_async.testing import assert_emitted, callback_on_mock_put, set_mock_value
+from ophyd_async.testing import assert_emitted
 from scanspec.specs import Line
 
 from test_rig_bluesky.plans import (