From c1ab1341592d472ec6d454d3687c00139f921212 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 21 Dec 2014 20:58:21 +1100 Subject: [PATCH] * Upd lemmas test --- tests/test_add_lemmas.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) 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