Implement unbinned pipeline tests

Christoph Deil
1 parent b7096783
Showing 3 changed files with 127 additions and 16 deletions
cscripts/__init__.py
cscripts/tests/test_binned.py
cscripts/tests/test_unbinned.py
@@ -34,19 +34,17 @@ from .cstsdist   import cstsdist
 from . import obsutils
  
  
-def test():
+def test(verbosity=2):
     """Run ctools tests.
     """
     import unittest
     from .tests.test_unbinned import TestUnbinnedAnalysis
-    from .tests.test_binned import TestBinnedAnalysis
-    # suite = unittest.TestLoader().loadTestsFromTestCase(TestUnbinnedAnalysis)
-    # unittest.TextTestRunner(verbosity=2).run(suite)
+
     test_suite = unittest.TestSuite()
  
     # Test cases must be listed manually here ...
     # we currently don't use unittest for test collection
     test_suite.addTest(unittest.makeSuite(TestUnbinnedAnalysis))
-    test_suite.addTest(unittest.makeSuite(TestBinnedAnalysis))
  
-    unittest.TextTestRunner(verbosity=2).run(test_suite)
+    test_runner = unittest.TextTestRunner(verbosity=verbosity)
+    test_runner.run(test_suite)
-"""Test binned analysis.
-"""
-import unittest
-
-class TestBinnedAnalysis(unittest.TestCase):
-
-    def test_it(self):
-        self.assertEqual('foo'.upper(), 'FOO')
 """Test unbinned analysis.
+
+TODO: This is basically a duplicate of `test/test_pipelines.py`
+      The duplication should be avoided somehow.
+
 """
+import os
+from tempfile import mkdtemp
 import unittest
+import ctools
+
  
 class TestUnbinnedAnalysis(unittest.TestCase):
+    """Test a simple unbinned analysis pipeline.
+    """
+
+    def setUp(self):
+        gammalib_dir = os.environ['GAMMALIB']
+        self.model_name = os.path.join(gammalib_dir, "share/models/crab.xml")
+        self.workdir = mkdtemp()
+
+    def test_unbinned_fits(self):
+        """
+        Test unbinned pipeline with FITS file saving.
+        """
+        # Set script parameters
+        model_name           = self.model_name
+        events_name          = os.path.join(self.workdir, "events.fits")
+        selected_events_name = os.path.join(self.workdir, "selected_events.fits")
+        result_name          = os.path.join(self.workdir, "results.xml")
+        caldb                = "prod2"
+        irf                  = "South_50h"
+        ra                   =   83.63
+        dec                  =   22.01
+        rad_sim              =   10.0
+        tstart               =    0.0
+        tstop                = 1800.0
+        emin                 =    0.1
+        emax                 =  100.0
+        rad_select           =    3.0
+
+        # Simulate events
+        sim = ctools.ctobssim()
+        sim["inmodel"]   = model_name
+        sim["outevents"] = events_name
+        sim["caldb"]     = caldb
+        sim["irf"]       = irf
+        sim["ra"]        = ra
+        sim["dec"]       = dec
+        sim["rad"]       = rad_sim
+        sim["tmin"]      = tstart
+        sim["tmax"]      = tstop
+        sim["emin"]      = emin
+        sim["emax"]      = emax
+        sim.execute()
+
+        # Select events
+        select = ctools.ctselect()
+        select["inobs"]  = events_name
+        select["outobs"] = selected_events_name
+        select["ra"]     = ra
+        select["dec"]    = dec
+        select["rad"]    = rad_select
+        select["tmin"]   = tstart
+        select["tmax"]   = tstop
+        select["emin"]   = emin
+        select["emax"]   = emax
+        select.execute()
+
+        # Perform maximum likelihood fitting
+        like = ctools.ctlike()
+        like["inobs"]    = selected_events_name
+        like["inmodel"]  = model_name
+        like["outmodel"] = result_name
+        like["caldb"]    = caldb
+        like["irf"]      = irf
+        like.execute()
+
+        # TODO: add asserts
+        assert True
+
+    def test_unbinned_mem(self):
+        """
+        Test unbinned in-memory pipeline.
+        """
+        # Set script parameters
+        model_name           = self.model_name
+        caldb                = "prod2"
+        irf                  = "South_50h"
+        ra                   =   83.63
+        dec                  =   22.01
+        rad_sim              =   10.0
+        tstart               =    0.0
+        tstop                = 1800.0
+        emin                 =    0.1
+        emax                 =  100.0
+        rad_select           =    3.0
+
+        # Simulate events
+        sim = ctools.ctobssim()
+        sim["inmodel"] = model_name
+        sim["caldb"]   = caldb
+        sim["irf"]     = irf
+        sim["ra"]      = ra
+        sim["dec"]     = dec
+        sim["rad"]     = rad_sim
+        sim["tmin"]    = tstart
+        sim["tmax"]    = tstop
+        sim["emin"]    = emin
+        sim["emax"]    = emax
+        sim.run()
+
+        # Select events
+        select = ctools.ctselect(sim.obs())
+        select["ra"]   = ra
+        select["dec"]  = dec
+        select["rad"]  = rad_select
+        select["tmin"] = tstart
+        select["tmax"] = tstop
+        select["emin"] = emin
+        select["emax"] = emax
+        select.run()
+
+        # Perform maximum likelihood fitting
+        like = ctools.ctlike(select.obs())
+        like.run()
  
-    def test_it(self):
-        self.assertEqual('foo'.upper(), 'FOO')
+        # TODO: add asserts
+        assert True