spaCy/tests/parser/test_ner.py
Matthew Honnibal d83255db17 * Fix ner test
2015-06-07 18:57:42 +02:00

11 lines
313 B
Python

def test_simple_types(EN):
tokens = EN(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'