diff --git a/tests/test_add_lemmas.py b/tests/test_add_lemmas.py index 8effbbb48..cf7cb1c7f 100644 --- a/tests/test_add_lemmas.py +++ b/tests/test_add_lemmas.py @@ -1,13 +1,14 @@ -from spacy.en import EN +from spacy.en import English import pytest +@pytest.fixture +def EN(): + return English(pos_tag=True, parse=False) @pytest.fixture -def tagged(): - EN.load() +def tagged(EN): string = u'Bananas in pyjamas are geese.' - tokens = EN.tokenize(string) - EN.set_pos(tokens) + tokens = EN(string, pos_tag=True) return tokens