Add ExperimentList.all_same_type().

newsfragments/758.misc

@@ -0,0 +1 @@
+Add `ExperimentList.all_same_type()` to quickly check if experiments are all the same type.

src/dxtbx/model/__init__.py

@@ -619,6 +619,16 @@ def all_laue(self):
         """Check if all the experiments are Laue experiments"""
         return all(exp.get_type() == ExperimentType.LAUE for exp in self)
 
+    def all_same_type(self):
+        """Check if all experiments are the same type"""
+        if len(self) <= 1:
+            return True
+        expt_type = self[0].get_type()
+        for i in range(1, len(self)):
+            if self[i].get_type() != expt_type:
+                return False
+        return True
+
     def to_dict(self):
         """Serialize the experiment list to dictionary."""
 

tests/model/test_experiment_list.py

@@ -1205,15 +1205,19 @@ def test_from_templates(dials_data):
 
 def test_experiment_list_all():
     experiments = ExperimentList()
+    assert experiments.all_same_type()
     for i in range(3):
         experiments.append(Experiment())
 
     assert experiments.all_stills()
+    assert experiments.all_same_type()
     experiments[0].goniometer = Goniometer()
     assert experiments.all_stills()
+    assert experiments.all_same_type()
     experiments[1].goniometer = Goniometer()
     experiments[2].goniometer = Goniometer()
     assert experiments.all_stills()
+    assert experiments.all_same_type()
 
     experiments[0].beam = BeamFactory.make_polychromatic_beam(
         direction=(0, 0, -1),
@@ -1222,6 +1226,7 @@ def test_experiment_list_all():
         wavelength_range=(1, 10),
     )
     assert not experiments.all_stills()
+    assert not experiments.all_same_type()
     experiments[1].beam = BeamFactory.make_polychromatic_beam(
         direction=(0, 0, -1),
         sample_to_source_distance=(100),
@@ -1235,39 +1240,48 @@ def test_experiment_list_all():
         wavelength_range=(1, 10),
     )
     assert experiments.all_laue()
+    assert experiments.all_same_type()
 
     experiments[0].beam = Beam()
     assert not experiments.all_laue()
+    assert not experiments.all_same_type()
     experiments[1].beam = Beam()
     experiments[2].beam = Beam()
     assert experiments.all_stills()
+    assert experiments.all_same_type()
 
     experiments[0].scan = Scan((1, 1000), (0, 0.05))
     assert not experiments.all_stills()
+    assert not experiments.all_same_type()
     experiments[1].scan = Scan((1, 1000), (0, 0.05))
     experiments[2].scan = Scan((1, 1000), (0, 0.05))
     assert experiments.all_rotations()
+    assert experiments.all_same_type()
 
     experiments[0].scan = ScanFactory.make_scan_from_properties(
         (1, 10), properties={"time_of_flight": list(range(10))}
     )
     assert not experiments.all_rotations()
+    assert not experiments.all_same_type()
     experiments[1].scan = ScanFactory.make_scan_from_properties(
         (1, 10), properties={"time_of_flight": list(range(10))}
     )
     experiments[2].scan = ScanFactory.make_scan_from_properties(
         (1, 10), properties={"time_of_flight": list(range(10))}
     )
     assert experiments.all_tof()
+    assert experiments.all_same_type()
 
     experiments[0].scan = ScanFactory.make_scan_from_properties(
         (1, 10), properties={"other_property": list(range(10))}
     )
     assert not experiments.all_tof()
+    assert not experiments.all_same_type()
     experiments[1].scan = ScanFactory.make_scan_from_properties(
         (1, 10), properties={"other_property": list(range(10))}
     )
     experiments[2].scan = ScanFactory.make_scan_from_properties(
         (1, 10), properties={"other_property": list(range(10))}
     )
     assert experiments.all_stills()
+    assert experiments.all_same_type()