2015-10-12 11:33:11 +03:00
|
|
|
import cloudpickle
|
2015-10-24 08:18:47 +03:00
|
|
|
import io
|
|
|
|
import os
|
2015-10-12 11:33:11 +03:00
|
|
|
import pickle
|
2015-10-24 08:18:47 +03:00
|
|
|
import pytest
|
|
|
|
import tempfile
|
2015-10-12 11:33:11 +03:00
|
|
|
|
|
|
|
@pytest.mark.models
|
|
|
|
def test_pickle_english(EN):
|
2015-10-13 12:20:10 +03:00
|
|
|
file_ = io.BytesIO()
|
2015-10-12 11:33:11 +03:00
|
|
|
cloudpickle.dump(EN, file_)
|
|
|
|
|
|
|
|
file_.seek(0)
|
|
|
|
|
|
|
|
loaded = pickle.load(file_)
|
2015-10-24 08:18:47 +03:00
|
|
|
assert loaded is not None
|
2015-10-12 11:33:11 +03:00
|
|
|
|
2015-10-24 08:18:47 +03:00
|
|
|
@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))
|
|
|
|
os.unlink(f.name)
|
|
|
|
doc = loaded_en(unicode('test parse'))
|
|
|
|
assert len(doc) == 2
|