def testDataMcGiven(self): puweight = pileupWeight(data="2012D", mc="Summer12_S10") self.assertEqual(puweight.enabled, True) self.assertEqual( puweight.data, os.path.join(aux.higgsAnalysisPath(), _pileupHistogramPath, "PileupHistogramData2012D.root")) self.assertEqual( puweight.mc, os.path.join(aux.higgsAnalysisPath(), _pileupHistogramPath, "PileupHistogramMCSummer12_S10.root"))
def testConstruct(self): f = File("NtupleAnalysis/python/main.py") self.assertEqual( f, os.path.join(aux.higgsAnalysisPath(), "NtupleAnalysis/python/main.py")) self.assertRaises(Exception, File, "NtupleFoo")
def getFiles(datasetName): basedir = os.path.join(aux.higgsAnalysisPath(), "NtupleAnalysis") name = _aliases.get(datasetName, datasetName) path = os.path.join(basedir, "data", name+".txt") if not os.path.exists(path): raise Exception("No files for dataset %s (i.e. file %s not found)" % (name, path)) def strip(line): ret = line.rstrip() if "#" in ret: ret = ret[0:ret.index("#")] return ret with open(path) as f: return filter(lambda l: len(l) > 0, map(strip, f.readlines()))
def getFiles(datasetName): basedir = os.path.join(aux.higgsAnalysisPath(), "NtupleAnalysis") name = _aliases.get(datasetName, datasetName) path = os.path.join(basedir, "data", name + ".txt") if not os.path.exists(path): raise Exception("No files for dataset %s (i.e. file %s not found)" % (name, path)) def strip(line): ret = line.rstrip() if "#" in ret: ret = ret[0:ret.index("#")] return ret with open(path) as f: return filter(lambda l: len(l) > 0, map(strip, f.readlines()))
def testSerialize(self): a = PSet(foo=File("NtupleAnalysis/python/main.py")) self.assertEqual( a.serialize_(), """{ "foo": "%s/NtupleAnalysis/python/main.py" }""" % aux.higgsAnalysisPath())
def File(fname): fullpath = os.path.join(aux.higgsAnalysisPath(), fname) if not os.path.exists(fullpath): raise Exception("The file %s does not exist" % self._fullpath) return fullpath
def testSerialize(self): a = PSet(foo=File("NtupleAnalysis/python/main.py")) self.assertEqual(a.serialize_(), """{ "foo": "%s/NtupleAnalysis/python/main.py" }""" % aux.higgsAnalysisPath())
def testConstruct(self): f = File("NtupleAnalysis/python/main.py") self.assertEqual(f, os.path.join(aux.higgsAnalysisPath(), "NtupleAnalysis/python/main.py")) self.assertRaises(Exception, File, "NtupleFoo")
# Example of adding a dataset which has its files defined in data/<dataset_name>.txt file #process.addDatasets(["TTbar_HBWB_HToTauNu_M_160_13TeV_pythia6"]) # Example of adding datasets from a multicrab directory import sys if len(sys.argv) != 2: print "Usage: ./exampleAnalysis.py <path-to-multicrab-directory>" sys.exit(0) #print sys.argv[1] process.addDatasetsFromMulticrab(sys.argv[1]) import HiggsAnalysis.NtupleAnalysis.tools.aux as aux PileupHistogramPath = os.path.join(aux.higgsAnalysisPath(), "NtupleAnalysis", "data", "PUWeights") process.addAnalyzer("generatorComparison", Analyzer("GeneratorComparison", histogramAmbientLevel = "Informative", tauPtCut = 41.0, tauEtaCut = 2.1, bjetEtCut = 30.0, bjetEtaCut = 2.4, lumi = 7274, runMin = 202807, runMax = 208686, ), #includeOnlyTasks="TauPlusX_\S+_2012D_Jan22" )
def testDataMcGiven(self): puweight = pileupWeight(data="2012D", mc="Summer12_S10") self.assertEqual(puweight.enabled, True) self.assertEqual(puweight.data, os.path.join(aux.higgsAnalysisPath(), _pileupHistogramPath, "PileupHistogramData2012D.root")) self.assertEqual(puweight.mc, os.path.join(aux.higgsAnalysisPath(), _pileupHistogramPath, "PileupHistogramMCSummer12_S10.root"))