def test_zero_centred_error(self):
     """Test 0-centred band (ERROR)"""
     fn0 = get_fns('TMath::Gaus', (0, 2), (-10, 10))
     fn1 = get_fns('TMath::Gaus', (0, 1), (-10, 10))
     self.hdata.FillRandom(fn0[0], 20000)
     self.href.FillRandom(fn1[0], 10000)
     cmpfn = ZeroCentredBandRef()
     dqscore = cmpfn.compare(self.hdata, self.href, 3)
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.ERROR)
     dqscore = cmpfn.compare(self.hdata, self.href, None)
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.ERROR)
    def setUp(self):
        self.href = TH1D('href1', '', 100, -10, 10)
        self.hdata = TH1D('hdata1', '', 100, -10, 10)
        self.hdata_bad = TH1D('hdata_bad', '', 100, -10, 10)

        fn = get_fns('TMath::Gaus', (0, 3), (-10, 10))
        self.href.FillRandom(fn[0], 10000)
        fn = get_fns('TMath::Gaus', (0, 3), (-10, 10))
        self.hdata.FillRandom(fn[0], 10000)
        fn = get_fns('TMath::Gaus', (1, 3), (-10, 10))
        self.hdata_bad.FillRandom(fn[0], 10000)
 def test_mean_width_diff_error(self):
     """Test mean and width check with reference (ERROR)"""
     fn0 = get_fns('TMath::Gaus', (0, 3), (-10, 10))
     fn1 = get_fns('TMath::Gaus', (1, 2), (-10, 10))
     self.hdata.FillRandom(fn0[0], 20000)
     self.href.FillRandom(fn1[0], 10000)
     cmpfn = MeanWidthDiffRef()
     # mean = 0 & 1, width = 3 & 2 (RMS)
     dqscore = cmpfn.compare(self.hdata, self.href, 0.1)
     self.assertTrue(dqscore['score'] < Score(30))
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.ERROR)
 def test_mean_width_diff_error(self):
     """Test mean and width check with reference (ERROR)"""
     fn0 = get_fns('TMath::Gaus', (0, 3), (-10, 10))
     fn1 = get_fns('TMath::Gaus', (1, 2), (-10, 10))
     self.hdata.FillRandom(fn0[0], 20000)
     self.href.FillRandom(fn1[0], 10000)
     cmpfn = MeanWidthDiffRef()
     # mean = 0 & 1, width = 3 & 2 (RMS)
     dqscore = cmpfn.compare(self.hdata, self.href, 0.1)
     self.assertTrue(dqscore['score'] < Score(30))
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.ERROR)
 def test_zero_centred_error(self):
     """Test 0-centred band (ERROR)"""
     fn0 = get_fns('TMath::Gaus', (0, 2), (-10, 10))
     fn1 = get_fns('TMath::Gaus', (0, 1), (-10, 10))
     self.hdata.FillRandom(fn0[0], 20000)
     self.href.FillRandom(fn1[0], 10000)
     cmpfn = ZeroCentredBandRef()
     dqscore = cmpfn.compare(self.hdata, self.href, 3)
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.ERROR)
     dqscore = cmpfn.compare(self.hdata, self.href, None)
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.ERROR)
Esempio n. 6
0
    def setUp(self):
        self.href = TH1D('href1', '', 100, -10, 10)
        self.hdata = TH1D('hdata1', '', 100, -10 ,10)
        self.hdata_bad = TH1D('hdata_bad', '', 100, -10 ,10)

        fn = get_fns('TMath::Gaus', (0, 3), (-10, 10))
        self.href.FillRandom(fn[0], 10000)
        fn = get_fns('TMath::Gaus', (0, 3), (-10, 10))
        self.hdata.FillRandom(fn[0], 10000)
        fn = get_fns('TMath::Gaus', (1, 3), (-10, 10))
        self.hdata_bad.FillRandom(fn[0], 10000)
 def test_mean_width_diff_ok(self):
     """Test mean and width check with reference (OK)"""
     fn = get_fns('TMath::Gaus', (0, 3), (-10, 10))
     self.hdata.FillRandom(fn[0], 20000)
     self.href.FillRandom(fn[0], 10000)
     cmpfn = MeanWidthDiffRef()
     # mean = 0, width = 3 (RMS)
     dqscore = cmpfn.compare(self.hdata, self.href, 0.1)
     self.assertTrue(dqscore['score'] > Score(90))
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.OK)
 def test_mean_width_diff_ok(self):
     """Test mean and width check with reference (OK)"""
     fn = get_fns('TMath::Gaus', (0, 3), (-10, 10))
     self.hdata.FillRandom(fn[0], 20000)
     self.href.FillRandom(fn[0], 10000)
     cmpfn = MeanWidthDiffRef()
     # mean = 0, width = 3 (RMS)
     dqscore = cmpfn.compare(self.hdata, self.href, 0.1)
     self.assertTrue(dqscore['score'] > Score(90))
     self.assertEqual(dqscore['lvl'], ERROR_LEVELS.OK)
Esempio n. 9
0
    def setUp(self):
        """Create dictionaries and ROOT files needed for testing."""

        self.rfdata = TFile('/tmp/fdata.root', 'recreate')
        self.rfref = TFile('/tmp/fref.root', 'recreate')

        fns = {}
        fns['pol0'] = get_simple_fns('5', (-10, 10))
        fns['gauss01'] = get_fns('TMath::Gaus', (0, 1), (-10, 10))
        fns['gauss02'] = get_fns('TMath::Gaus', (0, 2), (-10, 10))
        fns['gauss03'] = get_fns('TMath::Gaus', (0, 3), (-10, 10))
        fns['gauss12'] = get_fns('TMath::Gaus', (1, 2), (-10, 10))
        fns['landau03'] = get_fns('TMath::Landau', (0, 3), (-10, 10))

        N = 10000
        rho = N/100.0
        comp_fns = []
        comp_fns.append((FloorThreshold(), rho - 5*sqrt(rho), 'pol0', 'pol0', OK, 'FT1'))    # 5 sigma
        comp_fns.append((FloorThreshold(), rho - sqrt(rho), 'pol0', 'pol0', ERROR, 'FT2'))   # 1 sigma
        comp_fns.append((CeilingThreshold(), rho + 5*sqrt(rho), 'pol0', 'pol0', OK, 'CT1'))  # 5 sigma
        comp_fns.append((CeilingThreshold(), rho + sqrt(rho), 'pol0', 'pol0', ERROR, 'CT2')) # 1 sigma
        comp_fns.append((MeanWidthDiffRef(), 0.1, 'gauss03', 'gauss03', OK, 'MWDR1'))
        comp_fns.append((MeanWidthDiffRef(), 0.1, 'gauss03', 'gauss12', ERROR, 'MWDR2'))
        comp_fns.append((ZeroCentredBandRef(), 5, 'gauss01', 'gauss01', OK, 'ZCBR1'))        # 5 sigma
        comp_fns.append((ZeroCentredBandRef(), 3, 'gauss01', 'gauss02', ERROR, 'ZCBR2'))     # 3 sigma
        # comp_fns.append( , , 'landau03', 'landau03', )

        comb_dict = {}
        eval_dict = {}
        self.results = []
        for i in comp_fns:
            name = '{}_{}_{}'.format(i[5], i[2], id(i)) # unique form: <name>_<fn>_<id>
            hname = 'hist_' + name
            cname = name + '_Combiner' # should be of the form: *Combiner
            comb_dict[cname] = create_leaf_dict_with_path(hname)
            eval_dict[cname] = {'Function': i[0], 'Argument': i[1]}

            href = TH1D(hname, '', 100, -10 ,10)
            href.FillRandom(fns[i[2]][0], N)
            self.rfref.WriteTObject(href)
            del href
            hdata = TH1D(hname, '', 100, -10 ,10)
            hdata.FillRandom(fns[i[3]][0], N)
            self.rfdata.WriteTObject(hdata)
            del hdata

            self.results.append((cname, i[4]))

        comb_dict = {
            'MasterCombiner': merge_dicts(
                {"weight": 1.0, "minWW": 10, "minWE": 25, "minEW": 1, "minEE": 2},
                comb_dict)
        }
        self.results.append(('MasterCombiner', ERROR))

        # close files cleanly
        self.rfdata.Close()
        self.rfref.Close()

        self.mycombiner = Combiner(comb_dict, eval_dict,
                                   self.rfdata.GetName(), self.rfref.GetName())
        self.mycombiner.evaluate()
