File indexing completed on 2024-11-25 02:29:50
0001 import unittest
0002 from ROOT import TFile
0003 from .tree import Tree
0004
0005 class TreeTestCase(unittest.TestCase):
0006
0007 def test_fill(self):
0008 fi = TFile('tree.root','RECREATE')
0009 tr = Tree('test_tree', 'A test tree')
0010 tr.var('a')
0011 tr.var('b')
0012 tr.fill('a', 3)
0013 tr.fill('a', 4)
0014 tr.fill('b', 5)
0015 tr.tree.Fill()
0016 fi.Write()
0017 fi.Close()
0018
0019 def test_read(self):
0020 fi = TFile('tree.root')
0021 tr = fi.Get('test_tree')
0022 self.assertEqual(tr.GetEntries(), 1)
0023 tr.GetEntry(0)
0024 self.assertEqual(tr.a, 4)
0025
0026 def test_iterate(self):
0027 fi = TFile('tree.root')
0028 tr = fi.Get('test_tree')
0029 for ev in tr:
0030 self.assertEqual(ev.a, 4)
0031 break
0032
0033
0034 if __name__ == '__main__':
0035 unittest.main()
