Belle II Software development
Public Member Functions | Protected Member Functions | List of all members
TestStdV0 Class Reference
Inheritance diagram for TestStdV0:

Public Member Functions

def test_stdkshorts_list (self)
 
def test_belle_list (self)
 
def test_stdlambdas_list (self)
 
def test_kshort_signals (self)
 

Protected Member Functions

def _check_list (self, std_function=stdV0s.stdKshorts, expected_modules=["ParticleLoader", "ParticleListManipulator", "TreeFitter", "ParticleSelector", "ParticleLoader", "ParticleCombiner", "TreeFitter", "ParticleSelector", "ParticleListManipulator"], expected_lists=['V0_ToFit', 'V0_ToFit', 'all', 'RD', 'merged'])
 

Detailed Description

Test case for standard V0 lists

Definition at line 21 of file test_std_v0.py.

Member Function Documentation

◆ _check_list()

def _check_list (   self,
  std_function = stdV0s.stdKshorts,
  expected_modules = [                 "ParticleLoader",                 "ParticleListManipulator",                 "TreeFitter",                 "ParticleSelector",                 "ParticleLoader",                 "ParticleCombiner",                 "TreeFitter",                 "ParticleSelector",                 "ParticleListManipulator"],
  expected_lists = ['V0_ToFit', 'V0_ToFit', 'all', 'RD', 'merged'] 
)
protected 
check that a given function works

Definition at line 24 of file test_std_v0.py.

37 expected_lists=['V0_ToFit', 'V0_ToFit', 'all', 'RD', 'merged']):
38 """check that a given function works"""
39 testpath = create_path()
40 std_function(path=testpath)
41
42 loaded_modules = []
43 built_lists = []
44 for module in testpath.modules():
45 loaded_modules.append(module.type())
46 for param in module.available_params():
47 if module.type() == 'ParticleLoader' and param.name == 'decayStrings':
48 name = param.values[0].split(':')[1].split(' -> ')[0]
49 built_lists.append(name)
50 if module.type() == 'ParticleListManipulator' and param.name == 'outputListName':
51 name = str(param.values).split(':')[1].split(' -> ')[0]
52 built_lists.append(name)
53 if module.type() == 'ParticleCombiner' and param.name == 'decayString':
54 name = param.values.split(':')[1].split(' -> ')[0]
55 built_lists.append(name)
56
57 # Check that we load the correct modules
58 self.assertListEqual(
59 loaded_modules, expected_modules,
60 f"Loaded modules do not match the expected ones (function {std_function.__name__})")
61
62 # Check that we load the correct particle lists
63 self.assertListEqual(
64 built_lists, expected_lists,
65 f"Built particles lists do not match the expected ones (function {std_function.__name__})")
66

◆ test_belle_list()

def test_belle_list (   self) 
check that the builder function works with the legacy Belle Kshorts list

Definition at line 71 of file test_std_v0.py.

71 def test_belle_list(self):
72 """check that the builder function works with the legacy Belle Kshorts list"""
73 expected_modules = ["ParticleLoader",
74 "ParticleListManipulator",
75 "ParticleSelector",
76 "ParticleVertexFitter",
77 "ParticleSelector"]
78 self._check_list(std_function=stdV0s.goodBelleKshort, expected_modules=expected_modules,
79 expected_lists=["legacyGoodKS", "legacyGoodKS"])
80

◆ test_kshort_signals()

def test_kshort_signals (   self) 
check the number of signals in K_S0:merged and K_S0:scaled lists

Definition at line 100 of file test_std_v0.py.

