test no root file case

Author: RemDelaporteMathurin

libra_toolbox/neutron_detection/activation_foils/compass.py

@@ -246,16 +246,19 @@ def from_directory(cls, source_dir: str, name: str) -> "Measurement":
         detectors = [Detector(channel_nb=nb) for nb in time_values.keys()]
 
         # Get live and real count times
-        root_filename = glob.glob(os.path.join(source_dir, "*.root"))[0]
-        if not os.path.isfile(root_filename):
+        all_root_filenames = glob.glob(os.path.join(source_dir, "*.root"))
+        if len(all_root_filenames) == 1:
+            root_filename = all_root_filenames[0]
+        else:
+            root_filename = None
             print("No root file found, assuming all counts are live")
 
         for detector in detectors:
             detector.events = np.column_stack(
                 (time_values[detector.channel_nb], energy_values[detector.channel_nb])
             )
 
-            if os.path.isfile(root_filename):
+            if root_filename:
                 live_count_time, real_count_time = get_live_time_from_root(
                     root_filename, detector.channel_nb
                 )

test/neutron_detection/compass_test_data/complete_measurement_no_root/data/Data_CH1@V1725_292_Co60_0_872uCi_19Mar2014_250318_run2.csv

@@ -0,0 +1,11 @@
+BOARD;CHANNEL;TIMETAG;ENERGY;ENERGYSHORT;FLAGS
+0;5;234859459;2;2;0x4000
+0;5;421999310;0;1;0x4000
+0;5;535148093;1237;810;0x4000
+0;5;1623550122;589;396;0x4000
+0;5;5997211248;375;251;0x4000
+0;5;6685836624;515;340;0x4000
+0;5;11116032249;568;380;0x4000
+0;5;11281099382;1;0;0x4000
+0;5;12783039350;5;0;0x4000
+0;5;18306299412;2;0;0x4000

test/neutron_detection/compass_test_data/complete_measurement_no_root/data/Data_CH1@V1725_292_Co60_0_872uCi_19Mar2014_250318_run2_1.csv

@@ -0,0 +1,4 @@
+0;5;234859459;2;2;0x4000
+0;5;421999310;0;1;0x4000
+0;5;535148093;1237;810;0x4000
+0;5;1623550122;589;396;0x4000

test/neutron_detection/compass_test_data/complete_measurement_no_root/run.info

@@ -0,0 +1,31 @@
+id=Co60_0_872uCi_19Marc2014_250319_run3
+time.start=2025/03/19 09:47:46.724-0400
+time.stop=2025/03/19 09:53:05.651-0400
+time.real=00:05:18
+board.0-14-292.readout.rate=51.149 kb/s
+board.0-14-292.4.rejections.singles=0.0
+board.0-14-292.4.rejections.pileup=0.0
+board.0-14-292.4.rejections.saturation=1301.87
+board.0-14-292.4.rejections.energy=0.0
+board.0-14-292.4.rejections.psd=0.0
+board.0-14-292.4.rejections.timedistribution=0.0
+board.0-14-292.4.throughput=2627.31
+board.0-14-292.4.icr=3059.24
+board.0-14-292.4.ocr=1293.91
+board.0-14-292.4.calibration.energy.c0=0.0
+board.0-14-292.4.calibration.energy.c1=1.0
+board.0-14-292.4.calibration.energy.c2=0.0
+board.0-14-292.4.calibration.energy.uom=keV
+board.0-14-292.5.rejections.singles=0.0
+board.0-14-292.5.rejections.pileup=0.0
+board.0-14-292.5.rejections.saturation=717.247
+board.0-14-292.5.rejections.energy=0.0
+board.0-14-292.5.rejections.psd=0.0
+board.0-14-292.5.rejections.timedistribution=0.0
+board.0-14-292.5.throughput=1694.65
+board.0-14-292.5.icr=1703.71
+board.0-14-292.5.ocr=984.476
+board.0-14-292.5.calibration.energy.c0=0.0
+board.0-14-292.5.calibration.energy.c1=1.0
+board.0-14-292.5.calibration.energy.c2=0.0
+board.0-14-292.5.calibration.energy.uom=keV

test/neutron_detection/test_compass.py

@@ -277,14 +277,20 @@ def test_get_live_time_from_root(root_filename, channel, live_time, real_time):
     assert real_time_out == real_time
 
 
-def test_measurement_object_from_directory():
+@pytest.mark.parametrize("no_root", [True, False])
+def test_measurement_object_from_directory(no_root):
     """
     Test the Measurement object creation from a directory.
     """
-
-    test_directory = (
-        Path(__file__).parent / "compass_test_data/complete_measurement/data"
-    )
+    if no_root:
+        test_directory = (
+            Path(__file__).parent
+            / "compass_test_data/complete_measurement_no_root/data"
+        )
+    else:
+        test_directory = (
+            Path(__file__).parent / "compass_test_data/complete_measurement/data"
+        )
 
     measurement = compass.Measurement.from_directory(test_directory, name="test")