Add test for beam parsing

2017-07-22 01:48:35 +02:00 · 2017-07-22 01:48:35 +02:00 · 28244df4da
parent c86445bdfd
commit 28244df4da
1 changed files with 10 additions and 0 deletions
--- a/spacy/tests/parser/test_beam_parse.py
+++ b/spacy/tests/parser/test_beam_parse.py
@ -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)
+