Remove flyer from pre-declare and simplify test

bluesky · Mar 25, 2024 · ff62d2c · ff62d2c
1 parent 94469e2
commit ff62d2c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/core/test_flyer.py b/tests/core/test_flyer.py
@@ -8,6 +8,7 @@
 from bluesky.protocols import Descriptor, StreamAsset
 from bluesky.run_engine import RunEngine
 from event_model import ComposeStreamResourceBundle, compose_stream_resource
+
 from ophyd_async.core import (
     DEFAULT_TIMEOUT,
     DetectorControl,
@@ -174,7 +175,7 @@ def flying_plan():
             detector.controller.disarm.assert_called_once  # type: ignore
 
         yield from bps.open_run()
-        yield from bps.declare_stream(flyer, detector, name="main_stream")
+        yield from bps.declare_stream(*detector_list, name="main_stream", collect=True)
 
         yield from bps.kickoff(flyer)
         for detector in detector_list:
@@ -199,9 +200,8 @@ def flying_plan():
                 done = True
             yield from bps.collect(
                 *detector_list,
-                stream=True,
                 return_payload=False,
-                name="main_stream",
+                # name="main_stream",
             )
             yield from bps.sleep(0.01)
         yield from bps.wait(group="complete")