chain_test.ChainTestCase Class Reference

Inheritance diagram for chain_test.ChainTestCase:

Public Member Functions
def	setUp (self)
def	test_file (self)
def	test_get (self)
def	test_guess_treename (self)
def	test_iterate (self)
def	test_load_1 (self)
def	test_load_2 (self)
def	test_load_3 (self)
def	test_wrong_filename (self)

Public Attributes
	chain

Detailed Description

Definition at line 10 of file chain_test.py.

Member Function Documentation

def chain_test.ChainTestCase.setUp

(

self

)

Definition at line 12 of file chain_test.py.

    def setUp(self):
        self.chain = Chain(testfname, 'test_tree')

def chain_test.ChainTestCase.test_file

(

self

)

Test that the test file exists

Definition at line 15 of file chain_test.py.

    def test_file(self):
        '''Test that the test file exists'''
        self.assertTrue(os.path.isfile(testfname))

def chain_test.ChainTestCase.test_get

(

self

)

Test direct event access

Definition at line 50 of file chain_test.py.

References chain_test.ChainTestCase.chain, edm::FileLocator::Rule.chain, chain.Chain.chain, and testtree.create_tree().

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

def chain_test.ChainTestCase.test_guess_treename

(

self

)

Definition at line 23 of file chain_test.py.

References chain_test.ChainTestCase.chain, edm::FileLocator::Rule.chain, and chain.Chain.chain.

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

def chain_test.ChainTestCase.test_iterate

(

self

)

Test iteration

Definition at line 44 of file chain_test.py.

References chain_test.ChainTestCase.chain, edm::FileLocator::Rule.chain, and chain.Chain.chain.

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

def chain_test.ChainTestCase.test_load_1

(

self

)

Test that the chain has the correct number of entries

Definition at line 27 of file chain_test.py.

References chain_test.ChainTestCase.chain, edm::FileLocator::Rule.chain, and chain.Chain.chain.

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

def chain_test.ChainTestCase.test_load_2

(

self

)

Test chaining of two files.

Definition at line 31 of file chain_test.py.

    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 chain_test.ChainTestCase.test_load_3

(

self

)

Test LFN/root-fn loading

Definition at line 39 of file chain_test.py.

    def test_load_3(self):
        '''Test LFN/root-fn loading'''
        chain = Chain(["root://{0}".format(os.path.abspath(testfname))], 'test_tree')
        self.assertEqual(len(chain), 100)

def chain_test.ChainTestCase.test_wrong_filename

(

self

)

Definition at line 19 of file chain_test.py.

    def test_wrong_filename(self):
        self.assertRaises(ValueError,
                          Chain, 'non_existing_file.root')

Member Data Documentation

chain_test.ChainTestCase.chain

Definition at line 13 of file chain_test.py.

Referenced by chain_test.ChainTestCase.test_get(), chain_test.ChainTestCase.test_guess_treename(), chain_test.ChainTestCase.test_iterate(), and chain_test.ChainTestCase.test_load_1().