Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Test Issue #514: Serializer fails when new entity type has been added.

This commit is contained in:
Matthew Honnibal 2016-10-23 17:41:32 +02:00
parent f97548c6f1
commit 79aa03fe98
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -108,10 +108,20 @@ def test_serialize_after_adding_entity():
entity_recognizer.add_label('Food') entity_recognizer.add_label('Food')
entity_recognizer(doc) entity_recognizer(doc)
label_id = vocab.strings[u'Food'] label_id = vocab.strings[u'Food']
doc.ents = [(label_id, 5,6)] doc.ents = [(label_id, 5,6)]
assert [(ent.label_, ent.text) for ent in doc.ents] == [(u'Food', u'pasta')] assert [(ent.label_, ent.text) for ent in doc.ents] == [(u'Food', u'pasta')]
byte_string = doc.to_bytes() byte_string = doc.to_bytes()
@pytest.mark.models
def test_serialize_after_adding_entity(EN):
EN.entity.add_label(u'Food')
doc = EN(u'This is a sentence about pasta.')
label_id = EN.vocab.strings[u'Food']
doc.ents = [(label_id, 5,6)]
byte_string = doc.to_bytes()
doc2 = Doc(EN.vocab).from_bytes(byte_string)
assert [(ent.label_, ent.text) for ent in doc2.ents] == [(u'Food', u'pasta')]