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