explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-19 12:42:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add simple mention test

Browse Source
This commit is contained in:
Paul O'Leary McCann 2021-07-18 19:15:32 +09:00
parent 9b63cbb775
commit a4531be099
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/tests/pipeline/test_coref.py
View File

@ -6,7 +6,7 @@ from spacy.training import Example
from spacy.lang.en import English
from spacy.tests.util import make_tempdir
from spacy.pipeline.coref import DEFAULT_CLUSTERS_PREFIX
from spacy.ml.models.coref_util import select_non_crossing_spans
from spacy.ml.models.coref_util import select_non_crossing_spans, get_candidate_mentions
# fmt: off
TRAIN_DATA = [
@ -155,3 +155,9 @@ def test_crossing_spans():
guess = select_non_crossing_spans(idxs, starts, ends, limit)
guess = sorted(guess)
assert gold == guess
def test_mention_generator(nlp):
doc = nlp("I like text.") # four tokens
max_width = 20
mentions = get_candidate_mentions(doc, max_width)
assert len(mentions[0]) == 10