Functions
def	write_RecoTF (opts)

def	write_AthAnalysis_TestAlg ()

def	write_AnalysisBase_xAODChecker ()

def	writeTFileOpen ()

def	parse_options ()

def	main ()

Variables
	DAOD_FILE

	RAW_FILE

	RDO_FILE

	HAMMERCLOUD_FILE

	PROJECTS

	XAODCHECKER_TURLS

	RECO_TF_TURLS

	TFILE_OPEN_TURLs

Function Documentation

◆ main()

def generateARTtests.main ( )

Definition at line 280 of file generateARTtests.py.

 def main():
   #TODO: Extend to AnalysisBase test algorithms
   
   opts = parse_options()
   
   if not os.path.exists("./test/"):
     os.makedirs("./test/")
   
   for i in xrange(len(opts.athena_project)):
     if not opts.athena_project[i] in PROJECTS:
       print "\tAthena project \"",opts.athena_project[i],"\" not intended for DirectIOART!"
       continue
   
   if opts.mp and not "Athena" in opts.athena_project:
     opts.mp = False
   
   if opts.open_tests:
     writeTFileOpen()
   
   if "AthAnalysis" in opts.athena_project:
     write_AthAnalysis_TestAlg()
   
   if "Athena" in opts.athena_project or "AthDerivation" in opts.athena_project:
     write_RecoTF(opts)
   
   if "AnalysisBase" in opts.athena_project:
     write_AnalysisBase_xAODChecker()
  

◆ parse_options()

def generateARTtests.parse_options ( )

Definition at line 269 of file generateARTtests.py.

 def parse_options():
   import argparse
   parser = argparse.ArgumentParser()
   
   parser.add_argument("athena_project", help="Specify the Athena release flavour", type=str, nargs="*" )
   parser.add_argument("-o", "--tfile-open", help="Generate ART scripts for TFile::Open tests", action="store_true", dest="open_tests")
   parser.add_argument("-m", "--mp", help="Generate ART tests for AthenaMP", action="store_true")
   
   options = parser.parse_args()
   return options
  

◆ write_AnalysisBase_xAODChecker()

def generateARTtests.write_AnalysisBase_xAODChecker ( )

Definition at line 191 of file generateARTtests.py.

 def write_AnalysisBase_xAODChecker():
   #loop over turl list
   for key, value in XAODCHECKER_TURLS.items():
     name = os.path.join("test", "test_directioart_analysisbase_xaodchecker_{}.sh".format(key.lower()))
     print("\tGenerating ... {}".format(name))
     outfile = open(name,'w')
     # generate test scripts
     outstring = """#!/bin/bash\n
 # art-description: DirectIOART AnalysisBase xAODChecker protocol={protocol}
 # art-type: grid
 # art-include: 21.2/AnalysisBase\n
 set -e\n
 xAODChecker {turl}\n
 echo \"art-result: $? DirectIOART_AnalysisBase_xAODChecker_protocol_{protocol}\"\n
 PFC_FILE=PoolFileCatalog.xml
 touch $PFC_FILE
 echo \"<?xml version=\\"1.0\\" encoding=\\"UTF-8\\" standalone=\\"no\\" ?>\" >> $PFC_FILE
 echo \"<!-- Edited By POOL -->\" >> $PFC_FILE
 echo \"<!DOCTYPE POOLFILECATALOG SYSTEM \\"InMemory\\">\" >> $PFC_FILE
 echo \"<POOLFILECATALOG>\" >> $PFC_FILE
 echo \"\" >> $PFC_FILE
 echo \"</POOLFILECATALOG>\" >> $PFC_FILE""".format(turl=value, protocol=key)
     outfile.write(outstring)
     outfile.close()
     os.system("chmod +x " + name)
  

◆ write_AthAnalysis_TestAlg()

def generateARTtests.write_AthAnalysis_TestAlg ( )

Definition at line 170 of file generateARTtests.py.

 def write_AthAnalysis_TestAlg():
   # loop over turl list
   for i in xrange(len(TFILE_OPEN_TURLs)):
     # get values of dictionaries
     for key, value in TFILE_OPEN_TURLs[i].items():
       name = os.path.join("test", "test_directioart_athanalysis_alg_{}.sh".format(key.lower()))
       print("\tGenerating ... {}".format(name))
       outfile = open(name,'w')
       # generate tests scripts
       outstring = """#!/bin/bash\n
 # art-description: DirectIOART AthAnalysis ParticleSelectionAlg protocol={protocol}
 # art-type: grid
 # art-output: *.pool.root
 # art-include: 21.2/AthAnalysis\n
 set -e\n
 athena --filesInput=\"{turl}\" DirectIOART/DirectIOARTAthAnalysisJobOptions.py\n
 echo \"art-result: $? DirectIOART_AthAnalysis_ParticleSelectionAlg_protocol_{protocol}\"""".format(turl=value[0], protocol=key)
       outfile.write(outstring)
       outfile.close()
       os.system("chmod +x " + name)
  

