Add test function to ctools Python package

Christoph Deil
1 parent 551763bd
Showing 5 changed files with 65 additions and 13 deletions
cscripts/Makefile.am
cscripts/__init__.py
cscripts/tests/__init__.py
cscripts/tests/test_binned.py
cscripts/tests/test_unbinned.py
@@ -17,12 +17,19 @@ linkscripts = cspull cssens cstsdist cscaldb csresmap \
               cshessobs csobsdef csspec cslightcrv \
               csinfo
  
+testscripts = tests/__init__.py \
+              tests/test_unbinned.py \
+              tests/test_binned.py
+
+
 # Distribute scripts
-EXTRA_DIST         = $(binscripts) $(pythonscripts)
+EXTRA_DIST         = $(binscripts) $(pythonscripts) $(testscripts)
 binscriptsdir      = $(bindir)
 binscripts_DATA    = $(binscripts)
 pythonscriptsdir   = $(pythondir)/cscripts
 pythonscripts_DATA = $(pythonscripts)
+testscriptsdir   = $(pythondir)/cscripts/tests
+testscripts_DATA = $(testscripts)
  
 # Add hook rule to create symbolic links for cscripts
 install-data-hook:
+"""cscripts --- Python scripts from the ctools package.
+
+http://cta.irap.omp.eu/ctools/
+"""
+
+
 # List all scripts in this module
-__all__ = ["cscaldb", \
-           "cshessobs", \
-           "csinfo", \
-           "cslightcrv", \
-           "csobsdef", \
-           "cspull", \
-           "csresmap", \
-           "cssens", \
-           "csspec", \
-           "cstsdist", \
-           "obsutils"]
+__all__ = [
+    "cscaldb",
+    "cshessobs",
+    "csinfo",
+    "cslightcrv",
+    "csobsdef",
+    "cspull",
+    "csresmap",
+    "cssens",
+    "csspec",
+    "cstsdist",
+    "obsutils",
+    "test",
+]
  
 # Import
 from .cscaldb    import cscaldb
@@ -22,4 +31,22 @@ from .csresmap   import csresmap
 from .cssens     import cssens
 from .csspec     import csspec
 from .cstsdist   import cstsdist
-import obsutils
 \ No newline at end of file
+from . import obsutils
+
+
+def test():
+    """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)
+"""Tests that can be run for an installed ctools package.
+"""
+"""Test binned analysis.
+"""
+import unittest
+
+class TestBinnedAnalysis(unittest.TestCase):
+
+    def test_it(self):
+        self.assertEqual('foo'.upper(), 'FOO')
+"""Test unbinned analysis.
+"""
+import unittest
+
+class TestUnbinnedAnalysis(unittest.TestCase):
+
+    def test_it(self):
+        self.assertEqual('foo'.upper(), 'FOO')