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 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)
