TestParticleExtractorFromROE Class Reference

Inheritance diagram for TestParticleExtractorFromROE:

Collaboration diagram for TestParticleExtractorFromROE:

Public Member Functions
def	test_roepath (self)
def	test_mainpath (self)

Detailed Description

The unit test for ParticleExtractorFromROE module 

Definition at line 19 of file test_particleExtractorFromROE.py.

Member Function Documentation

test_mainpath()

def test_mainpath

(

self

)

 Test the extractParticlesFromROE function in main path 

Definition at line 84 of file test_particleExtractorFromROE.py.

    def test_mainpath(self):
        """ Test the extractParticlesFromROE function in main path """
 
        testFile = tempfile.NamedTemporaryFile()
 
        main = basf2.create_path()
 
        inputfile = b2test_utils.require_file('analysis/B0ToPiPiPi0.root', 'validation', py_case=self)
        ma.inputMdst(inputfile, path=main)
 
        ma.fillParticleList('pi+:sig', 'pionID > 0.5', path=main)
 
        ma.fillParticleList('gamma:all', '', path=main)
        ma.reconstructDecay('pi0:sig -> gamma:all gamma:all', '0.125 < InvM < 0.145', path=main)
 
        ma.reconstructDecay('B0:sig -> pi-:sig pi+:sig pi0:sig', '', path=main)
 
        ma.matchMCTruth('B0:sig', path=main)
        ma.applyCuts('B0:sig', 'isSignal==1', path=main)  # there must be only 1 candidate
 
        ma.buildRestOfEvent('B0:sig', fillWithMostLikely=True, path=main)
 
        plists = ['%s:in_roe' % ptype for ptype in ['pi+', 'gamma', 'K_L0', 'K+', 'p+', 'e+', 'mu+']]
        ma.extractParticlesFromROE(plists, maskName='all', path=main, signalSideParticleList='B0:sig')
 
        charged_inROE = ['nParticlesInList(%s:in_roe)' % ptype for ptype in ['pi+', 'K+', 'p+', 'e+', 'mu+']]
        neutral_inROE = ['nParticlesInList(%s:in_roe)' % ptype for ptype in ['gamma', 'K_L0']]
 
        default = ['nParticlesInList(pi+:all)', 'nParticlesInList(gamma:all)', 'nParticlesInList(K_L0:roe_default)']
        mostLikely = ['nParticlesInList(%s:mostlikely_roe)' % ptype for ptype in ['K+', 'p+', 'e+', 'mu+']]
 
        ma.variablesToNtuple('B0:sig', charged_inROE + neutral_inROE + default + mostLikely,
                             filename=testFile.name,
                             path=main)
 
        basf2.process(main)
 
        ntuplefile = TFile(testFile.name)
        ntuple = ntuplefile.Get('variables')
 
        self.assertFalse(ntuple.GetEntries() == 0, "Ntuple is empty.")
 
        nEntries = ntuple.GetEntries()
 
        # Number of pi+:all = Number of charged-FSPs in ROE + 2-pions in signal side
        cut = Belle2.MakeROOTCompatible.makeROOTCompatible('nParticlesInList(pi+:all)') + ' == ' \
            '(' + '+'.join([Belle2.MakeROOTCompatible.makeROOTCompatible(charged) for charged in charged_inROE]) + ' + 2)'
        nPass_chargedCheck = ntuple.GetEntries(cut)
        self.assertFalse(nPass_chargedCheck < nEntries, "Charged particles are not correctly extracted")
 
        # Number of mostLikely = Number of charged in ROE
        cut = Belle2.MakeROOTCompatible.makeROOTCompatible('nParticlesInList(e+:mostlikely_roe)') + ' == ' \
            + Belle2.MakeROOTCompatible.makeROOTCompatible('nParticlesInList(e+:in_roe)')
        nPass_electronCheck = ntuple.GetEntries(cut)
        self.assertFalse(nPass_electronCheck < nEntries, "MostLikely option seems broken")
 
 

test_roepath()

def test_roepath

(

self

)

 Test the extractParticlesFromROE function in roe_path 

Definition at line 22 of file test_particleExtractorFromROE.py.

---

The documentation for this class was generated from the following file:

• analysis/tests/test_particleExtractorFromROE.py