Unit tests for multiprocessing of cslightcrv and csphasecrv

Luigi Tibaldo
1 parent abb29aa8
Showing 2 changed files with 56 additions and 0 deletions
test/test_cslightcrv.py
test/test_csphasecrv.py
@@ -272,6 +272,35 @@ class Test(test):
         # Check light curve
         self._check_light_curve('cslightcrv_py5.fits', 2)
  
+        # Set-up cslightcrv without multiprocessing
+        lcrv = cscripts.cslightcrv()
+        lcrv['inobs'] = self._events
+        lcrv['inmodel'] = self._model
+        lcrv['srcname'] = 'Crab'
+        lcrv['caldb'] = self._caldb
+        lcrv['irf'] = self._irf
+        lcrv['tbinalg'] = 'LIN'
+        lcrv['tmin'] = '2020-01-01T00:00:00'
+        lcrv['tmax'] = '2020-01-01T00:05:00'
+        lcrv['tbins'] = 3
+        lcrv['method'] = '3D'
+        lcrv['enumbins'] = 0
+        lcrv['emin'] = 0.1
+        lcrv['emax'] = 100.0
+        lcrv['outfile'] = 'cslightcrv_py6.fits'
+        lcrv['logfile'] = 'cslightcrv_py6.log'
+        lcrv['chatter'] = 2
+        lcrv['publish'] = True
+        lcrv['nthreads'] = 1
+
+        # Run cslightcrv script and save light curve
+        lcrv.logFileOpen()  # Make sure we get a log file
+        lcrv.run()
+        lcrv.save()
+
+        # Check light curve
+        self._check_light_curve('cslightcrv_py6.fits', 3)
+
         # Return
         return
  
@@ -268,6 +268,33 @@ class Test(test):
         # Check phase curve
         self._check_phase_curve('csphasecrv_py5.fits', 2)
  
+        # Set-up csphasecrv without multiprocessing
+        pcrv = cscripts.csphasecrv()
+        pcrv['inobs']    = self._phased_events
+        pcrv['inmodel']  = self._model
+        pcrv['srcname']  = 'Crab'
+        pcrv['caldb']    = self._caldb
+        pcrv['irf']      = self._irf
+        pcrv['phbinalg'] = 'LIN'
+        pcrv['phbins']   = 5
+        pcrv['method']   = '3D'
+        pcrv['enumbins'] = 0
+        pcrv['emin']     = 0.1
+        pcrv['emax']     = 100.0
+        pcrv['outfile']  = 'csphasecrv_py6.fits'
+        pcrv['logfile']  = 'csphasecrv_py6.log'
+        pcrv['chatter']  = 2
+        pcrv['publish']  = True
+        pcrv['nthreads'] = 1
+
+        # Run csphasecrv script and save phase curve
+        pcrv.logFileOpen()   # Make sure we get a log file
+        pcrv.run()
+        pcrv.save()
+
+        # Check phase curve
+        self._check_phase_curve('csphasecrv_py6.fits', 5)
+
         # Return
         return