Belle II Software  release-05-01-25
treeFit_referenced_pipipi0.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 import unittest
4 import os
5 import tempfile
6 import basf2
7 import b2test_utils
8 import modularAnalysis as ma
9 from ROOT import Belle2
10 from ROOT import TFile
11 from ROOT import TNtuple
12 
13 
14 class TestTreeFits(unittest.TestCase):
15  """The unit test case for TreeFitter"""
16 
17  def testFit(self):
18  """Run the test fit"""
19 
20  testFile = tempfile.NamedTemporaryFile()
21 
22  main = basf2.create_path()
23 
24  inputfile = b2test_utils.require_file('analysis/tests/100_noBKG_B0ToPiPiPi0.root', py_case=self)
25  ma.inputMdst('default', inputfile, path=main)
26 
27  ma.fillParticleList('pi+:a', 'pionID > 0.5', path=main)
28 
29  ma.fillParticleList('gamma:a', '', path=main)
30  ma.reconstructDecay('pi0:a -> gamma:a gamma:a', '0.125 < InvM < 0.145', 0, path=main)
31 
32  ma.reconstructDecay('B0:rec -> pi-:a pi+:a pi0:a', '', 0, path=main)
33  ma.matchMCTruth('B0:rec', path=main)
34 
35  conf = 0
36  main.add_module('TreeFitter',
37  particleList='B0:rec',
38  confidenceLevel=conf,
39  massConstraintList=[],
40  massConstraintListParticlename=[],
41  expertUseReferencing=True,
42  ipConstraint=False,
43  updateAllDaughters=True)
44 
45  ntupler = basf2.register_module('VariablesToNtuple')
46  ntupler.param('fileName', testFile.name)
47  ntupler.param('variables', ['chiProb', 'M', 'isSignal'])
48  ntupler.param('particleList', 'B0:rec')
49  main.add_module(ntupler)
50 
51  basf2.process(main)
52 
53  ntuplefile = TFile(testFile.name)
54  ntuple = ntuplefile.Get('ntuple')
55 
56  self.assertFalse(ntuple.GetEntries() == 0, "Ntuple is empty.")
57 
58  allBkg = ntuple.GetEntries("isSignal == 0")
59  allSig = ntuple.GetEntries("isSignal > 0")
60 
61  truePositives = ntuple.GetEntries("(chiProb > 0) && (isSignal > 0)")
62  falsePositives = ntuple.GetEntries("(chiProb > 0) && (isSignal == 0)")
63 
64  mustBeZero = ntuple.GetEntries("(chiProb < {})".format(conf))
65 
66  print("True fit survivors: {0} out of {1} true candidates".format(truePositives, allSig))
67  print("False fit survivors: {0} out of {1} false candidates".format(falsePositives, allBkg))
68 
69  self.assertFalse(truePositives == 0, "No signal survived the fit.")
70 
71  self.assertTrue(falsePositives < 2596, "Background rejection increased.")
72 
73  self.assertTrue(truePositives > 34, "Signal rejection too high")
74  self.assertFalse(mustBeZero, "We should have dropped all candidates with confidence level less than {}.".format(conf))
75 
76  print("Test passed, cleaning up.")
77 
78 
79 if __name__ == '__main__':
80  unittest.main()
treeFit_referenced_pipipi0.TestTreeFits.testFit
def testFit(self)
Definition: treeFit_referenced_pipipi0.py:17
basf2.process
def process(path, max_event=0)
Definition: __init__.py:25
treeFit_referenced_pipipi0.TestTreeFits
Definition: treeFit_referenced_pipipi0.py:14
b2test_utils.require_file
def require_file(filename, data_type="", py_case=None)
Definition: __init__.py:47