Save all files from previous change

DiamondLightSource · Nov 2, 2023 · fe5a622 · fe5a622
1 parent 33e14e7
commit fe5a622
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/mx_bluesky/I24/serial/setup_beamline/__init__.py b/src/mx_bluesky/I24/serial/setup_beamline/__init__.py
@@ -1,6 +1,6 @@
 from . import pv, setup_beamline
 from .ca import caget, cagetstring, caput
-from .pv_abstract import Detector, Eiger, Pilatus
+from .pv_abstract import Detector, Eiger, ExperimentType, Extruder, FixedTarget, Pilatus
 
 __all__ = [
     "caget",
@@ -11,4 +11,7 @@
     "Pilatus",
     "pv",
     "setup_beamline",
+    "Extruder",
+    "FixedTarget",
+    "ExperimentType",
 ]
diff --git a/tests/I24/serial/setup_beamline/test_pv_abstract.py b/tests/I24/serial/setup_beamline/test_pv_abstract.py
@@ -1,4 +1,4 @@
-from mx_bluesky.I24.serial.setup_beamline import Eiger, Pilatus
+from mx_bluesky.I24.serial.setup_beamline import Eiger, Extruder, FixedTarget, Pilatus
 
 
 def test_eiger():
@@ -9,3 +9,8 @@ def test_eiger():
 def test_pilatus():
     pil = Pilatus()
     assert pil.image_size_mm == (423.636, 434.644)
+
+
+def test_experiment_types():
+    assert Extruder().expt_type.value == "Serial Jet"
+    assert FixedTarget().expt_type.value == "Serial Fixed"