def test_gen_load(self): aliases = gen.fs_content() zjets_cutflow = gen.filter(aliases, { "name": "cutflow", "sample": "zjets" }) wrp = gen.load(zjets_cutflow).next() self.assertTrue(isinstance(wrp.histo, TH1F)) self.assertAlmostEqual(wrp.histo.Integral(), 2889.0)
def gen_mixer_input(input_dict): for d in input_dict: yield op.norm_to_integral( op.merge( list( gen.load( gen.filter( gen.fs_content(), d )))))
def run(self): wrps = gen.fs_content() wrps = gen.filter( wrps, {"analyzer": re.compile("CutFlow*")} ) wrps = gen.filter_active_samples(wrps) wrps = sorted(wrps, key = sample_key_func) wrps = gen.load(wrps) grps = gen.group(wrps, sample_key_func) wrps = (self.combine_cutflow_histo(g) for g in grps) wrps = gen.pool_store_items(wrps) wrps = list(wrps) self.result = wrps count = len(wrps) self.message("INFO: "+self.name+" stored "+str(count)+" histos in pool.")
def test_gen_load(self): aliases = gen.fs_content() zjets_cutflow = gen.filter(aliases, {"name": "cutflow", "sample": "zjets"}) wrp = gen.load(zjets_cutflow).next() self.assertTrue(isinstance(wrp.histo, TH1F)) self.assertAlmostEqual(wrp.histo.Integral(), 2889.0)