PhysicsTools/HeppyCore/python/utils/testtree.py - master - githubmirror/cmssw

PhysicsTools/HeppyCore/python/utils/testtree.py

15 строк483 B

Shahzad Malik Muzaffar

[ANALYSIS] py2/3 compatibility:drop use of __future__

22 ноя 2024, 20:16

22 ноя 2024, 20:168d027cc

from builtins import range
from ROOT import TFile
from PhysicsTools.HeppyCore.statistics.tree import Tree

def create_tree(filename="test_tree.root"): 
    outfile = TFile(filename, 'recreate')
    tree = Tree('test_tree', 'A test tree')
    tree.var('var1')
    for i in range(100):
        tree.fill('var1', i)
        tree.tree.Fill()
    print('creating a tree', tree.tree.GetName(),\
        tree.tree.GetEntries(), 'entries in',\
        outfile.GetName())
    outfile.Write()