diff options
-rw-r--r-- | stanza/tests/pipeline/test_depparse.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/stanza/tests/pipeline/test_depparse.py b/stanza/tests/pipeline/test_depparse.py index 7ad29aa2..4ad97451 100644 --- a/stanza/tests/pipeline/test_depparse.py +++ b/stanza/tests/pipeline/test_depparse.py @@ -61,10 +61,13 @@ EN_DOC_DEPENDENCY_PARSES_GOLD = """ ('.', 2, 'punct') """.strip() +@pytest.fixture(scope="module") +def en_depparse_pipeline(): + nlp = stanza.Pipeline(dir=TEST_MODELS_DIR, lang='en', processors='tokenize,pos,lemma,depparse') + return nlp -def test_depparse(): - nlp = stanza.Pipeline(dir=TEST_MODELS_DIR, lang='en') - doc = nlp(EN_DOC) +def test_depparse(en_depparse_pipeline): + doc = en_depparse_pipeline(EN_DOC) assert EN_DOC_DEPENDENCY_PARSES_GOLD == '\n\n'.join([sent.dependencies_string() for sent in doc.sentences]) |