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

Format. Simplify code.

Browse Source
This commit is contained in:
Raphael Mitsch 2023-03-13 12:58:17 +01:00
parent 0753742a48
commit 5af8c711b0
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
spacy/tests/doc/test_span.py
View File

@ -706,12 +706,13 @@ def test_for_partial_ent_sents():
"""Spans may be associated with multiple sentences. These .sents should always be complete, not partial, sentences,
which this tests for.
"""
nlp = English()
text = ["Mahler's", "Symphony", "No.", "8", "was", "beautiful."]
doc = Doc(nlp.vocab, words=text, sent_starts=[1, 0, 0, 1, 0, 0])
doc = Doc(
English().vocab,
words=["Mahler's", "Symphony", "No.", "8", "was", "beautiful."],
sent_starts=[1, 0, 0, 1, 0, 0],
)
doc.set_ents([Span(doc, 1, 4, "WORK")])
# The specified entity is associated with both sentences in this doc, so we expect all sentences in the doc to be
# equal to the sentences referenced in ent.sents.
for doc_sent, ent_sent in zip(doc.sents, doc.ents[0].sents):
assert doc_sent == ent_sent

2
spacy/tokens/span.pyx
View File

@ -461,7 +461,7 @@ cdef class Span:
if start >= self.end:
break
elif i == self.doc.length - 1:
yield Span(self.doc, start, i + 1)
yield Span(self.doc, start, self.doc.length)
@property
def ents(self):