Python create_treeの例

コード例 #1

ファイル: test_multiprocessing.py プロジェクト: jlingema/heppy

 def setUp(self):
     self.fname = create_tree()
     self.fname2 = self.fname.replace(".root", "_2.root")
     shutil.copy(self.fname, self.fname2)
     rootfile = TFile(self.fname)
     self.nevents = rootfile.Get("test_tree").GetEntries()
     self.outdir = tempfile.mkdtemp()
     logging.disable(logging.CRITICAL)

コード例 #2

ファイル: test_multiprocessing.py プロジェクト: sisselbay/heppy

 def setUp(self):
     self.fname = create_tree()
     self.fname2 = self.fname.replace('.root', '_2.root')
     shutil.copy(self.fname, self.fname2)
     rootfile = TFile(self.fname)
     self.nevents = rootfile.Get('test_tree').GetEntries()
     self.outdir = tempfile.mkdtemp()
     logging.disable(logging.CRITICAL)

コード例 #3

ファイル: test_simple_example.py プロジェクト: sisselbay/heppy_VM

 def setUp(self):
     self.fname = create_tree()
     rootfile = TFile(self.fname)
     self.nevents = rootfile.Get('test_tree').GetEntries()
     self.outdir = tempfile.mkdtemp()
     logging.disable(logging.CRITICAL)

コード例 #4

ファイル: create_tree.py プロジェクト: jpazzini/heppy

from heppy.utils.testtree import create_tree
create_tree() 

コード例 #5

ファイル: test_chain.py プロジェクト: sisselbay/heppy_VM

 def setUp(self):
     self.fname = create_tree()
     rootfile = TFile(self.fname)
     self.nevents = rootfile.Get('test_tree').GetEntries()
     self.chain = Chain(testfname, 'test_tree')

コード例 #6

ファイル: create_tree.py プロジェクト: sisselbay/heppy_VM

from heppy.utils.testtree import create_tree

if __name__ == "__main__":

    import sys
    import pdb
    pdb.set_trace()
    if len(sys.argv) == 2:
        nentries = sys.argv[1]
        create_tree(nentries=nentries)
    else:
        create_tree()

コード例 #7

ファイル: chain_test.py プロジェクト: jpazzini/heppy

        chain = Chain(testfname)
        self.assertEqual(len(self.chain), 100)

    def test_load_1(self):
        '''Test that the chain has the correct number of entries'''
        self.assertEqual(len(self.chain), 100)

    def test_load_2(self):
        '''Test chaining of two files.'''
        tmpfile = testfname.replace('test_tree', 'test_tree_2_tmp')
        shutil.copyfile(testfname, tmpfile)
        chain = Chain(testfname.replace('.root', '*.root'), 'test_tree')
        self.assertEqual(len(chain), 200)
        os.remove(tmpfile)

    def test_iterate(self):
        '''Test iteration'''
        for ev in self.chain:
            pass
        self.assertTrue(True)

    def test_get(self):
        '''Test direct event access'''
        event = self.chain[2]
        self.assertEqual(event.var1, 2.)


if __name__ == '__main__':
    create_tree(testfname)
    unittest.main()

コード例 #8

ファイル: chain_test.py プロジェクト: alicerobson/heppy

        chain = Chain(testfname)
        self.assertEqual(len(self.chain), 100)        

    def test_load_1(self):
        '''Test that the chain has the correct number of entries'''
        self.assertEqual(len(self.chain), 100)

    def test_load_2(self):
        '''Test chaining of two files.'''
        tmpfile = testfname.replace('test_tree', 'test_tree_2_tmp')
        shutil.copyfile(testfname, tmpfile)
        chain = Chain(testfname.replace('.root', '*.root'), 'test_tree')
        self.assertEqual(len(chain), 200)
        os.remove(tmpfile)

    def test_iterate(self):
        '''Test iteration'''
        for ev in self.chain:
            pass
        self.assertTrue(True)

    def test_get(self):
        '''Test direct event access'''
        event = self.chain[2]
        self.assertEqual(event.var1, 2.)


if __name__ == '__main__':
    create_tree(testfname)
    unittest.main()

コード例 #9

ファイル: create_tree.py プロジェクト: SDHCAL/heppy

from heppy.utils.testtree import create_tree

if __name__ == "__main__":
    
    import sys 
    if len(sys.argv) == 2: 
        nentries = sys.argv[1]
        create_tree(nentries=nentries) 
    else: 
        create_tree()

コード例 #10

ファイル: test_chain.py プロジェクト: GaelTouquet/heppy

 def setUp(self):
     self.fname = create_tree()
     rootfile = TFile(self.fname)
     self.nevents = rootfile.Get('test_tree').GetEntries()
     self.chain = Chain(testfname, 'test_tree')

コード例 #11

ファイル: test_simple_example.py プロジェクト: jlingema/heppy

 def setUp(self):
     self.fname = create_tree()
     rootfile = TFile(self.fname)
     self.nevents = rootfile.Get('test_tree').GetEntries()
     self.outdir = tempfile.mkdtemp()
     logging.disable(logging.CRITICAL)

コード例 #12

ファイル: test_eventstfile.py プロジェクト: SDHCAL/heppy

 def setUp(self):
     self.fname = create_tree()
     rootfile = TFile(self.fname)
     self.events = Events(testfname, 'test_tree')

コード例 #13

 def setUp(self):
     self.fname = create_tree()
     rootfile = TFile(self.fname)
     self.events = Events(testfname, 'test_tree')

コード例 #14

ファイル: test_simple_example.py プロジェクト: GaelTouquet/heppy

 def setUp(self):
     self.fname = create_tree()
     rootfile = TFile(self.fname)
     self.nevents = rootfile.Get('test_tree').GetEntries()
     self.outdir = 'Out_test'