mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-11 20:28:20 +03:00
b84fd70cc3
Pickle exceptions in the MORPH_RULES format instead of the internal format after the recent `Morphology.__init__` changes.
23 lines
660 B
Python
23 lines
660 B
Python
import pytest
|
|
import pickle
|
|
from spacy.morphology import Morphology
|
|
from spacy.strings import StringStore
|
|
from spacy.lemmatizer import Lemmatizer
|
|
from spacy.lookups import Lookups
|
|
|
|
|
|
@pytest.fixture
|
|
def morphology():
|
|
tag_map = {"A": {"POS": "X"}, "B": {"POS": "NOUN"}}
|
|
exc = {"A": {"a": {"POS": "VERB"}}}
|
|
lemmatizer = Lemmatizer(Lookups())
|
|
return Morphology(StringStore(), tag_map, lemmatizer, exc=exc)
|
|
|
|
|
|
def test_morphology_pickle_roundtrip(morphology):
|
|
b = pickle.dumps(morphology)
|
|
reloaded_morphology = pickle.loads(b)
|
|
|
|
assert morphology.tag_map == reloaded_morphology.tag_map
|
|
assert morphology.exc == reloaded_morphology.exc
|