explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-10 19:57:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add entity recognizer to parser serialization tests

Browse Source
This commit is contained in:
ines 2017-06-02 18:40:06 +02:00
parent 0051c05964
commit d86e7cde93
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/tests/serialize/test_serialize_parser.py → spacy/tests/serialize/test_serialize_parser_ner.py
View File

@ -2,12 +2,17 @@
from __future__ import unicode_literals
from ..util import make_tempdir
from ...pipeline import NeuralDependencyParser as Parser
from ...pipeline import NeuralDependencyParser as DependencyParser
from ...pipeline import NeuralEntityRecognizer as EntityRecognizer
import pytest
def test_serialize_parser_roundtrip_bytes(en_vocab):
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()
@ -17,7 +22,8 @@ def test_serialize_parser_roundtrip_bytes(en_vocab):
assert new_parser.to_bytes() == parser_b
def test_serialize_parser_roundtrip_disk(en_vocab):
@pytest.mark.parametrize('Parser', test_parsers)
def test_serialize_parser_roundtrip_disk(en_vocab, Parser):
parser = Parser(en_vocab)
parser.model, _ = parser.Model(0)
with make_tempdir() as d: