def setUp(self): self.rfile = ROOT.TFile.Open('/tmp/testtree.root') self.tree = self.rfile.Get('testtree') self.nentries = self.tree.GetEntries() # total no of elements in data self.ndata = nplotted(self.tree, 'data') self.selector = Tselect(self.tree)
class test_Tselect(unittest.TestCase): def setUp(self): self.rfile = ROOT.TFile.Open('/tmp/testtree.root') self.tree = self.rfile.Get('testtree') self.nentries = self.tree.GetEntries() # total no of elements in data self.ndata = nplotted(self.tree, 'data') self.selector = Tselect(self.tree) def test_exprs(self): expr = ('foo>>hist', 'sz>4') self.selector.exprs = expr self.assertNotEqual(np.shape(expr), self.selector.shape) def test_fill_hist(self): self.selector.exprs = [('foo>>hist1', 'sz>4'), ('data>>hist2', 'sz>4')] hs = self.selector.fill_hists() self.assertAlmostEqual(hs[0].GetEntries(), self.nentries / 3., delta=1) self.assertAlmostEqual(hs[1].GetEntries(), 5 * self.nentries / 3., delta=2)
class test_Tselect(unittest.TestCase): def setUp(self): self.rfile = ROOT.TFile.Open('/tmp/testtree.root') self.tree = self.rfile.Get('testtree') self.nentries = self.tree.GetEntries() # total no of elements in data self.ndata = nplotted(self.tree, 'data') self.selector = Tselect(self.tree) def test_exprs(self): expr = ('foo>>hist', 'sz>4') self.selector.exprs = expr self.assertNotEqual(np.shape(expr), self.selector.shape) def test_fill_hist(self): self.selector.exprs = [ ('foo>>hist1', 'sz>4'), ('data>>hist2', 'sz>4') ] hs = self.selector.fill_hists() self.assertAlmostEqual(hs[0].GetEntries(), self.nentries/3., delta=1) self.assertAlmostEqual(hs[1].GetEntries(), 5*self.nentries/3., delta=2)