explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-27 09:44:36 +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 __future__ import unicode_literals
from ..util import make_tempdir 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 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 = Parser(en_vocab)
parser.model, _ = parser.Model(0) parser.model, _ = parser.Model(0)
parser_b = parser.to_bytes() parser_b = parser.to_bytes()
@ -17,7 +22,8 @@ def test_serialize_parser_roundtrip_bytes(en_vocab):
assert new_parser.to_bytes() == parser_b 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 = Parser(en_vocab)
parser.model, _ = parser.Model(0) parser.model, _ = parser.Model(0)
with make_tempdir() as d: with make_tempdir() as d: