explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 19:39:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Fix StringIO for Python 3

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/morphology/test_morphology_pickle.py
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_)

5
tests/parser/test_parser_pickle.py
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)

4
tests/test_pickle.py
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)

4
tests/vocab/test_vocab.py
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)