◆ write_RecoTF()

def generateARTtests.write_RecoTF ( opts )

Definition at line 69 of file generateARTtests.py.

 def write_RecoTF(opts):
   # in case of "Athena"
   if "Athena" in opts.athena_project:
     # loop over turl list of RDO input files
     for item in RECO_TF_TURLS["RDO"]:
       name = os.path.join("test", "test_directioart_athena_recotf_with_rdo_{}.sh".format(item.keys()[0].lower()))
       print("\tGenerating ... {}".format(name))
       outfile = open(name,'w')
       # generate test scripts
       outstring = """#!/bin/bash\n
 # art-description: DirectIOART Athena Reco_tf.py inputFile:RDO protocol={protocol}
 # art-type: grid
 # art-output: *.pool.root
 # art-include: 21.0/Athena
 # art-include: main/Athena\n
 set -e\n
 Reco_tf.py --AMI q221 --inputRDOFile {turl} --outputRDO_TRIGFile art.pool.root\n
 echo \"art-result: $? DirectIOART_Athena_RecoTF_inputRDO_protocol_{protocol}\"""".format(turl=item.values()[0], protocol=item.keys()[0])
       outfile.write(outstring)
       outfile.close()
       os.system("chmod +x " + name)
       # if AthenaMP
       if opts.mp:
         name = os.path.join("test", "test_directioart_athenamp_recotf_with_rdo_{}.sh".format(item.keys()[0].lower()))
         print("\tGenerating ... {}".format(name))
         outfile = open(name,'w')
         # generate test scripts
         outstring = """#!/bin/bash\n
 # art-description: DirectIOART AthenaMP Reco_tf.py inputFile:RDO protocol={protocol}
 # art-type: grid
 # art-output: *.pool.root
 # art-include: 21.0/Athena
 # art-include: main/Athena\n
 set -e\n
 export ATHENA_CORE_NUMBER=2
 Reco_tf.py --AMI q221 --multiprocess --inputRDOFile {turl} --outputRDO_TRIGFile art.pool.root\n
 echo \"art-result: $? DirectIOART_AthenaMP_RecoTF_inputRDO_protocol_{protocol}\"""".format(turl=item.values()[0], protocol=item.keys()[0])
         outfile.write(outstring)
         outfile.close()
         os.system("chmod +x " + name)
     
     # loop over turl list if RAW input files
     for item in RECO_TF_TURLS["RAW"]:
       name = os.path.join("test", "test_directioart_athena_recotf_with_raw_{}.sh".format(item.keys()[0].lower()))
       print("\tGenerating ... {}".format(name))
       outfile = open(name,'w')
       # generate test scripts
       outstring = """#!/bin/bash\n
 # art-description: DirectIOART Athena Reco_tf.py inputFile:RAW protocol={protocol}
 # art-type: grid
 # art-output: *.pool.root
 # art-include: 21.0/Athena
 # art-include: main/Athena\n
 set -e\n
 Reco_tf.py --AMI q431 --inputBSFile {turl} --outputESDFile art.pool.root\n
 echo \"art-result: $? DirectIOART_Athena_RecoTF_inputBS_protocol_{protocol}\"""".format(turl=item.values()[0], protocol=item.keys()[0])
       outfile.write(outstring)
       outfile.close()
       os.system("chmod +x " + name)
       # if AthenaMP
       if opts.mp:
         name = os.path.join("test", "test_directioart_athenamp_recotf_with_raw_{}.sh".format(item.keys()[0].lower()))
         print("\tGenerating ... {}".format(name))
         outfile = open(name,'w')
         # generate test scripts
         outstring = """#!/bin/bash\n
 # art-description: DirectIOART AthenaMP Reco_tf.py inputFile:RAW protocol={protocol}
 # art-type: grid
 # art-output: *.pool.root
 # art-include: 21.0/Athena
 # art-include: main/Athena\n
 set -e\n
 export ATHENA_CORE_NUMBER=2
 Reco_tf.py --AMI q431 --multiprocess --inputBSFile {turl} --outputESDFile art.pool.root\n
 echo \"art-result: $? DirectIOART_AthenaMP_RecoTF_inputBS_protocol_{protocol}\"""".format(turl=item.values()[0], protocol=item.keys()[0])
         outfile.write(outstring)
         outfile.close()
         os.system("chmod +x " + name)
   
   # in case of "AthDerivation"
   if "AthDerivation" in opts.athena_project:
     # loop over turl list
     for i in xrange(len(TFILE_OPEN_TURLs)):
       # get values of dictionaries
       for key, value in TFILE_OPEN_TURLs[i].items():
         name = os.path.join("test", "test_directioart_athderivation_recotf_with_aod_{}.sh".format(key.lower()))
         print("\tGenerating ... {}".format(name))
         outfile = open(name,'w')
         # generate test scripts
         outstring = """#!/bin/bash\n
 # art-description: DirectIOART AthDerivation Reco_tf.py inputFile:RDO protocol={protocol} reductionConf:TEST1
 # art-type: grid
 # art-output: *.pool.root
 # art-include: 21.2/AthDerivation\n
 set -e\n
 Reco_tf.py --inputAODFile {turl} --outputDAODFile art.pool.root --reductionConf TEST1 --maxEvents 100\n
 echo \"art-result: $? DirectIOART_AthDerivation_RecoTF_inputAOD_protocol_{protocol}_TEST1\"""".format(turl=value[0], protocol=key)
         outfile.write(outstring)
         outfile.close()
         os.system("chmod +x " + name)
  

