mirror of https://github.com/explosion/spaCy.git
35 lines
820 B
Python
35 lines
820 B
Python
import pytest
|
|
|
|
from ...pipeline import NeuralDependencyParser
|
|
from ...vocab import Vocab
|
|
|
|
|
|
@pytest.fixture
|
|
def vocab():
|
|
return Vocab()
|
|
|
|
|
|
@pytest.fixture
|
|
def parser(vocab):
|
|
parser = NeuralDependencyParser(vocab)
|
|
parser.add_label('nsubj')
|
|
parser.model, cfg = parser.Model(parser.moves.n_moves)
|
|
parser.cfg.update(cfg)
|
|
return parser
|
|
|
|
|
|
@pytest.fixture
|
|
def blank_parser(vocab):
|
|
parser = NeuralDependencyParser(vocab)
|
|
return parser
|
|
|
|
|
|
def test_to_from_bytes(parser, blank_parser):
|
|
assert parser.model is not True
|
|
assert blank_parser.model is True
|
|
assert blank_parser.moves.n_moves != parser.moves.n_moves
|
|
bytes_data = parser.to_bytes()
|
|
blank_parser.from_bytes(bytes_data)
|
|
assert blank_parser.model is not True
|
|
assert blank_parser.moves.n_moves == parser.moves.n_moves
|