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

Update serializer tests

Browse Source
This commit is contained in:
ines 2017-06-02 21:42:26 +02:00
parent c862527474
commit d21459f87d
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
spacy/tests/serialize/test_serialize_parser_ner.py
View File

@ -14,12 +14,11 @@ test_parsers = [DependencyParser, EntityRecognizer]
@pytest.mark.parametrize('Parser', test_parsers)
def test_serialize_parser_roundtrip_bytes(en_vocab, Parser):
parser = Parser(en_vocab)
parser.model, _ = parser.Model(0)
parser_b = parser.to_bytes()
parser.model, _ = parser.Model(10)
new_parser = Parser(en_vocab)
new_parser.model, _ = new_parser.Model(0)
new_parser = new_parser.from_bytes(parser_b)
assert new_parser.to_bytes() == parser_b
new_parser.model, _ = new_parser.Model(10)
new_parser = new_parser.from_bytes(parser.to_bytes())
assert new_parser.to_bytes() == parser.to_bytes()
@pytest.mark.parametrize('Parser', test_parsers)
@ -32,4 +31,4 @@ def test_serialize_parser_roundtrip_disk(en_vocab, Parser):
parser_d = Parser(en_vocab)
parser_d.model, _ = parser_d.Model(0)
parser_d = parser_d.from_disk(file_path)
assert parser.to_bytes() == parser_d.to_bytes()
assert parser.to_bytes(model=False) == parser_d.to_bytes(model=False)