def test_csv(): f = testdata.get_file('test_csv.root') tree = f.ParTree_Postselect tree.create_buffer(ignore_unsupported=True) output = StringIO() tree.csv(stream=output) f.close() # compare with existing txt output if sys.version_info[0] < 3: true_output_filename = testdata.get_filepath('test_csv.txt') else: true_output_filename = testdata.get_filepath('test_csv_new.txt') with open(true_output_filename, 'r') as true_output_file: true_output = true_output_file.read() assert_equal(output.getvalue(), true_output)
def test_csv(): from cStringIO import StringIO f = testdata.get_file("test_csv.root") tree = f.ParTree_Postselect tree.create_buffer(ignore_unsupported=True) output = StringIO() tree.csv(stream=output) f.close() # compare with existing txt output true_output_filename = testdata.get_filepath("test_csv.txt") with open(true_output_filename, "r") as true_output_file: true_output = true_output_file.read() assert_equal(output.getvalue(), true_output)
ROOT.TFile made easy by rootpy ============================== This example demonstrates how basic file operations are made easier in rootpy. """ print __doc__ import os import shutil import rootpy rootpy.log.basic_config_colorized() from rootpy.io import root_open, DoesNotExist from rootpy.plotting import Hist, Hist2D from rootpy import testdata from rootpy import asrootpy shutil.copyfile(testdata.get_filepath('test_file_2.root'), 'data.root') f = root_open('data.root') print f.a print f.a.b try: print f.a.b.c.d.e.f except AttributeError, e: print e for thing in f.walk(): print thing f.close()
""" ============================== ROOT.TFile made easy by rootpy ============================== This example demonstrates how basic file operations are made easier in rootpy. """ print(__doc__) import os import shutil from rootpy.io import root_open, DoesNotExist from rootpy.plotting import Hist, Hist2D from rootpy import testdata from rootpy import asrootpy shutil.copyfile(testdata.get_filepath('test_file_2.root'), 'data.root') f = root_open('data.root') print(f.a) print(f.a.b) try: print(f.a.b.c.d.e.f) except AttributeError as e: print(e) for thing in f.walk(): print(thing) f.close()