def test_call_ctypes_same_objects_different_builders(mockTree):
obj1 = BranchBuilder()
obj2 = BranchBuilder()
obj1.register_tree(mockTree)
obj2.register_tree(mockTree)
result1 = obj1(mockTree, 'jet_pt')
result2 = obj2(mockTree, 'jet_pt')
assert result1 is result2
def test_call_unknown_type(mockTree, caplog):
obj = BranchBuilder()
obj.register_tree(mockTree)
with caplog.at_level(logging.WARNING):
result = obj(mockTree, 'EventAuxiliary')
assert result is None
assert len(caplog.records) == 1
assert caplog.records[0].levelname == 'WARNING'
assert 'BranchBuilder' in caplog.records[0].name
assert 'unknown leaf type' in caplog.records[0].msg
def test_empty_chain(caplog):
tree_name = 'tree'
chain = ROOT.TChain(tree_name)
# add no files to the chain
obj = BranchBuilder()
obj.register_tree(chain)
with caplog.at_level(logging.WARNING):
assert obj(chain, 'var') is None
assert len(caplog.records) == 2
assert caplog.records[0].levelname == 'WARNING'
assert 'BranchBuilder' in caplog.records[0].name
assert 'cannot get' in caplog.records[0].msg
def test_register_tree_two_builders(mockTree):
obj1 = BranchBuilder()
obj2 = BranchBuilder()
obj1.register_tree(mockTree)
obj2.register_tree(mockTree)
assert 1 == mockTree.SetBranchStatus.call_count
def test_register_tree(mockTree):
obj = BranchBuilder()
obj.register_tree(mockTree)
assert [mock.call('*', 0)] == mockTree.SetBranchStatus.call_args_list
assert mockTree in obj.itsdict
def test_call_no_such_branch(mockTree):
obj = BranchBuilder()
obj.register_tree(mockTree)
result = obj(mockTree, 'no_such_branch')
assert result is None
def test_call_stdvector(mockTree):
obj = BranchBuilder()
obj.register_tree(mockTree)
result = obj(mockTree, 'trigger_path')
assert isinstance(result, ROOT.vector('string'))
def test_call_ctypes_same_objects_different_calls(mockTree):
obj = BranchBuilder()
obj.register_tree(mockTree)
result1 = obj(mockTree, 'jet_pt')
result2 = obj(mockTree, 'jet_pt')
assert result1 is result2
def test_call_ctypes(mockTree):
obj = BranchBuilder()
obj.register_tree(mockTree)
result = obj(mockTree, 'jet_pt')
assert isinstance(result, Branch)
def test_register_tree_two_builders(mockTree):
obj1 = BranchBuilder()
obj2 = BranchBuilder()
obj1.register_tree(mockTree)
obj2.register_tree(mockTree)
assert 1 == mockTree.SetBranchStatus.call_count
def test_register_tree(mockTree):
obj = BranchBuilder()
obj.register_tree(mockTree)
assert [mock.call('*', 0)] == mockTree.SetBranchStatus.call_args_list
assert mockTree in obj.itsdict