mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-26 01:46:28 +03:00
* Add test for simple NER case
This commit is contained in:
parent
49c2e72112
commit
0a62c32c14
20
tests/test_ner.py
Normal file
20
tests/test_ner.py
Normal file
|
@ -0,0 +1,20 @@
|
|||
from spacy.en import English
|
||||
|
||||
|
||||
nlp = English()
|
||||
|
||||
def test_simple_types():
|
||||
tokens = nlp(u'Mr. Best flew to New York on Saturday morning.')
|
||||
ents = list(tokens.ents)
|
||||
assert ents[0].start == 1
|
||||
assert ents[0].end == 2
|
||||
assert ents[0].label_ == 'PERSON'
|
||||
assert ents[1].start == 4
|
||||
assert ents[1].end == 6
|
||||
assert ents[1].label_ == 'GPE'
|
||||
assert ents[2].start == 7
|
||||
assert ents[2].end == 8
|
||||
assert ents[2].label_ == 'DATE'
|
||||
assert ents[3].start == 8
|
||||
assert ents[3].end == 9
|
||||
assert ents[3].label_ == 'TIME'
|
Loading…
Reference in New Issue
Block a user