spaCy/spacy/tests/test_pickle.py
2015-11-03 04:57:02 +01:00

33 lines
639 B
Python

import cloudpickle
import io
import os
import pickle
import pytest
import tempfile
try:
unicode
except NameError:
unicode = str
@pytest.mark.models
def test_pickle_english(EN):
file_ = io.BytesIO()
cloudpickle.dump(EN, file_)
file_.seek(0)
loaded = pickle.load(file_)
assert loaded is not None
@pytest.mark.models
def test_cloudpickle_to_file(EN):
f = tempfile.NamedTemporaryFile(delete=False)
p = cloudpickle.CloudPickler(f)
p.dump(EN)
f.close()
loaded_en = cloudpickle.load(open(f.name, 'rb'))
os.unlink(f.name)
doc = loaded_en(unicode('test parse'))
assert len(doc) == 2