spaCy/spacy/tests/regression/test_issue1305.py

12 lines
348 B
Python
Raw Normal View History

import pytest
2017-10-24 15:05:15 +00:00
import spacy
2017-10-24 15:05:15 +00:00
#@pytest.mark.models('en')
def test_issue1305():
'''Test lemmatization of English VBZ'''
2017-10-24 15:05:15 +00:00
nlp = spacy.load('en_core_web_sm')
assert nlp.vocab.morphology.lemmatizer('works', 'verb') == ['work']
doc = nlp(u'This app works well')
print([(w.text, w.tag_) for w in doc])
assert doc[2].lemma_ == 'work'