Add test for beam parsing

This commit is contained in:
Matthew Honnibal 2017-07-22 01:48:35 +02:00
parent c86445bdfd
commit 28244df4da

View File

@ -0,0 +1,10 @@
import spacy
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)