26 """Run the test fit"""
28 testFile = tempfile.NamedTemporaryFile()
30 main = basf2.create_path()
32 ma.inputMdst(b2test_utils.require_file('analysis/tests/150_noBKG_DtoPiNuNu.root'), path=main)
33 ma.fillParticleList('pi+:a', 'pionID > 0.5', path=main)
34 ma.fillParticleList('K+:a', 'kaonID > 0.5', path=main)
37 "D-:tag -> K+:a pi-:a pi-:a",
38 cut='1.859 < M < 1.879 and useCMSFrame(p)>2.0',
48 ma.reconstructDecay('Z0:rec -> D+:sig D-:tag pi-:a pi+:a', '', path=main)
49 ma.matchMCTruth('Z0:rec', path=main)
51 ma.setBeamConstrainedMomentum(
53 '^Z0:rec -> D+:sig D-:tag pi-:a pi+:a',
54 'Z0:rec -> D+:sig D-:tag pi-:a pi+:a',
58 ma.setBeamConstrainedMomentum(
60 'Z0:rec -> ^D+:sig D-:tag pi-:a pi+:a',
61 'Z0:rec -> D+:sig ^D-:tag ^pi-:a ^pi+:a',
66 main.add_module('TreeFitter',
67 particleList='Z0:rec',
69 massConstraintList=[],
70 expertBeamConstraintPDG=23,
71 treatAsInvisible='Z0:rec -> ^D+:sig D-:tag pi-:a pi+:a',
73 updateAllDaughters=True)
75 ma.applyCuts('Z0:rec',
76 "1.6<daughter(0,M)<2.2",
80 ma.rankByHighest("Z0:rec",
85 vm.addAlias("signalD", "daughter(0,isSignalAcceptMissingNeutrino)")
87 ntupler = basf2.register_module('VariablesToNtuple')
88 ntupler.param('fileName', testFile.name)
89 ntupler.param('variables', ['chiProb', 'M', 'signalD'])
90 ntupler.param('particleList', 'Z0:rec')
91 main.add_module(ntupler)
95 ntuplefile = TFile(testFile.name)
96 ntuple = ntuplefile.Get('ntuple')
98 self.assertFalse(ntuple.GetEntries() == 0, "Ntuple is empty.")
100 allBkg = ntuple.GetEntries("signalD == 0")
101 allSig = ntuple.GetEntries("signalD > 0")
103 truePositives = ntuple.GetEntries("(chiProb > 0) && (signalD > 0)")
104 falsePositives = ntuple.GetEntries("(chiProb > 0) && (signalD == 0)")
106 mustBeZero = ntuple.GetEntries(f"(chiProb < {conf})")
108 print(f"True fit survivors: {truePositives} out of {allSig} true candidates")
109 print(f"False fit survivors: {falsePositives} out of {allBkg} false candidates")
111 self.assertFalse(truePositives == 0, "No signal survived the fit.")
113 self.assertTrue(falsePositives < 1, "Background rejection too small.")
115 self.assertTrue(truePositives > 8, "Signal rejection too high")
116 self.assertFalse(mustBeZero, f"We should have dropped all candidates with confidence level less than {conf}.")
118 print("Test passed, cleaning up.")