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

Fix coref tests

Browse Source
This commit is contained in:
Paul O'Leary McCann 2022-05-25 18:32:47 +09:00
parent 303269c4b2
commit 6999436270
1 changed files with 2 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8,8 +8,7 @@ from spacy.tests.util import make_tempdir
from spacy.pipeline.coref import DEFAULT_CLUSTERS_PREFIX from spacy.pipeline.coref import DEFAULT_CLUSTERS_PREFIX
from spacy.ml.models.coref_util import ( from spacy.ml.models.coref_util import (
select_non_crossing_spans, select_non_crossing_spans,
get_candidate_mentions, get_sentence_ids,
get_sentence_map,
) )
# fmt: off # fmt: off
@ -159,22 +158,7 @@ def test_crossing_spans():
guess = sorted(guess) guess = sorted(guess)
assert gold == guess assert gold == guess
def test_mention_generator(snlp):
nlp = snlp
doc = nlp("I like text.") # four tokens
max_width = 20
mentions = get_candidate_mentions(doc, max_width)
assert len(mentions[0]) == 10
# check multiple sentences
doc = nlp("I like text. This is text.") # eight tokens, two sents
max_width = 20
mentions = get_candidate_mentions(doc, max_width)
assert len(mentions[0]) == 20
def test_sentence_map(snlp): def test_sentence_map(snlp):
doc = snlp("I like text. This is text.") doc = snlp("I like text. This is text.")
sm = get_sentence_map(doc) sm = get_sentence_ids(doc)
assert sm == [0, 0, 0, 0, 1, 1, 1, 1] assert sm == [0, 0, 0, 0, 1, 1, 1, 1]