Update lemmatizer test

This commit is contained in:
Matthew Honnibal 2017-10-12 22:50:52 +02:00
parent 9b90d235d1
commit cf6da9301a

View File

@ -57,6 +57,5 @@ def test_en_lemmatizer_punct(en_lemmatizer):
def test_en_lemmatizer_lemma_assignment(EN): def test_en_lemmatizer_lemma_assignment(EN):
text = "Bananas in pyjamas are geese." text = "Bananas in pyjamas are geese."
doc = EN.make_doc(text) doc = EN.make_doc(text)
assert all(t.lemma_ == '' for t in doc)
EN.tagger(doc) EN.tagger(doc)
assert all(t.lemma_ != '' for t in doc) assert all(t.lemma_ != '' for t in doc)