From f0d3672e170f19a2debe9282953696262681f47a Mon Sep 17 00:00:00 2001 From: Aaron Marquez Date: Thu, 15 Feb 2018 14:28:38 -0800 Subject: [PATCH] Changed loading EN model --- spacy/tests/regression/test_issue1959.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/spacy/tests/regression/test_issue1959.py b/spacy/tests/regression/test_issue1959.py index 2605b0088..0787af3b7 100644 --- a/spacy/tests/regression/test_issue1959.py +++ b/spacy/tests/regression/test_issue1959.py @@ -2,16 +2,14 @@ from __future__ import unicode_literals import pytest -from ..util import load_test_model - @pytest.mark.models('en') -def test_issue1959(): +def test_issue1959(EN): texts = ['Apple is looking at buying U.K. startup for $1 billion.'] - nlp = load_test_model('en_core_web_sm') - nlp.add_pipe(clean_component, name='cleaner', after='ner') - doc = nlp(texts[0]) - doc_pipe = [doc_pipe for doc_pipe in nlp.pipe(texts)] + # nlp = load_test_model('en_core_web_sm') + EN.add_pipe(clean_component, name='cleaner', after='ner') + doc = EN(texts[0]) + doc_pipe = [doc_pipe for doc_pipe in EN.pipe(texts)] assert doc == doc_pipe[0]