From 28244df4dafa8b83ea617349c479925663a8d173 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sat, 22 Jul 2017 01:48:35 +0200 Subject: [PATCH] Add test for beam parsing --- spacy/tests/parser/test_beam_parse.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 spacy/tests/parser/test_beam_parse.py diff --git a/spacy/tests/parser/test_beam_parse.py b/spacy/tests/parser/test_beam_parse.py new file mode 100644 index 000000000..da5f43d5e --- /dev/null +++ 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) +