Adding more CI tests for the remaining DQM modules

cta-observatory · Jul 22, 2024 · 377e43a · 377e43a
1 parent 904abab
commit 377e43a
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 6 deletions.
diff --git a/src/nectarchain/dqm/tests/test_camera_monitoring.py b/src/nectarchain/dqm/tests/test_camera_monitoring.py
@@ -1,6 +1,8 @@
+from astropy import time as astropytime
 from ctapipe.io import EventSource
 from ctapipe.utils import get_dataset_path
 from ctapipe_io_nectarcam.constants import HIGH_GAIN
+from tqdm import tqdm
 from traitlets.config import Config
 
 from nectarchain.dqm.camera_monitoring import CameraMonitoring
@@ -32,6 +34,13 @@ def test_camera_monitoring(self):
         reader1 = EventSource(input_url=path, config=config, max_events=1)
 
         Pix, Samp = CameraMonitoring(HIGH_GAIN).DefineForRun(reader1)
-        print(Pix, Samp)
-        # self.assertEqual(Pix + Samp, 1915)
+
+        CameraMonitoring(HIGH_GAIN).ConfigureForRun(path, Pix, Samp, reader1)
+
+        for evt in tqdm(reader1, total=1):
+            run_start1 = evt.nectarcam.tel[0].svc.date
+            SqlFileDate = astropytime.Time(run_start1, format="unix").iso.split(" ")[0]
+        # print("SqlFileDate", SqlFileDate)
+        # CameraMonitoring(HIGH_GAIN).FinishRun()
         assert Pix + Samp == 1915
+        assert SqlFileDate == "2023-01-23"
diff --git a/src/nectarchain/dqm/tests/test_mean_camera_display.py b/src/nectarchain/dqm/tests/test_mean_camera_display.py
@@ -1,6 +1,7 @@
 from ctapipe.io import EventSource
 from ctapipe.utils import get_dataset_path
 from ctapipe_io_nectarcam.constants import HIGH_GAIN
+from tqdm import tqdm
 from traitlets.config import Config
 
 from nectarchain.dqm.mean_camera_display import MeanCameraDisplayHighLowGain
@@ -32,6 +33,15 @@ def test_mean_camera_display(self):
         reader1 = EventSource(input_url=path, config=config, max_events=1)
 
         Pix, Samp = MeanCameraDisplayHighLowGain(HIGH_GAIN).DefineForRun(reader1)
-        print(Pix, Samp)
-        # self.assertEqual(Pix + Samp, 1915)
+
+        MeanCameraDisplayHighLowGain(HIGH_GAIN).ConfigureForRun(
+            path, Pix, Samp, reader1
+        )
+
+        for evt in tqdm(reader1, total=1):
+            # MeanCameraDisplayHighLowGain(HIGH_GAIN).ProcessEvent(evt, noped = False)
+            CameraAverage = evt.r0.tel[0].waveform[HIGH_GAIN].sum(axis=1)
+
+        # MeanCameraDisplayHighLowGain(HIGH_GAIN).FinishRun()
         assert Pix + Samp == 1915
+        assert CameraAverage.sum(axis=0) == 109723121
diff --git a/src/nectarchain/dqm/tests/test_trigger_statistics.py b/src/nectarchain/dqm/tests/test_trigger_statistics.py
@@ -1,6 +1,7 @@
 from ctapipe.io import EventSource
 from ctapipe.utils import get_dataset_path
 from ctapipe_io_nectarcam.constants import HIGH_GAIN
+from tqdm import tqdm
 from traitlets.config import Config
 
 from nectarchain.dqm.trigger_statistics import TriggerStatistics
@@ -32,6 +33,13 @@ def test_trigger_statistics(self):
         reader1 = EventSource(input_url=path, config=config, max_events=1)
 
         Pix, Samp = TriggerStatistics(HIGH_GAIN).DefineForRun(reader1)
-        print(Pix, Samp)
-        # self.assertEqual(Pix + Samp, 1915)
+
+        TriggerStatistics(HIGH_GAIN).ConfigureForRun(path, Pix, Samp, reader1)
+
+        for evt in tqdm(reader1, total=1):
+            time = evt.trigger.time.value
+            TriggerStatistics(HIGH_GAIN).ProcessEvent(evt, noped=False)
+
+        # TriggerStatistics(HIGH_GAIN).FinishRun()
         assert Pix + Samp == 1915
+        assert time == 1674462932.6398556