* Fix StringIO for Python3 tests

This commit is contained in:
Matthew Honnibal 2015-10-13 20:50:07 +11:00
parent e3f97cf938
commit c772df2f0c

View File

@ -1,7 +1,7 @@
# -*- coding: utf8 -*- # -*- coding: utf8 -*-
from __future__ import unicode_literals from __future__ import unicode_literals
import pickle import pickle
import StringIO import io
from spacy.strings import StringStore from spacy.strings import StringStore
@ -82,7 +82,7 @@ def test_massive_strings(sstore):
def test_pickle_string_store(sstore): def test_pickle_string_store(sstore):
hello_id = sstore[u'Hi'] hello_id = sstore[u'Hi']
string_file = StringIO.StringIO() string_file = io.BytesIO()
pickle.dump(sstore, string_file) pickle.dump(sstore, string_file)
string_file.seek(0) string_file.seek(0)