NanoAOD/test/test-btvNano.py

0001 #!/bin/env python3
0002
0003 import ROOT
0004
0005 f = ROOT.TFile.Open("btvNanoMC_NANO.root")
0006 t = f.Get("Events")
0007
0008 eps = 1e-10
0009
0010 def checkHist(maxMean, maxRMS):
0011     h = ROOT.gPad.GetPrimitive("htemp")
0012     assert h.GetMean() < maxMean
0013     assert h.GetRMS() < maxRMS
0014
0015 # Check JetCand matching, must be exact
0016 t.Draw("JetPFCands_pt - PFCands_pt[JetPFCands_pFCandsIdx]")
0017 checkHist(eps, eps)
0018 t.Draw("GenJetCands_pt - GenCands_pt[GenJetCands_genCandsIdx]")
0019 checkHist(eps, eps)
0020
0021 # Check matching of candidates to jets, RMS should be smaller than jet radius / 2
0022 t.Draw("PFCands_eta[JetPFCands_pFCandsIdx] - Jet_eta[JetPFCands_jetIdx]")
0023 checkHist(1e-2, 0.2)
0024 t.Draw("PFCands_eta[FatJetPFCands_pFCandsIdx] - FatJet_eta[FatJetPFCands_jetIdx]")
0025 checkHist(1e-2, 0.4)
0026
0027 # Check matching of JetSVs to SVs, must be exact
0028 t.Draw("JetSVs_mass - SV_mass[JetSVs_sVIdx]", "JetSVs_sVIdx>-1")
0029 checkHist(eps, eps)
0030
0031 # Check matching of SVs to leading jet, RMS should be smaller than jet radius / 2
0032 t.Draw("Jet_eta[0] - SV_eta[JetSVs_sVIdx]", "JetSVs_sVIdx>-1 && JetSVs_jetIdx==0")
0033 checkHist(0.1, 0.2)
0034
0035 # Check matching of PFCands to SVs
0036 t.Draw("PFCands_eta[JetPFCands_pFCandsIdx] - SV_eta[JetSVs_sVIdx[JetPFCands_jetSVIdx]]", "JetPFCands_jetSVIdx>-1 && JetSVs_sVIdx[JetPFCands_jetSVIdx]>-1")
0037 checkHist(1e-2, 0.2)
0038
0039 # Check matching of PFCands to GenCands
0040 t.Draw("PFCands_eta - GenCands_eta[PFCands_genCandIdx]", "PFCands_genCandIdx>-1")
0041 checkHist(1e-3, 1e-2)
0042
0043 # Check matching of Muons to GenParts
0044 t.Draw("Muon_eta - GenPart_eta[Muon_genPartIdx]", "Muon_genPartIdx>-1")
0045 checkHist(1e-3, 1e-2)
0046
0047 # Check matching of GenCands from B hadrons to GenPart mothers
0048 t.Draw("GenCands_eta - GenPart_eta[GenCands_genPartMotherIdx]", "GenCands_genPartMotherIdx>1 && GenCands_isFromB==2")
0049 checkHist(1e-3, 0.2)