def test_load_2(self): '''Test chaining of two files.''' tmpfile = testfname.replace('test_tree', 'test_tree_2_tmp') shutil.copyfile(testfname, tmpfile) chain = Chain(testfname.replace('.root', '*.root'), 'test_tree') self.assertEqual(len(chain), self.nevents * 2) os.remove(tmpfile)
def test_guess_treename(self): chain = Chain(testfname) self.assertEqual(len(self.chain), self.nevents)
def setUp(self): self.chain = Chain(testfname, 'test_tree')
def setUp(self): self.fname = create_tree() rootfile = TFile(self.fname) self.nevents = rootfile.Get('test_tree').GetEntries() self.chain = Chain(testfname, 'test_tree')
def test_load_3(self): '''Test LFN/root-fn loading''' chain = Chain(["root://{0}".format(os.path.abspath(testfname))], 'test_tree') self.assertEqual(len(chain), self.nevents)