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