def test_preloaded_nanoevents():
columns = [
'nMuon', 'Muon_pt', 'Muon_eta', 'Muon_phi', 'Muon_mass', 'Muon_charge',
'nJet', 'Jet_eta'
]
p = NanoEventsProcessor(columns=columns)
rootdir = uproot.open(os.path.abspath('tests/samples/nano_dy.root'))
tree = rootdir['Events']
arrays = tree.arrays(columns, how=dict)
src = SimplePreloadedColumnSource(arrays,
rootdir.file.uuid,
tree.num_entries,
object_path='/Events')
print(arrays)
events = NanoEventsFactory.from_preloaded(src,
metadata={
'dataset': 'ZJets'
}).events()
hists = p.process(events)
print(hists)
assert (hists['cutflow']['ZJets_pt'] == 18)
assert (hists['cutflow']['ZJets_mass'] == 6)
with pytest.raises(AttributeError):
print(events.Muon.matched_jet)
def test_preloaded_nanoevents():
columns = [
"nMuon",
"Muon_pt",
"Muon_eta",
"Muon_phi",
"Muon_mass",
"Muon_charge",
"nJet",
"Jet_eta",
]
p = NanoEventsProcessor(columns=columns)
rootdir = uproot.open(os.path.abspath("tests/samples/nano_dy.root"))
tree = rootdir["Events"]
arrays = tree.arrays(columns, how=dict)
src = SimplePreloadedColumnSource(
arrays, rootdir.file.uuid, tree.num_entries, object_path="/Events"
)
print(arrays)
events = NanoEventsFactory.from_preloaded(
src, metadata={"dataset": "ZJets"}
).events()
hists = p.process(events)
print(hists)
assert hists["cutflow"]["ZJets_pt"] == 18
assert hists["cutflow"]["ZJets_mass"] == 6
with pytest.raises(AttributeError):
print(events.Muon.matched_jet)
def test_preloaded_nanoevents():
columns = [
'nMuon', 'Muon_pt', 'Muon_eta', 'Muon_phi', 'Muon_mass', 'Muon_charge',
'nJet', 'Jet_eta'
]
p = NanoEventsProcessor(columns=columns)
tree = uproot.open(os.path.abspath('tests/samples/nano_dy.root'))['Events']
arrays = tree.arrays(columns, flatten=True, namedecode='ascii')
df = processor.PreloadedDataFrame(tree.numentries, arrays)
print(arrays)
events = NanoEvents.from_arrays(arrays, metadata={'dataset': 'ZJets'})
hists = p.process(events)
print(hists)
assert (hists['cutflow']['ZJets_pt'] == 18)
assert (hists['cutflow']['ZJets_mass'] == 6)
with pytest.raises(RuntimeError):
print(events.Muon.matched_jet)