◆ writeTFileOpen()

def generateARTtests.writeTFileOpen ( )

Definition at line 217 of file generateARTtests.py.

 def writeTFileOpen():
   name = os.path.join("test", "test_directioart_tfileopen.py")
   print("\tGenerating ... {}".format(name))
   outfile = open(name,'w')
  
   # generate frame of test scripts
   outstring = """#!/usr/bin/env python\n
 # art-description: DirectIOART TFile::Open
 # art-type: grid
 # art-include: main/Athena
 # art-include: main/AthSimulation
 # art-include: 21.0/Athena
 # art-include: 21.0/AthSimulation
 # art-include: 21.2/AthAnalysis
 # art-include: 21.2/AthDerivation
 # art-include: 21.2/AnalysisBase
 # art-include: 21.2/AnalysisTop
 # art-include: 21.3/Athena
 # art-include: 21.9/Athena\n
 import ROOT
 import os\n"""
   # loop over turl list
   for i in xrange(len(TFILE_OPEN_TURLs)):
     # get values of dictionaries
     for key, value in TFILE_OPEN_TURLs[i].items():
       # add turl to filenames string
       filenames = ""
       for j in xrange(len(value)):
         filenames += value[j]
         if j < len(value)-1: filenames += " | "
       outstring += """
 f = ROOT.TFile.Open(\"{turl}\")
 if f: print(\"art-result: 0 DirectIOART_TFileOpen_protocol_{protocol}\")
 else: print(\"art-result: 1 DirectIOART_TFileOpen_protocol_{protocol}\")
 del f\n""".format(turl=filenames, protocol=key)
       filenames = ""
   
   # back to main test scripts
   outstring += """
 outstr = \"\"\"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\" ?>
 <!-- Edited By POOL -->
 <!DOCTYPE POOLFILECATALOG SYSTEM \"InMemory\">
 <POOLFILECATALOG>
  
 </POOLFILECATALOG>\"\"\"
 with open(\"PoolFileCatalog.xml\", \"w\") as outfile:
   outfile.write(outstr)"""
   
   outfile.write(outstring)
   outfile.close()
   os.system("chmod +x " + name)
  

Variable Documentation

◆ DAOD_FILE

generateARTtests.DAOD_FILE

Definition at line 5 of file generateARTtests.py.

◆ HAMMERCLOUD_FILE

generateARTtests.HAMMERCLOUD_FILE

Definition at line 8 of file generateARTtests.py.

◆ PROJECTS

generateARTtests.PROJECTS

Definition at line 9 of file generateARTtests.py.

◆ RAW_FILE

generateARTtests.RAW_FILE

Definition at line 6 of file generateARTtests.py.

◆ RDO_FILE

generateARTtests.RDO_FILE

Definition at line 7 of file generateARTtests.py.

◆ RECO_TF_TURLS

generateARTtests.RECO_TF_TURLS

Definition at line 17 of file generateARTtests.py.

◆ TFILE_OPEN_TURLs

generateARTtests.TFILE_OPEN_TURLs

Definition at line 30 of file generateARTtests.py.

◆ XAODCHECKER_TURLS

generateARTtests.XAODCHECKER_TURLS

Definition at line 11 of file generateARTtests.py.

Functions

Variables