def writeTFile(self): import array import ROOT as r var = skimOptions.var() f = r.TFile(skimOptions.fileName(), "RECREATE") dirName, treeName = configuration.mainTree() pairs = [configuration.mainTree()] + configuration.otherTreesToKeepWhenSkimming() for i, (dirName, treeName) in enumerate(pairs): f.mkdir(dirName) f.cd(dirName) t = r.TTree(treeName, "title") a = array.array('i', [0]) t.Branch(var, a, '%s/I' % var) for run in range(i, i + 100): a[0] = run t.Fill() f.Write() f.Close()
def tearDown(self): import os os.remove(skimOptions.fileName())