Python BranchAddressManagerの例

コード例 #1

ファイル: test_BranchAddressManager.py プロジェクト: CMSRA1/AlphaTwirl

    def test_getArrays_no_branch(self):

        manager = BranchAddressManager()

        array_zet_pt, array_zet_pt_count = manager.getArrays(self.tree1, 'zet_pt')
        self.assertIsNone(array_zet_pt)
        self.assertIsNone(array_zet_pt_count)

コード例 #2

ファイル: test_BranchAddressManager.py プロジェクト: CMSRA1/AlphaTwirl

    def test_getArrays_same_objects(self):

        manager = BranchAddressManager()

        array_jet_pt_1, array_jet_pt_count_1 = manager.getArrays(self.tree1, 'jet_pt')
        array_jet_pt_2, array_jet_pt_count_2 = manager.getArrays(self.tree1, 'jet_pt')
        self.assertIs(array_jet_pt_1, array_jet_pt_2)
        self.assertIs(array_jet_pt_count_1, array_jet_pt_count_2)

コード例 #3

ファイル: test_BranchAddressManager.py プロジェクト: CMSRA1/AlphaTwirl

    def test_getArrays_no_count(self):

        manager = BranchAddressManager()

        array_met_pt, array_met_pt_count = manager.getArrays(self.tree1, 'met_pt')
        self.assertEqual('d', array_met_pt.typecode)
        self.assertEqual(1, array_met_pt.buffer_info()[1])
        self.assertIsNone(array_met_pt_count)

コード例 #4

ファイル: test_BranchAddressManager.py プロジェクト: eshwen/alphatwirl

    def test_getArrays_no_branch(self):

        manager = BranchAddressManager()

        array_zet_pt, array_zet_pt_count = manager.getArrays(
            self.tree1, 'zet_pt')
        self.assertIsNone(array_zet_pt)
        self.assertIsNone(array_zet_pt_count)

コード例 #5

ファイル: test_BranchAddressManager.py プロジェクト: CMSRA1/AlphaTwirl

    def test_getArrays_same_objects_different_managers(self):

        manager1 = BranchAddressManager()
        manager2 = BranchAddressManager()

        array_jet_pt_1, array_jet_pt_count_1 = manager1.getArrays(self.tree1, 'jet_pt')
        array_jet_pt_2, array_jet_pt_count_2 = manager2.getArrays(self.tree1, 'jet_pt')
        self.assertIs(array_jet_pt_1, array_jet_pt_2)
        self.assertIs(array_jet_pt_count_1, array_jet_pt_count_2)

コード例 #6

ファイル: test_BranchAddressManager.py プロジェクト: CMSRA1/AlphaTwirl

    def test_getArrays(self):

        manager = BranchAddressManager()

        array_jet_pt, array_jet_pt_count = manager.getArrays(self.tree1, 'jet_pt')
        self.assertEqual('d', array_jet_pt.typecode)
        self.assertEqual(13, array_jet_pt.buffer_info()[1])
        self.assertEqual('i', array_jet_pt_count.typecode)
        self.assertEqual(1, array_jet_pt_count.buffer_info()[1])

コード例 #7

ファイル: test_BranchAddressManager.py プロジェクト: eshwen/alphatwirl

    def test_getArrays_no_count(self):

        manager = BranchAddressManager()

        array_met_pt, array_met_pt_count = manager.getArrays(
            self.tree1, 'met_pt')
        self.assertEqual('d', array_met_pt.typecode)
        self.assertEqual(1, array_met_pt.buffer_info()[1])
        self.assertIsNone(array_met_pt_count)

コード例 #8

ファイル: test_BranchAddressManager.py プロジェクト: eshwen/alphatwirl

    def test_getArrays_same_objects(self):

        manager = BranchAddressManager()

        array_jet_pt_1, array_jet_pt_count_1 = manager.getArrays(
            self.tree1, 'jet_pt')
        array_jet_pt_2, array_jet_pt_count_2 = manager.getArrays(
            self.tree1, 'jet_pt')
        self.assertIs(array_jet_pt_1, array_jet_pt_2)
        self.assertIs(array_jet_pt_count_1, array_jet_pt_count_2)

コード例 #9

ファイル: test_BranchAddressManager.py プロジェクト: eshwen/alphatwirl

    def test_getArrays(self):

        manager = BranchAddressManager()

        array_jet_pt, array_jet_pt_count = manager.getArrays(
            self.tree1, 'jet_pt')
        self.assertEqual('d', array_jet_pt.typecode)
        self.assertEqual(13, array_jet_pt.buffer_info()[1])
        self.assertEqual('i', array_jet_pt_count.typecode)
        self.assertEqual(1, array_jet_pt_count.buffer_info()[1])

コード例 #10

ファイル: test_BranchAddressManager.py プロジェクト: eshwen/alphatwirl

    def test_getArrays_same_objects_different_managers(self):

        manager1 = BranchAddressManager()
        manager2 = BranchAddressManager()

        array_jet_pt_1, array_jet_pt_count_1 = manager1.getArrays(
            self.tree1, 'jet_pt')
        array_jet_pt_2, array_jet_pt_count_2 = manager2.getArrays(
            self.tree1, 'jet_pt')
        self.assertIs(array_jet_pt_1, array_jet_pt_2)
        self.assertIs(array_jet_pt_count_1, array_jet_pt_count_2)