Add test cases for cttsmap

Jürgen Knödlseder
1 parent d5a4fca2
Showing 3 changed files with 134 additions and 0 deletions
test/test_ctools.sh
test/test_cttsmap.py
test/test_python.py
@@ -36,6 +36,7 @@ $ECHO &quot;***************&quot;
 # ==================
 ctbin=../src/ctbin/ctbin
 ctlike=../src/ctlike/ctlike
+cttsmap=../src/cttsmap/cttsmap
 ctmodel=../src/ctmodel/ctmodel
 ctobssim=../src/ctobssim/ctobssim
 ctselect=../src/ctselect/ctselect
@@ -347,6 +348,34 @@ $ECHO &quot; ok&quot;
  
  
 #
+# Test cttsmap
+# ============
+$ECHO -n "Test cttsmap: "
+$cttsmap infile="selected_events.fits" \
+         srcmdl="data/crab.xml" \
+         srcname="Crab" \
+         caldb="irf" \
+         irf="cta_dummy_irf" \
+         outfile="tsmap.fits" \
+         nxpix=5 \
+         nypix=5 \
+         binsz=0.05 \
+         coordsys="CEL" \
+         xref=83.63 \
+         yref=22.01 \
+         proj="CAR"
+$ECHO -n "."
+if [ -s "tsmap.fits" ]
+then
+  $ECHO -n "."
+else
+  $ECHO " tsmap.fits file is not valid"
+  exit 1
+fi
+$ECHO " ok"
+
+
+#
 # Test ctexpcube
 # ==============
 $ECHO -n "Test ctexpcube: "
+#! /usr/bin/env python
+# ==========================================================================
+# This scripts performs unit tests for the cttsmap tool.
+#
+# Copyright (C) 2014 Juergen Knoedlseder
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+# ==========================================================================
+import gammalib
+import ctools
+
+
+# ========================== #
+# Test class for cttsmap tool #
+# ========================== #
+class Test(gammalib.GPythonTestSuite):
+    """
+    Test class for cttsmap tool.
+    """
+    # Constructor
+    def __init__(self):
+        """
+        Constructor.
+        """
+        # Call base class constructor
+        gammalib.GPythonTestSuite.__init__(self)
+
+        # Set members
+        self.events_name = "data/crab_events.fits"
+        self.model_name  = "data/crab.xml"
+        self.caldb       = "irf"
+        self.irf         = "cta_dummy_irf"
+
+        # Return
+        return
+
+    # Set test functions
+    def set(self):
+        """
+        Set all test functions.
+        """
+        # Set test name
+        self.name("cttsmap")
+
+        # Append tests
+        self.append(self.test_functional, "Test cttsmap functionality")
+
+        # Return
+        return
+
+    # Test cttsmap functionnality
+    def test_functional(self):
+        """
+        Test cttsmap functionnality.
+        """
+        # Set-up cttsmap
+        tsmap = ctools.cttsmap()
+        tsmap["infile"].filename(self.events_name)
+        tsmap["srcmdl"].filename(self.model_name)
+        tsmap["srcname"].string("Crab")
+        tsmap["caldb"].string(self.caldb)
+        tsmap["irf"].string(self.irf)
+        tsmap["outfile"].filename("tsmap.fits")
+        tsmap["nxpix"].integer(5)
+        tsmap["nypix"].integer(5)
+        tsmap["binsz"].real(0.02)
+        tsmap["coordsys"].string("CEL")
+        tsmap["proj"].string("CAR")
+        tsmap["xref"].real(83.63)
+        tsmap["yref"].real(22.01)
+
+        # Run tool
+        self.test_try("Run cttsmap")
+        try:
+            tsmap.run()
+            self.test_try_success()
+        except:
+            self.test_try_failure("Exception occured in cttsmap.")
+
+        # Save results
+        self.test_try("Save results")
+        try:
+            tsmap.save()
+            self.test_try_success()
+        except:
+            self.test_try_failure("Exception occured in saving results.")
+        
+        # Return
+        return
@@ -26,6 +26,7 @@ import test_ctobssim
 import test_ctselect
 import test_ctbin
 import test_ctlike
+import test_cttsmap
 import test_ctmodel
 import test_ctskymap
 import test_ctexpcube
@@ -50,6 +51,7 @@ if __name__ == &#39;__main__&#39;:
     suite_ctselect   = test_ctselect.Test()
     suite_ctbin      = test_ctbin.Test()
     suite_ctlike     = test_ctlike.Test()
+    suite_cttsmap    = test_cttsmap.Test()
     suite_ctmodel    = test_ctmodel.Test()
     suite_ctskymap   = test_ctskymap.Test()
     suite_ctexpcube  = test_ctexpcube.Test()
@@ -63,6 +65,7 @@ if __name__ == &#39;__main__&#39;:
     suite_ctselect.set()
     suite_ctbin.set()
     suite_ctlike.set()
+    suite_cttsmap.set()
     suite_ctmodel.set()
     suite_ctskymap.set()
     suite_ctexpcube.set()
@@ -76,6 +79,7 @@ if __name__ == &#39;__main__&#39;:
     suites.append(suite_ctselect)
     suites.append(suite_ctbin)
     suites.append(suite_ctlike)
+    suites.append(suite_cttsmap)
     suites.append(suite_ctmodel)
     suites.append(suite_ctskymap)
     suites.append(suite_ctexpcube)