22 """Run the test fit"""
24 testFile = tempfile.NamedTemporaryFile()
26 main = basf2.create_path()
28 inputfile = b2test_utils.require_file(
29 'analysis/1000_B_Jpsi_ks_pipi.root', 'validation', py_case=self)
30 ma.inputMdst(inputfile, path=main)
32 ma.fillParticleList('pi+:a', 'pionID > 0.5', path=main)
34 ma.reconstructDecay('K_S0:pipi -> pi+:a pi-:a', '', 0, path=main)
35 ma.matchMCTruth('K_S0:pipi', path=main)
38 main.add_module('TreeFitter',
39 particleList='K_S0:pipi',
41 massConstraintList=[],
43 updateAllDaughters=False)
45 ma.printMCParticles(path=main, suppressPrint=True, showStatus=True, showMomenta=True)
47 ntupler = basf2.register_module('VariablesToNtuple')
48 ntupler.param('fileName', testFile.name)
49 ntupler.param('variables', ['chiProb', 'M', 'isSignal', 'mcErrors', 'genParticleID',
50 'daughter(0,genParticleID)', 'daughter(1,genParticleID)'])
51 ntupler.param('particleList', 'K_S0:pipi')
52 main.add_module(ntupler)
56 ntuplefile = TFile(testFile.name)
57 ntuple = ntuplefile.Get('ntuple')
59 self.assertFalse(ntuple.GetEntries() == 0, "Ntuple is empty.")
61 allBkg = ntuple.GetEntries("isSignal == 0")
62 allSig = ntuple.GetEntries("isSignal > 0")
64 truePositives = ntuple.GetEntries("(chiProb > 0) && (isSignal > 0)")
65 falsePositives = ntuple.GetEntries("(chiProb > 0) && (isSignal == 0)")
67 mustBeZero = ntuple.GetEntries(f"(chiProb < {conf})")
69 print(f"True fit survivors: {truePositives} out of {allSig} true candidates")
70 print(f"False fit survivors: {falsePositives} out of {allBkg} false candidates")
72 self.assertFalse(truePositives == 0, "No signal survived the fit.")
74 self.assertTrue(falsePositives <= 3449, f"Too many false positives: {falsePositives} out of {allBkg} total bkg events.")
76 self.assertTrue(truePositives == 555, "Signal rejection too high")
77 self.assertFalse(mustBeZero, f"We should have dropped all candidates with confidence level less than {conf}.")
79 print("Test passed, cleaning up.")