def test_root2hdf5_chunked_selected(): try: import tables except ImportError: raise SkipTest from rootpy.root2hdf5 import root2hdf5 rfile = get_file('test_tree.root') hfilename = os.path.join(TEMPDIR, 'out.h5') root2hdf5(rfile, hfilename, entries=90, selection='i % 2 == 0') hfile = tables.openFile(hfilename) assert_equal(len(hfile.root.test), 500) hfile.close()
def test_root2hdf5(): try: import tables except ImportError: raise SkipTest from rootpy.root2hdf5 import root2hdf5 rfile = get_file('test_tree.root') hfilename = os.path.join(TEMPDIR, 'out.h5') root2hdf5(rfile, hfilename) hfile = tables.openFile(hfilename) assert_equal(len(hfile.root.test), 1000) hfile.close()
def test_root2hdf5_chunked(): try: import tables except ImportError: raise SkipTest from rootpy.root2hdf5 import root2hdf5, tables_open rfile = get_file("test_tree.root") hfilename = os.path.join(TEMPDIR, "out.h5") root2hdf5(rfile, hfilename, entries=10) hfile = tables_open(hfilename) assert_equal(len(hfile.root.test), 1000) hfile.close()
def main(): root2hdf5.root2hdf5( "/user/hermanse/analysis/rootfiles_test/run2126-00.root", "/user/hermanse/analysis/rootfiles_test/run2126-00.h5")
from __future__ import print_function import argparse import pandas as pd from rootpy.root2hdf5 import root2hdf5 parser = argparse.ArgumentParser(description='Convert ntuple from ROOT file to hdf5 store') parser.add_argument('root_file', type=argparse.FileType(), help='Path to the ROOT file') parser.add_argument('-i', '--input', help='Input tuple (default: %(default)s)', default='/ToolSvc.PatDebugTTTruthTool/DownstreamDebugTuple') parser.add_argument('-o', '--output', help='Output table in hdf5 (default: %(default)s)', default='DownstreamDebugTuple') args = parser.parse_args() input_filename = args.root_file.name output_filename = input_filename[:input_filename.rindex('.root')] + '.h5' print('Converting {} to {}'.format(input_filename, output_filename)) rpath = '' if '/' in args.input[1:]: rpath = args.input[:args.input.rindex('/')] root2hdf5(input_filename, output_filename, rpath) hdf = pd.HDFStore(output_filename) table = hdf.get(args.input) hdf.put(args.output, table, 'table') hdf.remove(args.input) hdf.close()
type=argparse.FileType(), help='Path to the ROOT file') parser.add_argument( '-i', '--input', help='Input tuple (default: %(default)s)', default='/ToolSvc.PatDebugTTTruthTool/DownstreamDebugTuple') parser.add_argument('-o', '--output', help='Output table in hdf5 (default: %(default)s)', default='DownstreamDebugTuple') args = parser.parse_args() input_filename = args.root_file.name output_filename = input_filename[:input_filename.rindex('.root')] + '.h5' print('Converting {} to {}'.format(input_filename, output_filename)) rpath = '' if '/' in args.input[1:]: rpath = args.input[:args.input.rindex('/')] root2hdf5(input_filename, output_filename, rpath) hdf = pd.HDFStore(output_filename) table = hdf.get(args.input) hdf.put(args.output, table, 'table') hdf.remove(args.input) hdf.close()