diff --git a/spacy/tests/tagger/test_add_lemmas.py b/spacy/tests/tagger/test_add_lemmas.py deleted file mode 100644 index 7fe9db86a..000000000 --- a/spacy/tests/tagger/test_add_lemmas.py +++ /dev/null @@ -1,12 +0,0 @@ -from __future__ import unicode_literals - -from spacy.en import English -import pytest - -@pytest.mark.models -def test_lemma_assignment(EN): - tokens = u'Bananas in pyjamas are geese .'.split(' ') - doc = EN.tokenizer.tokens_from_list(tokens) - assert all( t.lemma_ == u'' for t in doc ) - EN.tagger(doc) - assert all( t.lemma_ != u'' for t in doc ) diff --git a/spacy/tests/tagger/test_lemmatizer.py b/spacy/tests/tagger/test_lemmatizer.py index 31acc72e3..3bbd99eaa 100644 --- a/spacy/tests/tagger/test_lemmatizer.py +++ b/spacy/tests/tagger/test_lemmatizer.py @@ -91,3 +91,12 @@ def test_pickle_lemmatizer(lemmatizer): file_.seek(0) loaded = pickle.load(file_) + + +@pytest.mark.models +def test_lemma_assignment(EN): + tokens = u'Bananas in pyjamas are geese .'.split(' ') + doc = EN.tokenizer.tokens_from_list(tokens) + assert all( t.lemma_ == u'' for t in doc ) + EN.tagger(doc) + assert all( t.lemma_ != u'' for t in doc )