diff --git a/spacy/tests/parser/test_beam_parse.py b/spacy/tests/parser/test_beam_parse.py index da5f43d5e..dd77c6805 100644 --- a/spacy/tests/parser/test_beam_parse.py +++ b/spacy/tests/parser/test_beam_parse.py @@ -1,10 +1,11 @@ -import spacy +# coding: utf8 +from __future__ import unicode_literals + import pytest -@pytest.mark.models -def test_beam_parse(): - nlp = spacy.load('en_core_web_sm') - doc = nlp(u'Australia is a country', disable=['ner']) - ents = nlp.entity(doc, beam_width=2) - print(ents) +@pytest.mark.models('en') +def test_beam_parse(EN): + doc = EN(u'Australia is a country', disable=['ner']) + ents = EN.entity(doc, beam_width=2) + print(ents)