mirror of https://github.com/explosion/spaCy.git
22 lines
668 B
Python
22 lines
668 B
Python
import pytest
|
|
import pickle
|
|
from spacy.morphology import Morphology
|
|
from spacy.strings import StringStore
|
|
|
|
|
|
@pytest.fixture
|
|
def morphology():
|
|
morphology = Morphology(StringStore())
|
|
morphology.add("Feat1=Val1|Feat2=Val2")
|
|
morphology.add("Feat3=Val3|Feat4=Val4")
|
|
return morphology
|
|
|
|
|
|
def test_morphology_pickle_roundtrip(morphology):
|
|
b = pickle.dumps(morphology)
|
|
reloaded_morphology = pickle.loads(b)
|
|
feat = reloaded_morphology.get(morphology.strings["Feat1=Val1|Feat2=Val2"])
|
|
assert feat == "Feat1=Val1|Feat2=Val2"
|
|
feat = reloaded_morphology.get(morphology.strings["Feat3=Val3|Feat4=Val4"])
|
|
assert feat == "Feat3=Val3|Feat4=Val4"
|