def test_simple_var_access(self):
run_number = NTupleVariable(
'run_number', vtype='uint',
extract_function=lambda event: event.getRun()
)
self.assertEqual(
run_number.extract(self.event1).get('run_number'),
self.event1.getRun()
)
self.assertEqual(
run_number.extract(self.event2).get('run_number'),
self.event2.getRun()
)
event_number = NTupleVariable(
'event_number', vtype='uint',
extract_function=lambda event: event.id()
)
self.assertEqual(
event_number.extract(self.event1).get('event_number'),
self.event1.id()
)
self.assertEqual(
event_number.extract(self.event2).get('event_number'),
self.event2.id()
)
def test_collection_var_access(self):
first_e_pt = NTupleVariable(
'first_e_pt',
vtype='float',
extract_function=lambda event: event.electrons[0].pt)
self.assertEqual(
first_e_pt.extract(self.event1).get('first_e_pt'), self.e1.pt)
def test_collection_var_access(self):
first_e_pt = NTupleVariable(
'first_e_pt', vtype='float',
extract_function=lambda event: event.electrons[0].pt)
self.assertEqual(
first_e_pt.extract(self.event1).get('first_e_pt'),
self.e1.pt
)
def test_simple_var_access(self):
run_number = NTupleVariable(
'run_number',
vtype='uint',
extract_function=lambda event: event.getRun())
self.assertEqual(
run_number.extract(self.event1).get('run_number'),
self.event1.getRun())
self.assertEqual(
run_number.extract(self.event2).get('run_number'),
self.event2.getRun())
event_number = NTupleVariable(
'event_number',
vtype='uint',
extract_function=lambda event: event.id())
self.assertEqual(
event_number.extract(self.event1).get('event_number'),
self.event1.id())
self.assertEqual(
event_number.extract(self.event2).get('event_number'),
self.event2.id())
