#! /usr/bin/env python # ========================================================================== # This scripts performs unit tests for cscripts. # # Copyright (C) 2016 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 . # # ========================================================================== import os import sys import gammalib import cscripts sys.path.append(os.path.dirname(os.path.abspath(__file__))) import test_cscaldb import test_csfindobs import test_csinfo import test_cslightcrv import test_csmodelinfo import test_csmodelmerge import test_csobs2caldb import test_csobsdef import test_csobsinfo import test_cspull import test_csresmap import test_csroot2caldb import test_cssens import test_csspec import test_cstsdist import test_cstsmapmerge import test_cstsmapsplit import test_csworkflow import test_csiactcopy import test_csiactdata import test_csiactobs # ================== # # Perform unit tests # # ================== # def test(installed=False, debug=False): """ Perform cscripts unit testing. Parameters ---------- installed : bool, optional Flag indicating whether the script has been installed or not debug : bool, optional Enables debugging """ # If we have an installed version then create a temporary directory and # copy over all information that is needed if installed: # Create temporary working directory import tempfile path = tempfile.mkdtemp() os.chdir(path) # Get test directory import inspect testdir = inspect.getfile(cscripts.tests) dirname = os.path.dirname(testdir) # Copy over test data and irf os.system('cp -r %s %s' % (dirname+'/data', 'data')) # Set test data environment variable os.environ['TEST_DATA'] = 'data' # ... otherwise set the calibration database to the one shipped with the # package; we don't need to set the 'TEST_DATA', this is done by the # test environment else: os.environ['CALDB'] = '%s/caldb' % (os.environ['TEST_SRCDIR']) # Set PFILES environment variable try: os.mkdir('pfiles') except: pass os.environ['PFILES'] = 'pfiles' # Copy over pfiles if not installed: os.system('cp -r %s/cscripts/*.par pfiles/'% (os.environ['TEST_SRCDIR'])) os.system('chmod u+w pfiles/*') else: os.system('cp -r %s/syspfiles/*.par pfiles/' % (os.environ['CTOOLS'])) os.system('chmod u+w pfiles/*') # Allocate test suite container suites = gammalib.GTestSuites('cscripts unit testing') # Allocate test suites suite_cscaldb = test_cscaldb.Test() suite_csinfo = test_csinfo.Test() suite_cslightcrv = test_cslightcrv.Test() suite_csmodelinfo = test_csmodelinfo.Test() suite_csmodelmerge = test_csmodelmerge.Test() suite_csobs2caldb = test_csobs2caldb.Test() suite_csobsdef = test_csobsdef.Test() suite_csobsinfo = test_csobsinfo.Test() suite_cspull = test_cspull.Test() suite_csresmap = test_csresmap.Test() suite_csroot2caldb = test_csroot2caldb.Test() suite_cssens = test_cssens.Test() suite_csspec = test_csspec.Test() suite_cstsdist = test_cstsdist.Test() suite_cstsmapmerge = test_cstsmapmerge.Test() suite_cstsmapsplit = test_cstsmapsplit.Test() suite_csworkflow = test_csworkflow.Test() # Setup unit tests suite_cscaldb.set() suite_csinfo.set() suite_cslightcrv.set() suite_csmodelinfo.set() suite_csmodelmerge.set() suite_csobs2caldb.set() suite_csobsdef.set() suite_csobsinfo.set() suite_cspull.set() suite_csresmap.set() suite_csroot2caldb.set() suite_cssens.set() suite_csspec.set() suite_cstsdist.set() suite_cstsmapmerge.set() suite_cstsmapsplit.set() suite_csworkflow.set() # Append tests to container suites.append(suite_cscaldb) suites.append(suite_csinfo) suites.append(suite_cslightcrv) suites.append(suite_csmodelinfo) suites.append(suite_csmodelmerge) suites.append(suite_csobs2caldb) suites.append(suite_csobsdef) suites.append(suite_csobsinfo) suites.append(suite_cspull) suites.append(suite_csresmap) suites.append(suite_csroot2caldb) suites.append(suite_cssens) suites.append(suite_csspec) suites.append(suite_cstsdist) suites.append(suite_cstsmapmerge) suites.append(suite_cstsmapsplit) suites.append(suite_csworkflow) # Append tests for Python 2.6+ (the IACT cscripts depend on the json # module which is only available since Python 2.6+ ver = sys.version.split()[0] if ver >= '2.6.0': # Check for VHEFITS environment variable if 'VHEFITS' in os.environ: # If environment variable exists then unset for the test cases. # Since Python is executed in a subprocess this will not impact # the environment variable in the parent shell. del os.environ['VHEFITS'] # Allocate test suites suite_csfindobs = test_csfindobs.Test() suite_csiactcopy = test_csiactcopy.Test() suite_csiactdata = test_csiactdata.Test() suite_csiactobs = test_csiactobs.Test() # Setup unit tests suite_csfindobs.set() suite_csiactcopy.set() suite_csiactdata.set() suite_csiactobs.set() # Append tests to container suites.append(suite_csfindobs) suites.append(suite_csiactcopy) suites.append(suite_csiactdata) suites.append(suite_csiactobs) # Run test suite success = suites.run() # If we have a non-installed version then save test results if not installed: suites.save('reports/cscripts.xml') # If debuging is requested then print test suites if debug: print(suites) # Set return code if success: rc = 0 else: rc = 1 # Remove temporary direction if installed: os.system('rm -rf %s' % (path)) # Exit with return code sys.exit(rc) # ======================== # # Main routine entry point # # ======================== # if __name__ == '__main__': # Run tests test()