Alexander Ziegler / ctools

Commit 16f51af681e5497d317a1af9873768710f7f38aa

Authored by Michael Mayer
1 parent 2bab4887

Added cstssplit (code, docs, test)

Inline Side-by-side
Showing 8 changed files with 684 additions and 0 deletions
cscripts/Makefile.am
  View file @16f51af
... ... @@ -22,6 +22,7 @@ pythonscripts = __init__.py \
22 22 csiactcopy.py \
23 23 csiactdata.py \
24 24 csiactobs.py \
  25 + cstssplit.py \
25 26 obsutils.py
26 27  
27 28 # Parfiles to be distributed in $(prefix)/syspfiles
... ... @@ -43,6 +44,7 @@ parfiles = cscaldb.par \
43 44 csfindobs.par \
44 45 csiactcopy.par \
45 46 csiactdata.par \
  47 + cstssplit.par \
46 48 csiactobs.par
47 49  
48 50 # Scripts for which symbolic links will be created in bindir
... ... @@ -65,6 +67,7 @@ linkscripts = cscaldb \
65 67 csfindobs \
66 68 csiactcopy \
67 69 csiactdata \
  70 + cstssplit \
68 71 csiactobs
69 72  
70 73 # Test scripts to be distributed in $(pyexecdir)/cscripts/tests
... ... @@ -88,6 +91,7 @@ testscripts = tests/__init__.py \
88 91 ../test/test_csfindobs.py \
89 92 ../test/test_csiactcopy.py \
90 93 ../test/test_csiactdata.py \
  94 + ../test/test_cstssplit.py \
91 95 ../test/test_csiactobs.py
92 96  
93 97 # Test data to be distributed in $(pyexecdir)/cscripts/tests/data
... ...
cscripts/__init__.py.in
  View file @16f51af
... ... @@ -40,6 +40,7 @@ __all__ = [
40 40 "cstsdist",
41 41 "cstsmapmerge",
42 42 "csworkflow",
  43 + "cstssplit",
43 44 "obsutils"
44 45 ]
45 46 if cur_version > req_version:
... ... @@ -65,6 +66,7 @@ from cscripts.csspec import csspec
65 66 from cscripts.cstsdist import cstsdist
66 67 from cscripts.cstsmapmerge import cstsmapmerge
67 68 from cscripts.csworkflow import csworkflow
  69 +from cscripts.cstssplit import cstssplit
