* Fix StringIO for Python 3

This commit is contained in:
Matthew Honnibal 2015-10-13 20:20:10 +11:00
parent 7a15d1b60c
commit 719221a215
4 changed files with 9 additions and 8 deletions

View File

@ -1,7 +1,7 @@
import pytest
import pickle
import StringIO
import io
from spacy.morphology import Morphology
@ -12,6 +12,6 @@ from spacy.strings import StringStore
def test_pickle():
morphology = Morphology(StringStore(), {}, Lemmatizer({}, {}, {}))
file_ = StringIO.StringIO()
file_ = io.BytesIO()
pickle.dump(morphology, file_)

View File

@ -2,12 +2,13 @@ import pytest
import pickle
import cloudpickle
import StringIO
import io
@pytest.mark.models
def test_pickle(EN):
file_ = StringIO.StringIO()
file_ = io.BytesIO()
cloudpickle.dump(EN.parser, file_)
file_.seek(0)

View File

@ -1,12 +1,12 @@
import pytest
import StringIO
import io
import cloudpickle
import pickle
@pytest.mark.models
def test_pickle_english(EN):
file_ = StringIO.StringIO()
file_ = io.BytesIO()
cloudpickle.dump(EN, file_)
file_.seek(0)

View File

@ -1,6 +1,6 @@
from __future__ import unicode_literals
import pytest
import StringIO
import io
import cloudpickle
import pickle
@ -46,7 +46,7 @@ def test_symbols(en_vocab):
def test_pickle_vocab(en_vocab):
file_ = StringIO.StringIO()
file_ = io.BytesIO()
cloudpickle.dump(en_vocab, file_)
file_.seek(0)