Make plan_stub test dir and move tests into it

bluesky · May 21, 2024 · cb6b852 · cb6b852
1 parent ef98501
commit cb6b852
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/src/ophyd_async/plan_stubs/__init__.py b/src/ophyd_async/plan_stubs/__init__.py
@@ -1,6 +1,7 @@
 from .fly import (
     prepare_static_seq_table_flyer_and_detectors_with_same_trigger,
-    time_resolved_fly_and_collect_with_static_seq_table)
+    time_resolved_fly_and_collect_with_static_seq_table,
+)
 
 __all__ = [
     "time_resolved_fly_and_collect_with_static_seq_table",

diff --git a/src/ophyd_async/plan_stubs/fly.py b/src/ophyd_async/plan_stubs/fly.py
@@ -2,8 +2,7 @@
 
 import bluesky.plan_stubs as bps
 
-from ophyd_async.core.detector import (DetectorTrigger, StandardDetector,
-                                       TriggerInfo)
+from ophyd_async.core.detector import DetectorTrigger, StandardDetector, TriggerInfo
 from ophyd_async.core.flyer import HardwareTriggeredFlyable
 from ophyd_async.core.utils import in_micros
 from ophyd_async.panda._table import SeqTable, SeqTableRow, seq_table_from_rows

diff --git a/tests/test_flyer_with_panda.py → tests/plan_stubs/test_fly.py b/tests/test_flyer_with_panda.py → tests/plan_stubs/test_fly.py
@@ -17,8 +17,9 @@
 from ophyd_async.epics.signal.signal import epics_signal_rw
 from ophyd_async.panda import CommonPandaBlocks
 from ophyd_async.panda._trigger import StaticSeqTableTriggerLogic
-from ophyd_async.plan_stubs import \
-    prepare_static_seq_table_flyer_and_detectors_with_same_trigger
+from ophyd_async.plan_stubs import (
+    prepare_static_seq_table_flyer_and_detectors_with_same_trigger,
+    time_resolved_fly_and_collect_with_static_seq_table)
 
 
 class DummyWriter(DetectorWriter):