22 def test_roepath(self):
23 """ Test the extractParticlesFromROE function in roe_path """
24
25 testFile = tempfile.NamedTemporaryFile()
26
27 main = basf2.create_path()
28
30 ma.inputMdst(inputfile, path=main)
31
32 ma.fillParticleList('pi+:sig', 'pionID > 0.5', path=main)
33
34 ma.fillParticleList('gamma:all', '', path=main)
35 ma.reconstructDecay('pi0:sig -> gamma:all gamma:all', '0.125 < InvM < 0.145', path=main)
36
37 ma.reconstructDecay('B0:sig -> pi-:sig pi+:sig pi0:sig', '', path=main)
38
39 ma.matchMCTruth('B0:sig', path=main)
40 ma.applyCuts('B0:sig', 'isSignal==1', path=main)
41
42 ma.buildRestOfEvent('B0:sig', fillWithMostLikely=True, path=main)
43
44 roe_path = basf2.create_path()
45 deadEndPath = basf2.create_path()
46 ma.signalSideParticleFilter('B0:sig', '', roe_path, deadEndPath)
47
48 plists = [f'{ptype}:in_roe' for ptype in ['pi+', 'gamma', 'K_L0', 'K+', 'p+', 'e+', 'mu+']]
49 ma.extractParticlesFromROE(plists, maskName='all', path=roe_path)
50
51 charged_inROE = [f'nParticlesInList({ptype}:in_roe)' for ptype in ['pi+', 'K+', 'p+', 'e+', 'mu+']]
52 neutral_inROE = [f'nParticlesInList({ptype}:in_roe)' for ptype in ['gamma', 'K_L0']]
53
54 default = ['nParticlesInList(pi+:all)', 'nParticlesInList(gamma:all)', 'nParticlesInList(K_L0:roe_default)']
55 mostLikely = [f'nParticlesInList({ptype}:mostlikely_roe)' for ptype in ['K+', 'p+', 'e+', 'mu+']]
56
57 ma.variablesToNtuple('', charged_inROE + neutral_inROE + default + mostLikely,
58 filename=testFile.name,
59 path=roe_path)
60
61 main.for_each('RestOfEvent', 'RestOfEvents', roe_path)
62
63 basf2.process(main)
64
65 ntuplefile = TFile(testFile.name)
66 ntuple = ntuplefile.Get('variables')
67
68 self.assertFalse(ntuple.GetEntries() == 0, "Ntuple is empty.")
69
70 nEntries = ntuple.GetEntries()
71
72
75 nPass_chargedCheck = ntuple.GetEntries(cut)
76 self.assertFalse(nPass_chargedCheck < nEntries, "Charged particles are not correctly extracted")
77
78
81 nPass_electronCheck = ntuple.GetEntries(cut)
82 self.assertFalse(nPass_electronCheck < nEntries, "MostLikely option seems broken")
83