def test_ROOT(): try: import ROOT except ImportError: raise SkipTest filename = os.path.join(DIRNAME, 'test.root') root_file = ROOT.TFile.Open(filename, 'recreate') root_file.Close() grl = GRL(GRLA) grl.save(filename + ':/lumi') grl2 = GRL(filename + ':/lumi') assert_equal(grl, grl2) root_file = ROOT.TFile.Open(filename, 'recreate') root_file.mkdir('dir') root_file.Close() grl = GRL(GRLA) grl.save(filename + ':/dir/lumi') grl2 = GRL(filename + ':/dir/lumi') assert_equal(grl, grl2) root_file = ROOT.TFile.Open(filename) tobj = root_file.Get('dir/lumi') assert_true(isinstance(tobj, ROOT.TObjString)) root_file.Close() os.unlink(filename)
def test_ROOT(): try: import ROOT except ImportError: raise SkipTest filename = os.path.join(DIRNAME, 'test.root') root_file = ROOT.TFile.Open(filename, 'recreate') root_file.Close() grl = GRL(GRLA) grl.save(filename + ':/lumi') grl2 = GRL(filename + ':/lumi') assert_equal(grl, grl2) root_file = ROOT.TFile.Open(filename, 'recreate') root_file.mkdir('dir') root_file.Close() grl = GRL(GRLA) grl.save(filename + ':/dir/lumi') grl2 = GRL(filename + ':/dir/lumi') assert_equal(grl, grl2) root_file = ROOT.TFile.Open(filename) tobj = root_file.Get('dir/lumi') assert_true(isinstance(tobj, ROOT.TObjString)) root_file.Close() os.unlink(filename)
def work(self): # merge GRL XML strings grl = GRL() for fname in self.files: with root_open(fname) as f: for key in f.Lumi.keys(): grl |= GRL(str(key.ReadObj().GetString()), from_string=True) grl.save('grl.xml')
def work(self): # merge GRL XML strings grl = GRL() for fname in self.files: with root_open(fname) as f: for key in f.Lumi.keys(): grl |= GRL(str(key.ReadObj().GetString()), from_string=True) grl.save('grl.xml')
def test_read_yaml(): try: import yaml except ImportError: raise SkipTest grl = GRL(GRLA) grl.save('test.yml') grl2 = GRL('test.yml') assert_equal(grl, grl2) os.unlink('test.yml')
def merge(inputs, output, metadata): # merge output trees root_output = output + '.root' subprocess.call(['hadd', root_output] + inputs) if metadata.datatype == datasets.DATA: # merge GRLs grl = GRL() for input in inputs: grl |= GRL('%s:/lumi' % input) grl.save('%s:/lumi' % root_output)
def test_read_yaml(): try: import yaml except ImportError: raise SkipTest grl = GRL(GRLA) grl.save('test.yml') grl2 = GRL('test.yml') assert_equal(grl, grl2) os.unlink('test.yml')
def save_test(): grl = GRL(GRLA) grl.save('testA.xml') os.unlink('testA.xml') assert_raises(ValueError, grl.save, 'testB.badext')
def save_test(): grl = GRL(GRLA) grl.save('testA.xml') os.unlink('testA.xml') assert_raises(ValueError, grl.save, 'testB.badext')