68 70 from cscripts import obsutils
69 71 if cur_version > req_version:
70 72 from cscripts.csfindobs import csfindobs
... ...
cscripts/cstssplit.par 0 → 100644
  View file @16f51af
  1 +#############################################################################
  2 +# #
  3 +# CTA Science Analysis Tools #
  4 +# #
  5 +# ------------------------------------------------------------------------- #
  6 +# #
  7 +# File: cstssplit.par #
  8 +# Version: 1.1.0 #
  9 +# #
  10 +# Author: Michael Mayer #
  11 +# michael.mayer@physik.hu-berlin.de #
  12 +# HU Berlin #
  13 +# #
  14 +# Purpose: Parameter file for the cttsmap application #
  15 +# #
  16 +#############################################################################
  17 +#
  18 +# General parameters
  19 +#===================
  20 +inobs, f, a, "events.fits",,, "Input event list, counts cube or observation definition XML file"
  21 +inmodel, f, a, "$CTOOLS/share/models/crab.xml",,, "Input model XML file"
  22 +srcname, s, a, "Crab",,, "Test source name"
  23 +expcube, f, a, "NONE",,, "Input exposure cube file (only needed for stacked analysis)"
  24 +psfcube, f, a, "NONE",,, "Input PSF cube file (only needed for stacked analysis)"
  25 +edispcube, f, a, "NONE",,, "Input energy dispersion cube file (only needed for stacked analysis)"
  26 +bkgcube, f, a, "NONE",,, "Input background cube file (only needed for stacked analysis)"
  27 +caldb, s, a, "prod2",,, "Calibration database"
  28 +irf, s, a, "South_0.5h",,, "Instrument response function"
  29 +edisp, b, h, no,,, "Apply energy dispersion?"
  30 +outmap, f, a, "tsmap.fits",,, "Output Test Statistic map file"
  31 +bins_per_job, i, a, 5,,, "Number of TS map bins per running job"
  32 +compute_null, b, a, yes,,,"Compute null hypothesis first?"
  33 +outfile, f, a, "commands.txt",,, "ASCII file containing all commands"
  34 +
  35 +#
  36 +# Spatial binning parameters
  37 +#===========================
  38 +usepnt, b, h, no,,, "Use pointing instead of xref/yref parameters?"
  39 +nxpix, i, a, 200,,, "Size of the X axis in pixels"
  40 +nypix, i, a, 200,,, "Size of the Y axis in pixels"
  41 +binsz, r, a, 0.02,,, "Image scale (in degrees/pixel)"
  42 +coordsys, s, a, "CEL", CEL|GAL,,"Coordinate system (CEL - celestial, GAL - galactic)"
  43 +xref, r, a, 83.63,0,360, "First coordinate of image center in degrees (RA or galactic l)"
  44 +yref, r, a, 22.01,-90,90, "Second coordinate of image center in degrees (DEC or galactic b)"
  45 +proj, s, a, "CAR", AIT|AZP|CAR|MER|MOL|STG|TAN,, "Projection method"
  46 +
  47 +#
  48 +# Standard parameters
  49 +#====================
  50 +publish, b, h, no,,, "Publish Test Statistic map on VO Hub?"
  51 +chatter, i, h, 2,0,4, "Chattiness of output"
  52 +clobber, b, h, yes,,, "Overwrite existing output files with new output files?"
  53 +debug, b, h, no,,, "Debugging mode activated"
  54 +mode, s, h, "ql",,, "Mode of automatic parameters"
  55 +logfile, f, h, "cstssplit.log",,, "Log filename"