100 def test_kshort_signals(self):
101 """check the number of signals in K_S0:merged and K_S0:scaled lists"""
102
103 main = create_path()
104
105 inputfile = b2test_utils.require_file('analysis/1000_B_Jpsi_ks_pipi.root', 'validation', py_case=self)
106 ma.inputMdst(inputfile, path=main)
107
108 stdV0s.stdKshorts(path=main) # -> K_S0:merged
109 stdV0s.scaleErrorKshorts(path=main) # -> K_S0:scaled
110
111 ma.matchMCTruth('K_S0:merged', path=main)
112 ma.matchMCTruth('K_S0:scaled', path=main)
113
114 testFile = tempfile.NamedTemporaryFile()
115 ma.variablesToNtuple('K_S0:merged', ['isSignal', 'M'], filename=testFile.name, treename='merged', path=main)
116 ma.variablesToNtuple('K_S0:scaled', ['isSignal', 'M'], filename=testFile.name, treename='scaled', path=main)
117
118 process(main)
119
120 ntuplefile = TFile(testFile.name)
121 ntuple_merged = ntuplefile.Get('merged')
122 ntuple_scaled = ntuplefile.Get('scaled')
123
124 allSig_merged = ntuple_merged.GetEntries("isSignal == 1")
125 allSig_scaled = ntuple_scaled.GetEntries("isSignal == 1")
126
127 print(f"Number of signal K_S0:merged: {allSig_merged}")
128 print(f"Number of signal K_S0:scaled: {allSig_scaled}")
129
130 self.assertTrue(allSig_merged > 999, "Number of signal K_S0:merged is too small.")
131 self.assertTrue(allSig_scaled > 1000, "Number of signal K_S0:scaled is too small.")
132
133 tightMSig_merged = ntuple_merged.GetEntries("isSignal == 1 && M > 0.48 && M < 0.52")
134 tightMSig_scaled = ntuple_scaled.GetEntries("isSignal == 1 && M > 0.48 && M < 0.52")
135
136 print(f"Number of signal K_S0:merged with 0.48<M<0.52: {tightMSig_merged}")
137 print(f"Number of signal K_S0:scaled with 0.48<M<0.52: {tightMSig_scaled}")
138
139 self.assertTrue(tightMSig_merged > 962, "Number of signal K_S0:merged with 0.48<M<0.52 is too small.")
140 self.assertTrue(tightMSig_scaled > 962, "Number of signal K_S0:scaled with 0.48<M<0.52 is too small.")
141
142 print("Test passed, cleaning up.")
143
144
b2test_utils.require_file
def require_file(filename, data_type="", py_case=None)
Definition: __init__.py:54
stdV0s.stdKshorts
def stdKshorts(prioritiseV0=True, fitter='TreeFit', path=None, updateAllDaughters=False, writeOut=False)
Definition: stdV0s.py:17
stdV0s.scaleErrorKshorts
def scaleErrorKshorts(prioritiseV0=True, fitter='TreeFit', scaleFactors_V0=[1.125927, 1.058803, 1.205928, 1.066734, 1.047513], scaleFactorsNoPXD_V0=[1.125927, 1.058803, 1.205928, 1.066734, 1.047513], d0Resolution_V0=[0.001174, 0.000779], z0Resolution_V0=[0.001350, 0.000583], d0MomThr_V0=0.500000, z0MomThr_V0=0.00000, scaleFactors_RD=[1.149631, 1.085547, 1.151704, 1.096434, 1.086659], scaleFactorsNoPXD_RD=[1.149631, 1.085547, 1.151704, 1.096434, 1.086659], d0Resolution_RD=[0.00115328, 0.00134704], z0Resolution_RD=[0.00124327, 0.0013272], d0MomThr_RD=0.500000, z0MomThr_RD=0.500000, path=None)
Definition: stdV0s.py:101

◆ test_stdkshorts_list()

def test_stdkshorts_list (   self) 
check that the builder function works with the stdKshorts list

Definition at line 67 of file test_std_v0.py.

67 def test_stdkshorts_list(self):
68 """check that the builder function works with the stdKshorts list"""
69 self._check_list()
70

◆ test_stdlambdas_list()

def test_stdlambdas_list (   self) 
check that the builder function works with the stdLambdas list

Definition at line 81 of file test_std_v0.py.

81 def test_stdlambdas_list(self):
82 """check that the builder function works with the stdLambdas list"""
83 expected_modules = ["ParticleLoader",
84 "ParticleListManipulator",
85 "TreeFitter",
86 "ParticleSelector",
87 "DuplicateVertexMarker",
88 "ParticleSelector",
89 "ParticleLoader",
90 "ParticleLoader",
91 "ParticleCombiner",
92 "TreeFitter",
93 "ParticleSelector",
94 "DuplicateVertexMarker",
95 "ParticleSelector",
96 "ParticleListManipulator"]
97 expected_lists = ['V0_ToFit', 'V0_ToFit', 'all', 'all', 'RD', 'merged']
98 self._check_list(std_function=stdV0s.stdLambdas, expected_modules=expected_modules, expected_lists=expected_lists)
99
The documentation for this class was generated from the following file: