2015-07-22 23:19:11 +00:00
|
|
|
import pytest
|
2015-04-13 19:33:54 +00:00
|
|
|
|
2015-07-22 23:19:11 +00:00
|
|
|
@pytest.mark.models
|
2015-06-07 16:02:24 +00:00
|
|
|
def test_simple_types(EN):
|
|
|
|
|
tokens = EN(u'Mr. Best flew to New York on Saturday morning.')
|
2015-04-13 19:33:54 +00:00
|
|
|
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'
|