... ...
cscripts/cstssplit.py 0 → 100755
  View file @16f51af
  1 +#!/usr/bin/env python
  2 +# ==========================================================================
  3 +# Create commands to split TS map computation
  4 +#
  5 +# Copyright (C) 2016 Michael Mayer
  6 +#
  7 +# This program is free software: you can redistribute it and/or modify
  8 +# it under the terms of the GNU General Public License as published by
  9 +# the Free Software Foundation, either version 3 of the License, or
  10 +# (at your option) any later version.
  11 +#
  12 +# This program is distributed in the hope that it will be useful,
  13 +# but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15 +# GNU General Public License for more details.
  16 +#
  17 +# You should have received a copy of the GNU General Public License
  18 +# along with this program. If not, see <http://www.gnu.org/licenses/>.
  19 +#
  20 +# ==========================================================================
  21 +import sys
  22 +import os
  23 +import math
  24 +import gammalib
  25 +import ctools
  26 +
  27 +
  28 +# ============== #
  29 +# cstssplit class #
  30 +# ============== #
  31 +class cstssplit(ctools.cscript):
  32 + """
  33 + Generates commands to split TS map computation
  34 +
  35 + This class implements the creation of file containing commands to split
  36 + the computation of a TS map into several machines. It derives from
  37 + the ctools.cscript class which provides support for parameter files,
  38 + command line arguments, and logging. In that way the Python script
  39 + behaves just as a regular ctool.
  40 + """
  41 +
  42 + # Consructor
  43 + def __init__(self, *argv):
  44 + """
  45 + Constructor.
  46 + """
  47 +
  48 + # Set name
  49 + self._name = 'cstssplit'
  50 + self._version = '1.1.0'
  51 + self._outmap = gammalib.GFilename()
  52 + self._bins_per_job = 0
  53 + self._compute_null = False
  54 + self._outfile = gammalib.GFilename()
  55 + self._map = gammalib.GSkyMap()
  56 + self._cmd = []
  57 +
  58 + # Initialise observation container from constructor arguments.
  59 + self._obs, argv = self._set_input_obs(argv)
  60 +
  61 + # Initialise application by calling the appropriate class
  62 + # constructor.
  63 + self._init_cscript(argv)
  64 +
  65 + # Return
  66 + return
  67 +
  68 +
  69 + # Private methods
  70 + def _get_parameters(self):
  71 + """
  72 + Get parameters from parfile and setup the observation.
  73 + """
  74 +
  75 + # If there are no observations in container then get some ...
  76 + if self._obs.size() == 0:
  77 + self._obs = self._get_observations()
  78 +
  79 + # Set models if we have none
  80 + if self._obs.models().size() == 0:
  81 + self._obs.models(self['inmodel'].filename())
  82 +
  83 + # Get TS map paramters
  84 + self._map = self._create_map(self._obs)
  85 + self._srcname = self['srcname'].string()
  86 + self._outmap = self['outmap'].filename()
  87 +
  88 + # Get additional parameters
  89 + self._bins_per_job = self['bins_per_job'].integer()
  90 + self._compute_null = self['compute_null'].boolean()
  91 + self._outfile = self['outfile'].filename()
  92 +
  93 + # Write input parameters into logger
  94 + if self._logTerse():
  95 + self._log_parameters()
  96 + self._log('\n')
  97 +
  98 + # Return
  99 + return
  100 +
  101 +
  102 + def _compute_null_hypothesis(self):
  103 + """
  104 + Compute null hypothesis
  105 +
  106 + Returns:
  107 + Log-likelihood of null hypothesis
  108 + """
  109 +
  110 + # Store original models
  111 + models_orig = self._obs.models()
  112 +
  113 + # Get models instance
  114 + models0 = self._obs.models()
  115 +
  116 + # Remove test source
  117 + models0.remove(self._srcname)
  118 +
  119 + # Initialise optimizer
  120 + opt = gammalib.GOptimizerLM()
  121 +
  122 + # Set models for null hypothesis
  123 + self._obs.models(models0)
  124 +
  125 + # Optimise observation container
  126 + self._obs.optimize(opt)
  127 +
  128 + # Set original models to container
  129 + self._obs.models(models_orig)
  130 +
  131 + # Return optimised log-likelihood of null hypothesis
  132 + return -(opt.value())
  133 +
  134 +
  135 + def run(self):
  136 + """
  137 + Run the script.
  138 + """
  139 +
  140 + # Switch screen logging on in debug mode
  141 + if self._logDebug():
  142 + self._log.cout(True)
  143 +
  144 + # Get parameters
  145 + self._get_parameters()
  146 +
  147 + # Write information into logger
  148 + if self._logTerse():
  149 + self._log('\n')
  150 + self._log.header1('Test source')
  151 + model = self._obs.models()[self._srcname]
  152 + self._log(str(model))
  153 + self._log('\n')
  154 +
  155 + # Set log-likelihood to zero
  156 + logL0 = 0.0
  157 +
  158 + # Pre-compute null hypothesis if requested
  159 + if self._compute_null:
  160 +
  161 + # Write information into logger
  162 + if self._logTerse():
  163 + self._log('\n')
  164 + self._log.header1('Compute null hypothesis')
  165 +
  166 + # Compute null hypothesis
  167 + logL0 = self._compute_null_hypothesis()
  168 +
  169 + # Write likelihood into logger
  170 + if self._logTerse():
  171 + self._log(gammalib.parformat('Source removed'))
  172 + self._log(self._srcname)
  173 + self._log('\n')
  174 + self._log(gammalib.parformat('Log-likelihood'))
  175 + self._log(repr(logL0))
  176 + self._log('\n')
  177 +
  178 + # Get parameters of TS map ctool
  179 + pars = gammalib.GApplicationPars('cttsmap.par')
  180 +
  181 + # Compute total number of jobs
  182 + nbins = self._map.npix()
  183 + njobs = int(math.ceil(float(nbins) / float(self._bins_per_job)) + 0.1)
  184 +
  185 + # Set parameters to be skipped now, we will deal with them later
  186 + skip_pars = ['binmin', 'binmax', 'logL0', 'outmap', 'logfile']
  187 +
  188 + # Set tool name for computation
  189 + base_command = 'cttsmap'
  190 +
  191 + # Write information into logger
  192 + if self._logTerse():
  193 + self._log('\n')
  194 + self._log.header1('Create commands')
  195 + self._log(gammalib.parformat('Number of cttsmap calls'))
  196 + self._log(str(njobs))
  197 + self._log('\n')
  198 +
  199 + # Loop over TS map parameters
  200 + for par in pars:
  201 +
  202 + # Skip if we deal with them later
  203 + if par.name() in skip_pars:
  204 + continue
  205 +
  206 + # Skip if they need to be queried
  207 + # This way we ensure we pass only parameters
  208 + # that were queried before
  209 + if self[par.name()].is_query():
  210 + continue
  211 +
  212 + # Set TS map parameter according to input from
  213 + # this script
  214 + par.value(self[par.name()].value())
  215 +
  216 + # Append command to set parameter
  217 + base_command += ' ' + par.name() + '=' + par.value()
  218 +
  219 + # Append null hypothesis parameter
  220 + base_command += ' logL0=' + repr(logL0)
  221 +
  222 + # Set binning to start from zero
  223 + binmin = 0
  224 + binmax = 0
  225 +
  226 + # Clear command sequence
  227 + self._cmd = []
  228 +
  229 + # Loop over jobs and create commands
  230 + for job in range(njobs):
  231 +
  232 + # Set specific outmap file name
  233 + outmap = self._outmap.url().replace('.fits', '_'+str(job)+'.fits')
  234 +
  235 + # Set bin numbers to be computed in this job
  236 + binmin = binmax
  237 + binmax = binmin + self._bins_per_job
  238 + if binmax > nbins:
  239 + binmax = nbins
  240 +
  241 + # Complete command by binning parameters
  242 + sliced_command = base_command + ' binmin='+str(binmin)+' binmax='+str(binmax)
  243 + sliced_command += ' outmap='+outmap + ' logfile='+outmap.replace('.fits','.log')
  244 +
  245 + # Append command to list of commands
  246 + self._cmd.append(sliced_command)
  247 +
  248 + # Write information into logger
  249 + if self._logExplicit():
  250 + self._log('\n')
  251 + self._log.header2('Commands')
  252 + for cmd in self._cmd:
  253 + self._log(cmd)
  254 + self._log('\n')
  255 + self._log('\n')
  256 +
  257 + # Return
  258 + return
  259 +
  260 +
  261 + def execute(self):
  262 + """
  263 + Execute the script.
  264 + """
  265 +
  266 + # Run the script
  267 + self.run()
  268 +
  269 + # Save results
  270 + self.save()
  271 +
  272 + # Return
  273 + return
  274 +
  275 +
  276 + def save(self):
  277 + """
  278 + Save commands to ASCII file
  279 + """
  280 +
  281 + # Open file
  282 + f = open(self._outfile.url(), 'w')
  283 +
  284 + # Write commands to file
  285 + for cmd in self._cmd:
  286 + f.write(cmd + '\n')
  287 +
  288 + # Close file
  289 + f.close()
  290 +
  291 + # Return
  292 + return
  293 +# ======================== #
  294 +# Main routine entry point #
  295 +# ======================== #
  296 +if __name__ == '__main__':
  297 + """
  298 + Generates splitted TS map commands
  299 + """
  300 +
  301 + # Create instance of application
  302 + app = cstssplit(sys.argv)
  303 +
  304 + # Execute application
  305 + app.execute()
  306 +
  307 +#
