def test_getattr_same_objects(): tree = MockTree() events = Events(tree) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder branch1 = MockBranch() branchBuilder.next = branch1 jet_pt1 = events.jet_pt assert branch1 is jet_pt1 branch2 = MockBranch() branchBuilder.next = branch2 jet_pt2 = events.jet_pt assert branch1 is jet_pt2 assert branch2 is not jet_pt2 it = iter(events) event = next(it) branch3 = MockBranch() branchBuilder.next = branch3 jet_pt3 = event.jet_pt assert branch1 is jet_pt3 assert branch3 is not jet_pt3
def test_getattr_exception(): tree = MockTree() events = Events(tree) events.buildBranch = MockBranchBuilder() with pytest.raises(AttributeError): events.no_such_branch
def test_getattr_same_objects(self): tree = MockTree() events = BEvents(tree) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder branch1 = MockBranch() branchBuilder.next = branch1 jet_pt1 = events.jet_pt self.assertIs(branch1, jet_pt1) branch2 = MockBranch() branchBuilder.next = branch2 jet_pt2 = events.jet_pt self.assertIs(branch1, jet_pt2) self.assertIsNot(branch2, jet_pt2) it = iter(events) event = next(it) branch3 = MockBranch() branchBuilder.next = branch3 jet_pt3 = event.jet_pt self.assertIs(branch1, jet_pt3) self.assertIsNot(branch3, jet_pt3)
def test_getattr_same_objects(): tree = MockTree() events = Events(tree) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder branch1 = MockBranch() branchBuilder.next = branch1 jet_pt1 = events.jet_pt assert branch1 is jet_pt1 branch2 = MockBranch() branchBuilder.next = branch2 jet_pt2 = events.jet_pt assert branch1 is jet_pt2 assert branch2 is not jet_pt2 it = iter(events) event = next(it) branch3 = MockBranch() branchBuilder.next = branch3 jet_pt3 = event.jet_pt assert branch1 is jet_pt3 assert branch3 is not jet_pt3
def test_getattr_exception(): tree = MockTree() events = Events(tree) events.buildBranch = MockBranchBuilder() with pytest.raises(AttributeError): events.no_such_branch
def test_getattr(): tree = MockTree() events = Events(tree) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder branchBuilder.next = MockBranch() jet_pt = events.jet_pt assert isinstance(jet_pt, MockBranch)
def test_getattr(self): tree = MockTree() events = BEvents(tree) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder branchBuilder.next = MockBranch() jet_pt = events.jet_pt self.assertIsInstance(jet_pt, MockBranch)
def test_getattr(): tree = MockTree() events = Events(tree) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder branchBuilder.next = MockBranch() jet_pt = events.jet_pt assert isinstance(jet_pt, MockBranch)
def test_getattr_getentry(): tree = MockTree() events = Events(tree, start=10) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder assert -1 == events.iEvent assert -1 == tree.ievent assert not tree.getEntry_called branchBuilder.next = MockBranch() jet_pt = events.jet_pt assert not tree.getEntry_called it = iter(events) event = next(it) tree.getEntry_called = False assert 0 == events.iEvent assert 10 == tree.ievent assert not tree.getEntry_called jet_pt = event.jet_pt assert not tree.getEntry_called tree.getEntry_called = False met_pt = event.met_pt assert tree.getEntry_called assert 0 == events.iEvent assert 10 == tree.ievent event = next(it) tree.getEntry_called = False assert 1 == events.iEvent assert 11 == tree.ievent assert not tree.getEntry_called jet_pt = event.jet_pt assert not tree.getEntry_called tree.getEntry_called = False met_pt = event.met_pt assert not tree.getEntry_called assert 1 == events.iEvent assert 11 == tree.ievent tree.getEntry_called = False njets = event.njets assert tree.getEntry_called assert 1 == events.iEvent assert 11 == tree.ievent
def test_getattr_getentry(self): tree = MockTree() events = BEvents(tree, start=10) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder self.assertEqual(-1, events.iEvent) self.assertEqual(-1, tree.ievent) self.assertFalse(tree.getEntry_called) branchBuilder.next = MockBranch() jet_pt = events.jet_pt self.assertFalse(tree.getEntry_called) it = iter(events) event = next(it) tree.getEntry_called = False self.assertEqual(0, events.iEvent) self.assertEqual(10, tree.ievent) self.assertFalse(tree.getEntry_called) jet_pt = event.jet_pt self.assertFalse(tree.getEntry_called) tree.getEntry_called = False met_pt = event.met_pt self.assertTrue(tree.getEntry_called) self.assertEqual(0, events.iEvent) self.assertEqual(10, tree.ievent) event = next(it) tree.getEntry_called = False self.assertEqual(1, events.iEvent) self.assertEqual(11, tree.ievent) self.assertFalse(tree.getEntry_called) jet_pt = event.jet_pt self.assertFalse(tree.getEntry_called) tree.getEntry_called = False met_pt = event.met_pt self.assertFalse(tree.getEntry_called) self.assertEqual(1, events.iEvent) self.assertEqual(11, tree.ievent) tree.getEntry_called = False njets = event.njets self.assertTrue(tree.getEntry_called) self.assertEqual(1, events.iEvent) self.assertEqual(11, tree.ievent)
def test_getattr_getentry(): tree = MockTree() events = Events(tree, start=10) branchBuilder = MockBranchBuilder() events.buildBranch = branchBuilder assert -1 == events.iEvent assert -1 == tree.ievent assert not tree.getEntry_called branchBuilder.next = MockBranch() jet_pt = events.jet_pt assert not tree.getEntry_called it = iter(events) event = next(it) tree.getEntry_called = False assert 0 == events.iEvent assert 10 == tree.ievent assert not tree.getEntry_called jet_pt = event.jet_pt assert not tree.getEntry_called tree.getEntry_called = False met_pt = event.met_pt assert tree.getEntry_called assert 0 == events.iEvent assert 10 == tree.ievent event = next(it) tree.getEntry_called = False assert 1 == events.iEvent assert 11 == tree.ievent assert not tree.getEntry_called jet_pt = event.jet_pt assert not tree.getEntry_called tree.getEntry_called = False met_pt = event.met_pt assert not tree.getEntry_called assert 1 == events.iEvent assert 11 == tree.ievent tree.getEntry_called = False njets = event.njets assert tree.getEntry_called assert 1 == events.iEvent assert 11 == tree.ievent
def test_init_branch_status(): tree = MockTree() assert [] == tree.branchstatus events = Events(tree) events.buildBranch = MockBranchBuilder() assert [('*', 0)] == tree.branchstatus
def test_init_branch_status(self): tree = MockTree() self.assertEqual([], tree.branchstatus) events = BEvents(tree) events.buildBranch = MockBranchBuilder() self.assertEqual([('*', 0)], tree.branchstatus)
def test_getattr_exception(self): tree = MockTree() events = BEvents(tree) events.buildBranch = MockBranchBuilder() self.assertRaises(AttributeError, events.__getattr__, 'no_such_branch')
def test_init_branch_status(): tree = MockTree() assert [ ] == tree.branchstatus events = Events(tree) events.buildBranch = MockBranchBuilder() assert [('*', 0)] == tree.branchstatus