jpt-archive/saucebrush
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use cStringIO on Python 2.x, io.StringIO on 3.x

Browse Source
This commit is contained in:
Jeremy Carbaugh 2012-03-11 23:17:02 -07:00
parent 79976e3c78
commit 7f69f6a769
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
saucebrush/tests/emitters.py
View File

@ -1,6 +1,6 @@
from __future__ import unicode_literals
from contextlib import closing
from io import BytesIO, StringIO
from io import StringIO
import unittest
from saucebrush.emitters import DebugEmitter, CSVEmitter, CountEmitter
@ -16,8 +16,8 @@ class EmitterTestCase(unittest.TestCase):
def test_csv_emitter(self):
try:
import cStringIO # if Python 2.x then use BytesIO
io = BytesIO()
import cStringIO # if Python 2.x then use old cStringIO
io = cStringIO.StringIO()
except:
io = StringIO() # if Python 3.x then use StringIO