File indexing completed on 2024-11-25 02:29:50
0001 import unittest
0002
0003 from .eventstfile import Events
0004 from PhysicsTools.HeppyCore.utils.testtree import create_tree
0005
0006 testfname = 'test_tree.root'
0007
0008 class EventsTFileTestCase(unittest.TestCase):
0009
0010 def test(self):
0011 events = Events(testfname, 'test_tree')
0012 event = events.to(2)
0013 for iev, ev in enumerate(events):
0014 self.assertEqual(iev, ev.var1)
0015
0016 if __name__ == '__main__':
0017 create_tree(testfname)
0018 unittest.main()