def test_BeliefTriplet():
nlp = load_danish(transformer=None)
doc = nlp("Dette består af to sætninger")
span = next(doc.sents)
path = (0, 1, 2, 3)
bt = BeliefTriplet(
head_id=path[0],
relation_ids=path[1:-1],
tail_id=path[-1],
span=span,
confidence=1,
)
assert len(bt.relation_list) == 2
assert isinstance(bt.confidence, float)
assert isinstance(bt.head_span, Span)
assert isinstance(bt.head, str)
path = (3, 1, 2, 0)
rev_bt = BeliefTriplet(
head_id=path[0],
relation_ids=path[1:-1],
tail_id=path[-1],
span=span,
confidence=1,
)
assert bt < rev_bt, "belief triplets should be sorted according to their heads"
def simple_triplets() -> List[BeliefTriplet]:
nlp = load_danish(transformer=None)
doc = nlp("Dette består af to sætninger")
span = next(doc.sents)
path = (0, 1, 2, 3)
bt = BeliefTriplet(
head_id=path[0],
relation_ids=path[1:-1],
tail_id=path[-1],
span=span,
confidence=1,
)
doc = nlp("Mit navn er Kenneth. .")
span = next(doc.sents)
path = (0, 1, 2)
bt_ = BeliefTriplet(
head_id=path[0],
relation_ids=path[1:-1],
tail_id=path[-1],
span=span,
confidence=1,
)
return [bt, bt_]
def create_graph_with_filters(triplet_filters, group_filters):
nlp = bg.load_danish()
bp = bg.BeliefParser(nlp=nlp)
graph = bg.BeliefGraph(parser=bp, triplet_filters=[], group_filters=[])
graph.add_texts(EXAMPLES)
graph.replace_filters(triplet_filters=triplet_filters,
group_filters=group_filters)
return graph
def test_belief_parser():
nlp = bg.load_danish()
belief_parser = bg.BeliefParser(nlp=nlp)
triplets = belief_parser.parse_texts(EXAMPLE)
assert isinstance(triplets, Iterable)
triplets = list(triplets)
assert isinstance(triplets[0], bg.BeliefTriplet)
for t in triplets:
print(t)
def simple_graph():
nlp = bg.load_danish()
bp = bg.BeliefParser(nlp=nlp)
graph = bg.BeliefGraph(parser=bp, triplet_filters=[], group_filters=[])
return graph