本文整理汇总了Python中heppy.framework.looper.Looper类的典型用法代码示例。如果您正苦于以下问题：Python Looper类的具体用法？Python Looper怎么用？Python Looper使用的例子？那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了Looper类的15个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: TestAnalysis_ee_Z_bb_fccsw
class TestAnalysis_ee_Z_bb_fccsw(unittest.TestCase):
def setUp(self):
random.seed(0xdeadbeef)
self.outdir = tempfile.mkdtemp()
import logging
logging.disable(logging.CRITICAL)
def tearDown(self):
shutil.rmtree(self.outdir)
logging.disable(logging.NOTSET)
def test_beff_cms_fccsw(self):
'''check that the analysis runs with papas inputs from cmssw
'''
from heppy.papas.detectors.CMS import cms
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_Z_bbbar_with_papas_rec.root'])
config.components[0].files = [fname]
for s in config.sequence:
if hasattr( s,'detector'):
s.detector = cms
self.looper = Looper( self.outdir, config,
nEvents=100,
nPrint=0 )
self.looper.loop()示例2: TestAnalysis_ee_Z
class TestAnalysis_ee_Z(unittest.TestCase):
def setUp(self):
random.seed(0xdeadbeef)
self.outdir = tempfile.mkdtemp()
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_Z_ddbar.root'])
config.components[0].files = [fname]
self.looper = Looper( self.outdir, config,
nEvents=100,
nPrint=0,
timeReport=True)
import logging
logging.disable(logging.CRITICAL)
def tearDown(self):
shutil.rmtree(self.outdir)
logging.disable(logging.NOTSET)
def test_analysis(self):
'''Check for an almost perfect match with reference.
Will fail if physics algorithms are modified,
so should probably be removed from test suite,
or better: be made optional.
'''
self.looper.loop()
self.looper.write()
rootfile = '/'.join([self.outdir,
'heppy.analyzers.GlobalEventTreeProducer.GlobalEventTreeProducer_1/tree.root'])
mean, sigma = plot(rootfile)
self.assertAlmostEqual(mean, 90.13, 1)
self.assertAlmostEqual(sigma, 10.54, 1)示例3: test_process_event
def test_process_event(self):
loop = Looper( self.outdir, config,
nEvents=None,
nPrint=0 )
loop.process(10)
self.assertEqual(loop.event.input.var1, 10)
loop.process(10)示例4: test_analysis_sorting
def test_analysis_sorting(self):
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_ZH_Zmumu_Hbb.root'])
config.components[0].files = [fname]
looper = Looper( self.outdir, config,
nEvents=50,
nPrint=0 )
looper.process(0)
self.assertTrue(test_sorted(looper.event.rec_particles))示例5: test_skip
def test_skip(self):
first = 10
loop = Looper( self.outdir, config,
nEvents=None,
firstEvent=first,
nPrint=0 )
loop.loop()
loop.write()
# input file has 200 entries
# we skip 10 entries, so we process 190.
self.assertEqual(loop.nEvProcessed, self.nevents-first)示例6: test_1
def test_1(self):
'''
'''
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_ZZ_4mu.root'])
config.components[0].files = [fname]
looper = Looper( self.outdir, config,
nEvents=50,
nPrint=0 )
looper.loop()
looper.write()示例7: test_all_events_processed
def test_all_events_processed(self):
loop = Looper( self.outdir, config,
nEvents=None,
nPrint=0 )
loop.loop()
loop.write()
logfile = open('/'.join([self.outdir, 'log.txt']))
nev_processed = None
for line in logfile:
if line.startswith('number of events processed:'):
nev_processed = int(line.split(':')[1])
logfile.close()
self.assertEqual(nev_processed, self.nevents)
# checking the looper itself.
self.assertEqual(loop.nEvProcessed, self.nevents)示例8: runLoop
def runLoop( comp, outDir, config, options):
if options.input is not None:
comp.files = [options.input]
fullName = '/'.join( [outDir, comp.name ] )
# import pdb; pdb.set_trace()
config.components = [comp]
memcheck = 2 if getattr(options,'memCheck',False) else -1
loop = Looper( fullName,
config,
options.nevents, 0,
nPrint = options.nprint,
timeReport = options.timeReport,
quiet = options.quiet,
memCheckFromEvent = memcheck,
stopFlag = _globalGracefulStopFlag)
# print loop
if options.iEvent is None:
loop.loop()
loop.write()
# print loop
else:
# loop.InitOutput()
iEvent = int(options.iEvent)
loop.process( iEvent )
return loop示例9: setUp
def setUp(self):
random.seed(0xDEADBEEF)
self.outdir = tempfile.mkdtemp()
fname = "/".join([os.environ["HEPPY"], "test/data/ee_ZH_Zmumu_Hbb.root"])
config.components[0].files = [fname]
self.looper = Looper(self.outdir, config, nEvents=50, nPrint=0, timeReport=True)
import logging
logging.disable(logging.CRITICAL)示例10: TestAnalysis_ee_Z
class TestAnalysis_ee_Z(unittest.TestCase):
def setUp(self):
random.seed(0xdeadbeef)
self.outdir = tempfile.mkdtemp()
import logging
#Alice: I turned this off otherwise no pdebug output. Ask Colin why it is on and if this is OK
#logging.disable(logging.CRITICAL)
def tearDown(self):
shutil.rmtree(self.outdir)
logging.disable(logging.NOTSET)
# NB If this test fails and If you expected the physics results to change
# then it means that the cpp version of papas will need to be updated.
# Please
# (1) take the physics_clic.txt file that is produced by this test and rename it to update
# the required_clic_physics_dd.txt file in heppy/test/data
# (2) notify alice to change papas cpp
def test_z_clic(self):
'''Check Z mass in ee->Z->ddbar (CLIC).
Will fail if physics algorithms are modified,
so should probably be removed from test suite,
or better: be made optional.
'''
random.seed(0xdeadbeef)
from heppy.papas.detectors.CLIC import clic
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_Z_ddbar.root'])
config.components[0].files = [fname]
for s in config.sequence:
if hasattr( s,'detector'):
s.detector = clic
if hasattr(s, 'debug_filename'):
s.debug_filename = 'physics_clic.txt'
self.looper = Looper( self.outdir, config,
nEvents=10,
nPrint=0 )
self.looper.loop()
self.looper.write()
self.assertEqual(0,os.system("source $HEPPY/test/data/pdebug_python_check.sh physics_clic.txt $HEPPY/test/data/required_clic_physics_dd.txt"))示例11: setUp
def setUp(self):
random.seed(0xdeadbeef)
self.outdir = tempfile.mkdtemp()
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_Z_ddbar.root'])
config.components[0].files = [fname]
self.looper = Looper( self.outdir, config,
nEvents=100,
nPrint=0,
timeReport=True)
import logging
logging.disable(logging.CRITICAL)示例12: TestAnalysis_ee_ZH
class TestAnalysis_ee_ZH(unittest.TestCase):
def setUp(self):
random.seed(0xDEADBEEF)
self.outdir = tempfile.mkdtemp()
fname = "/".join([os.environ["HEPPY"], "test/data/ee_ZH_Zmumu_Hbb.root"])
config.components[0].files = [fname]
self.looper = Looper(self.outdir, config, nEvents=50, nPrint=0, timeReport=True)
import logging
logging.disable(logging.CRITICAL)
def tearDown(self):
shutil.rmtree(self.outdir)
logging.disable(logging.NOTSET)
def test_analysis(self):
"""Check for an almost perfect match with reference.
Will fail if physics algorithms are modified,
so should probably be removed from test suite,
or better: be made optional.
"""
self.looper.loop()
self.looper.write()
rootfile = "/".join([self.outdir, "heppy.analyzers.examples.zh.ZHTreeProducer.ZHTreeProducer_1/tree.root"])
mean, sigma = plot(rootfile)
self.assertAlmostEqual(mean, 120.7, 1)
self.assertAlmostEqual(sigma, 20.3, 1)
def test_analysis_sorting(self):
self.looper.process(0)
self.assertTrue(test_sorted(self.looper.event.rec_particles))示例13: TestAnalysis_ee_Z_bb
class TestAnalysis_ee_Z_bb(unittest.TestCase):
def setUp(self):
random.seed(0xdeadbeef)
self.outdir = tempfile.mkdtemp()
import logging
logging.disable(logging.CRITICAL)
def tearDown(self):
shutil.rmtree(self.outdir)
logging.disable(logging.NOTSET)
def test_beff_cms(self):
'''Check b matching probability and b tag efficiency in CMS
'''
from heppy.papas.detectors.CMS import cms
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_Z_bbbar.root'])
config.components[0].files = [fname]
for s in config.sequence:
if hasattr( s,'detector'):
s.detector = cms
self.looper = Looper( self.outdir, config,
nEvents=500,
nPrint=0 )
self.looper.loop()
self.looper.write()
rootfile = '/'.join([self.outdir,
'heppy.analyzers.JetTreeProducer.JetTreeProducer_1/jet_tree.root '])
plotter = Plotter(rootfile)
self.assertAlmostEqual(plotter.bfrac(), 0.867, places=2)
self.assertAlmostEqual(plotter.beff(), 0.71, places=2)
def test_fake_cms(self):
'''Check fake rate in CMS
'''
from heppy.papas.detectors.CMS import cms
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_Z_ddbar.root'])
config.components[0].files = [fname]
for s in config.sequence:
if hasattr( s,'detector'):
s.detector = cms
self.looper = Looper( self.outdir, config,
nEvents=100,
nPrint=0 )
self.looper.loop()
self.looper.write()
rootfile = '/'.join([self.outdir,
'heppy.analyzers.JetTreeProducer.JetTreeProducer_1/jet_tree.root '])示例14: test_analysis_ee_ZH_mumubb
def test_analysis_ee_ZH_mumubb(self):
'''Check for an almost perfect match with reference.
Will fail if physics algorithms are modified,
so should probably be removed from test suite,
or better: be made optional.
'''
from heppy.papas.detectors.CMS import cms
for s in config.sequence:
if hasattr( s,'detector'):
s.detector = cms
if hasattr(s, 'debug_filename'):
s.debug_filename = 'physics_cms.txt'
# import pdb; pdb.set_trace()
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_ZH_Zmumu_Hbb.root'])
config.components[0].files = [fname]
looper = Looper( self.outdir, config,
nEvents=10,
nPrint=0 )
looper.loop()
looper.write()
self.assertEqual(0,os.system("source $HEPPY/test/data/pdebug_python_check.sh physics_cms.txt $HEPPY/test/data/required_cms_physics.txt"))示例15: test_beff_cms_fccsw
def test_beff_cms_fccsw(self):
'''check that the analysis runs with papas inputs from cmssw
'''
from heppy.papas.detectors.CMS import cms
fname = '/'.join([os.environ['HEPPY'],
'test/data/ee_Z_bbbar_with_papas_rec.root'])
config.components[0].files = [fname]
for s in config.sequence:
if hasattr( s,'detector'):
s.detector = cms
self.looper = Looper( self.outdir, config,
nEvents=100,
nPrint=0 )
self.looper.loop()