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)