mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-24 00:46:28 +03:00
* Add tests for Span stuff
This commit is contained in:
parent
c2b9a61ee2
commit
3b5ea3731a
23
tests/test_span.py
Normal file
23
tests/test_span.py
Normal file
|
@ -0,0 +1,23 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
from spacy.en import English
|
||||
from spacy.spans import Spans
|
||||
|
||||
import pytest
|
||||
import re
|
||||
|
||||
|
||||
EN = English()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def doc():
|
||||
return EN('This is a sentence. This is another sentence. And a third.')
|
||||
|
||||
|
||||
def test_sent_spans(doc):
|
||||
sents = list(doc.sents)
|
||||
assert sents[0].start == 0
|
||||
assert sents[0].end == 5
|
||||
assert len(sents) == 3
|
||||
assert sum(len(sent) for sent in sents) == len(doc)
|
23
tests/test_times.py
Normal file
23
tests/test_times.py
Normal file
|
@ -0,0 +1,23 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
from spacy.en import English
|
||||
|
||||
import pytest
|
||||
|
||||
NLU = English()
|
||||
|
||||
def test_am_pm():
|
||||
numbers = ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12']
|
||||
variants = ['a.m.', 'am', 'p.m.', 'pm']
|
||||
spaces = ['', ' ']
|
||||
for num in numbers:
|
||||
for var in variants:
|
||||
for space in spaces:
|
||||
string = u"The meeting was at %s%s%s wasn't it?" % (num, space, var)
|
||||
tokens = NLU(string)
|
||||
assert tokens[4].orth_ == '%s%s%s' % (num, space, var)
|
||||
ents = list(tokens.ents)
|
||||
assert len(ents) == 1
|
||||
assert ents[0].label_ == 'TIME', string
|
||||
if ents[0].start == 4 and ents[0].end == 5:
|
||||
assert ents[0].orth_ == '%s%s%s' % (num, space, var)
|
Loading…
Reference in New Issue
Block a user