mirror of https://github.com/explosion/spaCy.git
36 lines
745 B
Python
36 lines
745 B
Python
import cloudpickle
|
|
import io
|
|
import os
|
|
import pickle
|
|
import pytest
|
|
import tempfile
|
|
|
|
try:
|
|
unicode
|
|
except NameError:
|
|
unicode = str
|
|
|
|
# These cause the addition of temp files, that are then not deleted
|
|
@pytest.mark.xfail
|
|
@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.xfail
|
|
@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
|