def test_combination_index(self):
# Test a simple combination in the barrel
layers = [5, 6, 7, 8, 9, 10]
radius = [0]*6
self.assertEquals(combinations.combination_index(layers, radius), 2016)
# Test one with a disk (PS)
layers = [5, 6, 7, 8, 9, 11]
self.assertEquals(combinations.combination_index(layers, radius), 3040)
# Test one with a disk (2S)
radius[5] = 70
self.assertEquals(combinations.combination_index(layers, radius), 2100192)
def fit_and_draw(base_var, h_name, layers_list, radius_list):
comb_index = combination_index(layers_list, radius_list)
h = [gaussian_fit(base_var+"_"+str(comb_index)+".root", h_name.replace("summary", str(comb_index)))]
for layer in layers_list:
layers_list_removed = list(layers_list)
radius_list_removed = list(radius_list)
layers_list_removed.remove(layer)
del radius_list_removed[layers_list.index(layer)]
comb_index = combination_index(layers_list_removed, radius_list_removed)
h.append(gaussian_fit(base_var+"_"+str(comb_index)+".root", h_name.replace("summary", str(comb_index))))
fit_results = []
for i in range(len(h)):
fit_results.append(h[i][1])
return h, fit_results