... ...
doc/source/users/reference_manual/cstssplit.rst 0 → 100644
  View file @16f51af
  1 +.. _cstssplit:
  2 +
  3 +cstssplit
  4 +=======
  5 +
  6 +Generate commands to split TS map computation
  7 +
  8 +
  9 +Synopsis
  10 +--------
  11 +
  12 +This tool allows to generate a command sequence in order to compute a
  13 +Test Statistic (TS) map using e.g. several machines. The command sequence
  14 +is stored in an ASCII file. Each line contains an individual command.
  15 +Each command will call the tool :ref:`cttsmap` with the appropriate parameters.
  16 +To split the computation, each TS map that is computed has a suffix to the filename
  17 +according to the number of the command. The resulting individual TS maps can then be
  18 +merged via :ref:`cstsmapmerge`.
  19 +
  20 +
  21 +General parameters
  22 +------------------
  23 +
  24 +``inobs [file]``
  25 + Input event list, counts cube or observation definition XML file.
  26 +
  27 +``inmodel [file]``
  28 + Input model XML file.
  29 +
  30 +``srcname [string]``
  31 + Name of source model for which the TS map should be computed.
  32 +
  33 +``expcube [file]``
  34 + Input exposure cube file (only needed for stacked analysis).
  35 +
  36 +``psfcube [file]``
  37 + Input PSF cube file (only needed for stacked analysis).
  38 +
  39 +``bkgcube [file]``
  40 + Input background cube file (only needed for stacked analysis).
  41 +
  42 +``caldb [string]``
  43 + Calibration database.
  44 +
  45 +``irf [string]``
  46 + Instrumental response function.
  47 +
  48 +``(edisp = no) [boolean]``
  49 + Applies energy dispersion to response computation.
  50 +
  51 +``outmap [file]``
  52 + Output Test Statistic map file.
  53 +
  54 +``(usepnt = no) [boolean]``
  55 + Use CTA pointing direction for map centre instead of xref/yref parameters?
  56 +
  57 +``nxpix [integer]``
  58 + Number of map pixels in Right Ascension or Galactic longitude.
  59 +
  60 +``nypix [integer]``
  61 + Number of map pixels in Declination or Galactic latitude.
  62 +
  63 +``binsz [real]``
  64 + Map pixel size (in degrees/pixel).
  65 +
  66 +``coordsys <CEL|GAL> [string]``
  67 + Coordinate system (CEL - celestial, GAL - galactic).
  68 +
  69 +``xref [real]``
  70 + Right Ascension / Galactic longitude of map centre (J2000, in degrees).
  71 +
  72 +``yref [real]``
  73 + Declination / Galactic latitude of map centre (J2000, in degrees).
  74 +
  75 +``proj <AIT|AZP|CAR|MER|MOL|STG|TAN> [string]``
  76 + Projection method.
  77 +
  78 +``bins_per_job [integer]``
  79 + Number of bins of the TS map that should be computed within one command
  80 +
  81 +``compute_null [boolean]``
  82 + Pre-computes the null hypothesis globally and passes the value to the individual commands
  83 + If 'false', the null hypothesis will be computed in each run of :ref:`cttsmap` individually.
  84 +
  85 +``outfile [file]``
  86 + Output ASCII file name where the commands are written to.
  87 +
  88 +
  89 +Standard parameters
  90 +-------------------
  91 +
  92 +``(publish = no) [boolean]``
  93 + Specifies whether the TS map should be published on VO Hub.
  94 +
  95 +``(chatter = 2) [integer]``
  96 + Verbosity of the executable:
  97 + chatter = 0: no information will be logged
  98 +
  99 + chatter = 1: only errors will be logged
  100 +
  101 + chatter = 2: errors and actions will be logged
  102 +
  103 + chatter = 3: report about the task execution
  104 +
  105 + chatter = 4: detailed report about the task execution
  106 +
  107 +``(clobber = yes) [boolean]``
  108 + Specifies whether an existing output TS map file should be overwritten.
  109 +
  110 +``(debug = no) [boolean]``
  111 + Enables debug mode. In debug mode the executable will dump any log file output to the console.
  112 +
  113 +``(mode = ql) [string]``
  114 + Mode of automatic parameters (default is "ql", i.e. "query and learn").
  115 +
  116 +``(logfile = cttsmap.log) [string]``
  117 + Name of log file.
  118 +
  119 +
  120 +Related tools or scripts
  121 +------------------------
  122 +
  123 +:ref:`cttsmap`
  124 +:ref:`cstsmapmerge`
  125 +
  126 +
... ...
doc/source/users/reference_manual/index.rst
  View file @16f51af
... ... @@ -50,7 +50,9 @@ cscripts
50 50 cssens --- Computes CTA sensitivity <cssens>
51 51 csspec --- Computes spectral points <csspec>
52 52 cstsdist --- Generates TS distribution <cstsdist>
  53 + cstssplit --- Creates commands to slice the ts map computations <cstssplit>
53 54 cstsmapmerge --- Merges slices from ts map computations <cstsmapmerge>
  55 +
54 56  
55 57  
56 58 Scripts to manage an IACT database
... ...
test/test_cstssplit.py 0 → 100755
  View file @16f51af
  1 +#! /usr/bin/env python
  2 +# ==========================================================================
  3 +# This scripts performs unit tests for the cstssplit script.
  4 +#
  5 +# Copyright (C) 2016 Michael Mayer
  6 +#
  7 +# This program is free software: you can redistribute it and/or modify
  8 +# it under the terms of the GNU General Public License as published by
  9 +# the Free Software Foundation, either version 3 of the License, or
  10 +# (at your option) any later version.
  11 +#
  12 +# This program is distributed in the hope that it will be useful,
  13 +# but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15 +# GNU General Public License for more details.
  16 +#
  17 +# You should have received a copy of the GNU General Public License
  18 +# along with this program. If not, see <http://www.gnu.org/licenses/>.
  19 +#
  20 +# ==========================================================================
  21 +import os
  22 +import gammalib
  23 +import cscripts
  24 +
  25 +
  26 +# =============================== #
  27 +# Test class for cstssplit script #
  28 +# =============================== #
  29 +class Test(gammalib.GPythonTestSuite):
  30 + """
  31 + Test class for cstssplit script.
  32 +
  33 + This test class makes unit tests for the csstsplit script by using it
  34 + from the command line and from Python.
  35 + """
  36 +
  37 + # Constructor
  38 + def __init__(self):
  39 + """
  40 + Constructor.
  41 + """
  42 + # Call base class constructor
  43 + gammalib.GPythonTestSuite.__init__(self)
  44 +
  45 + # Return
  46 + return
  47 +
  48 + # Set test functions
  49 + def set(self):
  50 + """
  51 + Set all test functions.
  52 + """
  53 + # Set test name
  54 + self.name('cstssplit')
  55 +
  56 + # Append tests
  57 + self.append(self._test_cmd, "Test cstssplit on command line")
  58 + #self.append(self._test_python, "Test cstssplit from Python")
  59 +
  60 + # Return
  61 + return
  62 +
  63 + # Test cstssplit on command line
  64 + def _test_cmd(self):
  65 + """
  66 + Test cstssplit on the command line.
  67 + """
  68 + # Kluge to set the command (installed version has no README file)
  69 + if os.path.isfile("README.md"):
  70 + cstssplit = "../cscripts/cstssplit.py"
  71 + else:
  72 + cstssplit = "cstssplit"
  73 +
  74 + # Setup cstssplit command
  75 + cmd = cstssplit+' inobs="selected_events.fits"'+ \
  76 + ' inmodel="data/crab.xml"'+ \
  77 + ' srcname="Crab"'+ \
  78 + ' caldb="irf"'+ \
  79 + ' irf="cta_dummy_irf"'+ \
  80 + ' outmap="tsmap.fits"'+ \
  81 + ' nxpix=5'+ \
  82 + ' nypix=5'+ \
  83 + ' binsz=0.05'+ \
  84 + ' coordsys="CEL"'+ \
  85 + ' xref=83.63'+ \
  86 + ' yref=22.01'+ \
  87 + ' proj="CAR"'+ \
  88 + ' compute_null=yes'+ \
  89 + ' bins_per_job=5'+ \
  90 + ' outfile="tsmap_cmds1.txt"'
  91 +
  92 +
  93 + # Execute cstssplit, make sure we catch any exception
  94 + try:
  95 + rc = os.system(cmd+" >/dev/null 2>&1")
  96 + except:
  97 + pass
  98 +
  99 + # Check if execution was successful
  100 + self.test_assert(rc == 0,
  101 + "Successful cstsssplit execution on command line")
  102 +
  103 + # Check observation definition XML file
  104 + self._check_cmdfile('tsmap_cmds1.txt')
  105 +
  106 + # Setup cstssplit command
  107 + cmd = cstssplit+' inobs="selected_events.fits"'+ \
  108 + ' inmodel="data/crab.xml"'+ \
  109 + ' srcname="Does_not_exist"'+ \
  110 + ' caldb="irf"'+ \
  111 + ' irf="cta_dummy_irf"'+ \
  112 + ' outmap="tsmap.fits"'+ \
  113 + ' nxpix=5'+ \
  114 + ' nypix=5'+ \
  115 + ' binsz=0.05'+ \
  116 + ' coordsys="CEL"'+ \
  117 + ' xref=83.63'+ \
  118 + ' yref=22.01'+ \
  119 + ' proj="CAR"'+ \
  120 + ' compute_null=yes'+ \
  121 + ' bins_per_job=5'+ \
  122 + ' outfile="tsmap_cmds2.txt"'
  123 +
  124 +
  125 + # Execute cstssplit, make sure we catch any exception
  126 + try:
  127 + rc = os.system(cmd+" >/dev/null 2>&1")
  128 + except:
  129 + pass
  130 +
  131 + # Check if execution failed
  132 + self.test_assert(rc != 0,
  133 + "Failure of cstssplit execution on command line")
  134 +
  135 + # Return
  136 + return
  137 +
  138 + # Test cstssplit from Python
  139 + def _test_python(self):
  140 + """
  141 + Test cstssplit from Python.
  142 + """
  143 + # Set-up cstssplit
  144 + tssplit = cscripts.cstssplit()
  145 + tssplit['inobs'] = 'obs_selected.xml'
  146 + tssplit['inmodel'] = 'data/crab.xml'
  147 + tssplit['xref'] = 83.6331
  148 + tssplit['yref'] = 22.01
  149 + tssplit['coordsys'] = 'CEL'
  150 + tssplit['proj'] = 'CAR'
  151 + tssplit['binsz'] = 0.05
  152 + tssplit['nxpix'] = 5
  153 + tssplit['nypix'] = 5
  154 + tssplit['compute_null'] = False
  155 + tssplit['bins_per_job'] = 5
  156 + tssplit['outfile'] = 'tsmap_cmd_py1.txt'
  157 +
  158 +
  159 + # Run cstssplit script and save run list
  160 + tssplit.logFileOpen() # Make sure we get a log file
  161 + tssplit.run()
  162 + tssplit.save()
  163 +
  164 + # Check observation definition XML file
  165 + self._check_cmdfile('tsmap_cmd_py1.txt')
  166 +
  167 + # Return
  168 + return
  169 +
  170 + # Check observation definition XML file
  171 + def _check_cmdfile(self, filename):
  172 + """
  173 + Check ascii files containing commands
  174 + """
  175 +
  176 + # Create file name instance
  177 + fname = gammalib.GFilename(filename)
  178 +
  179 + # Check if execution was successful
  180 + self.test_assert(fname.exists(),
  181 + "Output of cstssplit exists")
  182 +
  183 + # Return
  184 + return
... ...
test/test_python_cscripts.py
  View file @16f51af
... ... @@ -42,6 +42,7 @@ import test_csworkflow
42 42 import test_csiactcopy
43 43 import test_csiactdata
44 44 import test_csiactobs
  45 +import test_cstssplit
45 46  
46 47  
47 48 # ================== #
... ... @@ -113,6 +114,7 @@ def test(installed=False, debug=False):
113 114 suite_cstsdist = test_cstsdist.Test()
114 115 suite_cstsmapmerge = test_cstsmapmerge.Test()
115 116 suite_csworkflow = test_csworkflow.Test()
  117 + suite_cstssplit = test_cstssplit.Test()
116 118  
117 119 # Setup unit tests
118 120 suite_cscaldb.set()
... ... @@ -130,6 +132,7 @@ def test(installed=False, debug=False):
130 132 suite_cstsdist.set()
131 133 suite_cstsmapmerge.set()
132 134 suite_csworkflow.set()
  135 + suite_cstssplit.set()
133 136  
134 137 # Append tests to container
135 138 suites.append(suite_cscaldb)
... ... @@ -147,6 +150,7 @@ def test(installed=False, debug=False):
147 150 suites.append(suite_cstsdist)
148 151 suites.append(suite_cstsmapmerge)
149 152 suites.append(suite_csworkflow)
  153 + suites.append(suite_cstssplit)
150 154  
151 155 # Append tests for Python 2.6+ (the IACT cscripts depend on the json
152 156 # module which is only available since Python 2.6+
... ...