explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-26 17:24:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

Test Issue #514: Serialization fails after adding a new entity label.

Browse Source
This commit is contained in:
Matthew Honnibal 2016-10-23 16:40:27 +02:00
parent 936e6246aa
commit 4de30a8e38
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
spacy/tests/serialize/test_serialization.py
View File

@ -97,3 +97,21 @@ def test_serialize_empty_doc():
assert b == b''
loaded = Doc(vocab).from_bytes(b)
assert len(loaded) == 0
def test_serialize_after_adding_entity():
# Re issue #514
vocab = spacy.en.English.Defaults.create_vocab()
entity_recognizer = spacy.en.English.Defaults.create_entity()
doc = Doc(vocab, words=u'This is a sentence about pasta .'.split())
entity_recognizer.add_label('Food')
entity_recognizer(doc)
label_id = vocab.strings[u'Food']
doc.ents = [(label_id, 5,6)]
assert [(ent.label_, ent.text) for ent in doc.ents] == [(u'Food', u'pasta')]
byte_string = doc.to_bytes()