From f3a9a7226a0367c82f1d0f5483b106ccc08d6b32 Mon Sep 17 00:00:00 2001 From: John Bauer Date: Tue, 13 Sep 2022 19:52:32 -0700 Subject: Squeeze a little bit more - only use depparse in the depparse pipeline --- stanza/tests/pipeline/test_depparse.py | 9 ++++++--- 1 file 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]) -- cgit v1.2.3