Esempio n. 10
0
    def setUp(self):
        """Create dictionaries and ROOT files needed for testing."""

        self.tdir = tempfile.mkdtemp()
        self.rfdata = TFile(os.path.join(self.tdir, 'fdata.root'), 'recreate')
        self.rfref = TFile(os.path.join(self.tdir, 'fref.root'), 'recreate')

        fns = {}
        fns['pol0'] = get_simple_fns('5', (-10, 10))
        fns['gauss01'] = get_fns('TMath::Gaus', (0, 1), (-10, 10))
        fns['gauss02'] = get_fns('TMath::Gaus', (0, 2), (-10, 10))
        fns['gauss03'] = get_fns('TMath::Gaus', (0, 3), (-10, 10))
        fns['gauss12'] = get_fns('TMath::Gaus', (1, 2), (-10, 10))
        fns['landau03'] = get_fns('TMath::Landau', (0, 3), (-10, 10))

        N = 10000
        rho = N / 100.0
        comp_fns = []
        comp_fns.append((FloorThreshold(), rho - 5 * TMath.Sqrt(rho), 'pol0',
                         'pol0', OK, 'FT1'))  # 5 sigma
        comp_fns.append((FloorThreshold(), rho - TMath.Sqrt(rho), 'pol0',
                         'pol0', ERROR, 'FT2'))  # 1 sigma
        comp_fns.append((CeilingThreshold(), rho + 5 * TMath.Sqrt(rho), 'pol0',
                         'pol0', OK, 'CT1'))  # 5 sigma
        comp_fns.append((CeilingThreshold(), rho + TMath.Sqrt(rho), 'pol0',
                         'pol0', ERROR, 'CT2'))  # 1 sigma
        comp_fns.append(
            (MeanWidthDiffRef(), 0.1, 'gauss03', 'gauss03', OK, 'MWDR1'))
        comp_fns.append(
            (MeanWidthDiffRef(), 0.1, 'gauss03', 'gauss12', ERROR, 'MWDR2'))
        comp_fns.append((ZeroCentredBandRef(), 5, 'gauss01', 'gauss01', OK,
                         'ZCBR1'))  # 5 sigma
        comp_fns.append((ZeroCentredBandRef(), 3, 'gauss01', 'gauss02', ERROR,
                         'ZCBR2'))  # 3 sigma
        # comp_fns.append( , , 'landau03', 'landau03', )

        comb_dict = {}
        eval_dict = {}
        childNames = []
        self.results = []
        for i in comp_fns:
            name = '{}_{}_{}'.format(i[5], i[2],
                                     id(i))  # unique form: <name>_<fn>_<id>
            hname = 'hist_' + name
            cname = name + '_Combiner'  # should be of the form: *Combiner
            childNames.append(cname)
            comb_dict[cname] = {
                "weight": 1.0,
                "maxError": 50,
                "maxWarning": 80,
                "path": hname
            }
            eval_dict[cname] = {'Function': i[0], 'Argument': i[1]}

            href = TH1D(hname, '', 100, -10, 10)
            href.FillRandom(fns[i[2]][0], N)
            self.rfref.WriteTObject(href)
            del href
            hdata = TH1D(hname, '', 100, -10, 10)
            hdata.FillRandom(fns[i[3]][0], N)
            self.rfdata.WriteTObject(hdata)
            del hdata

            self.results.append((cname, i[4]))

        comb_dict['MasterCombiner'] = {
            "weight": 1.0,
            "minWW": 10,
            "minWE": 25,
            "minEW": 1,
            "minEE": 2,
            "children": childNames
        }
        self.results.append(('MasterCombiner', ERROR))

        # close files cleanly
        self.rfdata.Close()
        self.rfref.Close()

        self.mycombiner = RootCombiner(comb_dict, eval_dict,
                                       self.rfdata.GetName(),
                                       self.rfref.GetName(), 0)
        self.mycombiner.evaluate()
Esempio n. 11
0
 def setUp(self):
     self.hist = TH1D('hist', '', 100, -10, 10)
     fn = get_fns('TMath::Gaus', (0, 3), (-10, 10))
     self.hist.FillRandom(fn[0], 10000)
     self.delta = 0.05       # 5%