Python bindings to oxyroot. Makes reading .root files blazing fast 馃殌
at main 548 B view raw
1import oxyroot 2import uproot 3import numpy as np 4import os 5 6print(oxyroot.__version__) 7 8def test_read_from_uproot(): 9 # Create a dummy ROOT file for testing 10 11 input = np.array([4.1, 5.2, 6.3]) 12 file_name = "test.root" 13 14 with uproot.recreate(file_name) as f: 15 f.mktree("tree1", {"branch1": np.float64}) 16 f["tree1"].extend({"branch1": input}) 17 18 19 output = oxyroot.open(file_name)["tree1"]["branch1"].array() 20 assert(type(output) is np.ndarray) 21 assert(np.array_equal(input, output)) 22 23 os.remove(file_name)