mirror of https://github.com/explosion/spaCy.git
20 lines
636 B
Python
20 lines
636 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)
|
|
assert reloaded_morphology.get(morphology.strings["Feat1=Val1|Feat2=Val2"]) == "Feat1=Val1|Feat2=Val2"
|
|
assert reloaded_morphology.get(morphology.strings["Feat3=Val3|Feat4=Val4"]) == "Feat3=Val3|Feat4=Val4"
|