Johan Bregeon / ctools

#! /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 <http://www.gnu.org/licenses/>.
#
# ==========================================================================
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 test data in "data" directory
        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'])

    # Create a local "pfiles" directory and set PFILES environment variable
    try:
        os.mkdir('pfiles')
    except:
        pass
    os.environ['PFILES'] = 'pfiles'

    # Copy the cscripts parameter files into the "pfiles" directory. For a
    # non-installed test we copy the parameter files from the "cscripts"
    # source directory into the "pfiles" directory, for an installed version
    # we get all parameter files from the "syspfiles" directory. Also make
    # sure that all parameter files are writable.
    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 the environment variable exists then unset it 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()