Skip to content
Snippets Groups Projects
Commit 277168b7 authored by Federico Betti's avatar Federico Betti Committed by Miroslav Saur
Browse files

remove branches with MASSWITHHYPOTHESES from qmtests

parent 4afb67a0
No related branches found
No related tags found
1 merge request!1208remove branches with MASSWITHHYPOTHESES from qmtests
......@@ -88,7 +88,7 @@ f.Close()
# Check there are no NaN values in the ntuple except where expected.
from DaVinciTests.QMTest.check_helpers import list_fields_with_nan
l_branches_with_nans = ['B0_TRUEP', 'B0_TRUEPT', 'B0_TRUEPX', 'B0_TRUEPY', 'B0_TRUEPZ', 'B0_TRUEENERGY', 'B0_TRUEORIGINVERTEX_X', 'B0_TRUEORIGINVERTEX_Y', 'B0_TRUEORIGINVERTEX_Z', 'B0_TRUEENDVERTEX_X', 'B0_TRUEENDVERTEX_Y', 'B0_TRUEENDVERTEX_Z', 'B0_MASSWITHHYPOTHESES', 'Kaon_PROBNN_D', 'Kaon_PROBNN_MU', 'Kaon_TRUEP', 'Kaon_TRUEPT', 'Kaon_TRUEPX', 'Kaon_TRUEPY', 'Kaon_TRUEPZ', 'Kaon_TRUEENERGY', 'Kaon_BREMENERGY', 'Kaon_BREMBENDCORR', 'Kaon_BREMPIDE', 'Kaon_ECALPIDE', 'Kaon_ECALPIDMU', 'Kaon_HCALPIDE', 'Kaon_HCALPIDMU', 'Kaon_ELECTRONSHOWEREOP', 'Kaon_ELECTRONSHOWERDLL', 'Kaon_ELECTRONMATCH_CHI2', 'Kaon_BREMHYPOMATCH_CHI2', 'Kaon_ELECTRONENERGY', 'Kaon_BREMHYPOENERGY', 'Kaon_BREMHYPODELTAX', 'Kaon_BREMTRACKBASEDENERGY', 'Kaon_HCALEOP', 'Ds_TRUEP', 'Ds_TRUEPT', 'Ds_TRUEPX', 'Ds_TRUEPY', 'Ds_TRUEPZ', 'Ds_TRUEENERGY', 'Ds_TRUEORIGINVERTEX_X', 'Ds_TRUEORIGINVERTEX_Y', 'Ds_TRUEORIGINVERTEX_Z', 'Ds_TRUEENDVERTEX_X', 'Ds_TRUEENDVERTEX_Y', 'Ds_TRUEENDVERTEX_Z', 'Ds_BPVCORRMERR', 'Ds_BPVLTIME', 'Ds_MASSWITHHYPOTHESES', 'pip_PROBNN_D', 'pip_PROBNN_MU', 'pip_TRUEP', 'pip_TRUEPT', 'pip_TRUEPX', 'pip_TRUEPY', 'pip_TRUEPZ', 'pip_TRUEENERGY', 'pip_BREMENERGY', 'pip_BREMBENDCORR', 'pip_BREMPIDE', 'pip_ECALPIDE', 'pip_ECALPIDMU', 'pip_HCALPIDE', 'pip_HCALPIDMU', 'pip_ELECTRONSHOWEREOP', 'pip_ELECTRONSHOWERDLL', 'pip_ELECTRONMATCH_CHI2', 'pip_BREMHYPOMATCH_CHI2', 'pip_ELECTRONENERGY', 'pip_BREMHYPOENERGY', 'pip_BREMHYPODELTAX', 'pip_BREMTRACKBASEDENERGY', 'pip_HCALEOP']
l_branches_with_nans = ['B0_TRUEP', 'B0_TRUEPT', 'B0_TRUEPX', 'B0_TRUEPY', 'B0_TRUEPZ', 'B0_TRUEENERGY', 'B0_TRUEORIGINVERTEX_X', 'B0_TRUEORIGINVERTEX_Y', 'B0_TRUEORIGINVERTEX_Z', 'B0_TRUEENDVERTEX_X', 'B0_TRUEENDVERTEX_Y', 'B0_TRUEENDVERTEX_Z', 'Kaon_PROBNN_D', 'Kaon_PROBNN_MU', 'Kaon_TRUEP', 'Kaon_TRUEPT', 'Kaon_TRUEPX', 'Kaon_TRUEPY', 'Kaon_TRUEPZ', 'Kaon_TRUEENERGY', 'Kaon_BREMENERGY', 'Kaon_BREMBENDCORR', 'Kaon_BREMPIDE', 'Kaon_ECALPIDE', 'Kaon_ECALPIDMU', 'Kaon_HCALPIDE', 'Kaon_HCALPIDMU', 'Kaon_ELECTRONSHOWEREOP', 'Kaon_ELECTRONSHOWERDLL', 'Kaon_ELECTRONMATCH_CHI2', 'Kaon_BREMHYPOMATCH_CHI2', 'Kaon_ELECTRONENERGY', 'Kaon_BREMHYPOENERGY', 'Kaon_BREMHYPODELTAX', 'Kaon_BREMTRACKBASEDENERGY', 'Kaon_HCALEOP', 'Ds_TRUEP', 'Ds_TRUEPT', 'Ds_TRUEPX', 'Ds_TRUEPY', 'Ds_TRUEPZ', 'Ds_TRUEENERGY', 'Ds_TRUEORIGINVERTEX_X', 'Ds_TRUEORIGINVERTEX_Y', 'Ds_TRUEORIGINVERTEX_Z', 'Ds_TRUEENDVERTEX_X', 'Ds_TRUEENDVERTEX_Y', 'Ds_TRUEENDVERTEX_Z', 'Ds_BPVCORRMERR', 'Ds_BPVLTIME', 'pip_PROBNN_D', 'pip_PROBNN_MU', 'pip_TRUEP', 'pip_TRUEPT', 'pip_TRUEPX', 'pip_TRUEPY', 'pip_TRUEPZ', 'pip_TRUEENERGY', 'pip_BREMENERGY', 'pip_BREMBENDCORR', 'pip_BREMPIDE', 'pip_ECALPIDE', 'pip_ECALPIDMU', 'pip_HCALPIDE', 'pip_HCALPIDMU', 'pip_ELECTRONSHOWEREOP', 'pip_ELECTRONSHOWERDLL', 'pip_ELECTRONMATCH_CHI2', 'pip_BREMHYPOMATCH_CHI2', 'pip_ELECTRONENERGY', 'pip_BREMHYPOENERGY', 'pip_BREMHYPODELTAX', 'pip_BREMTRACKBASEDENERGY', 'pip_HCALEOP']
l_branches_with_nans += ["B0_TRUEPRIMARYVERTEX_X", "B0_TRUEPRIMARYVERTEX_Y", "B0_TRUEPRIMARYVERTEX_Z", "Kaon_TRUEPRIMARYVERTEX_X", "Kaon_TRUEPRIMARYVERTEX_Y", "Kaon_TRUEPRIMARYVERTEX_Z"]
l_branches_with_nans += ["Ds_TRUEPRIMARYVERTEX_X", "Ds_TRUEPRIMARYVERTEX_Y", "Ds_TRUEPRIMARYVERTEX_Z", "pip_TRUEPRIMARYVERTEX_X", "pip_TRUEPRIMARYVERTEX_Y", "pip_TRUEPRIMARYVERTEX_Z"]
l_branches_with_nans += ["Kaon_IS_NOT_H","Kaon_IS_PHOTON","Kaon_SHOWER_SHAPE","pip_IS_NOT_H","pip_IS_PHOTON","pip_SHOWER_SHAPE"]
......
......@@ -50,10 +50,8 @@ if df.shape != (41, 26):
# Check there are no NaN values in the ntuple except where expected.
from DaVinciTests.QMTest.check_helpers import list_fields_with_nan
l_branches_with_nans = ['Bs_phi_mass_kk', 'Bs_phi_mass_kpi'] # MASSWITHHYPOTHESES variables
l_test = list_fields_with_nan(ntuple.name, "TDirectoryName/TTreeName")
if sorted(l_test) != sorted(l_branches_with_nans):
if len(l_test) > 0:
causes.append("Unexpected list of branches with NaN values")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment