Don't insist on cStringIO.
This commit is contained in:
parent
62b8871561
commit
5c4501af57
|
@ -1,7 +1,8 @@
|
|||
from test import test_support
|
||||
import unittest
|
||||
import codecs
|
||||
import sys, cStringIO, _testcapi
|
||||
import sys, _testcapi
|
||||
from StringIO import StringIO
|
||||
|
||||
class Queue(object):
|
||||
"""
|
||||
|
@ -103,7 +104,7 @@ class ReadTest(unittest.TestCase, MixInCheckStateHandling):
|
|||
|
||||
def test_readline(self):
|
||||
def getreader(input):
|
||||
stream = cStringIO.StringIO(input.encode(self.encoding))
|
||||
stream = StringIO(input.encode(self.encoding))
|
||||
return codecs.getreader(self.encoding)(stream)
|
||||
|
||||
def readalllines(input, keepends=True, size=None):
|
||||
|
@ -215,7 +216,7 @@ class ReadTest(unittest.TestCase, MixInCheckStateHandling):
|
|||
' break\r\n',
|
||||
' \r\n',
|
||||
]
|
||||
stream = cStringIO.StringIO("".join(s).encode(self.encoding))
|
||||
stream = StringIO("".join(s).encode(self.encoding))
|
||||
reader = codecs.getreader(self.encoding)(stream)
|
||||
for (i, line) in enumerate(reader):
|
||||
self.assertEqual(line, s[i])
|
||||
|
@ -253,7 +254,7 @@ class ReadTest(unittest.TestCase, MixInCheckStateHandling):
|
|||
s3 = "next line.\r\n"
|
||||
|
||||
s = (s1+s2+s3).encode(self.encoding)
|
||||
stream = cStringIO.StringIO(s)
|
||||
stream = StringIO(s)
|
||||
reader = codecs.getreader(self.encoding)(stream)
|
||||
self.assertEqual(reader.readline(), s1)
|
||||
self.assertEqual(reader.readline(), s2)
|
||||
|
@ -268,7 +269,7 @@ class ReadTest(unittest.TestCase, MixInCheckStateHandling):
|
|||
s5 = "againokay.\r\n"
|
||||
|
||||
s = (s1+s2+s3+s4+s5).encode(self.encoding)
|
||||
stream = cStringIO.StringIO(s)
|
||||
stream = StringIO(s)
|
||||
reader = codecs.getreader(self.encoding)(stream)
|
||||
self.assertEqual(reader.readline(), s1)
|
||||
self.assertEqual(reader.readline(), s2)
|
||||
|
@ -286,7 +287,7 @@ class UTF16Test(ReadTest):
|
|||
def test_only_one_bom(self):
|
||||
_,_,reader,writer = codecs.lookup(self.encoding)
|
||||
# encode some stream
|
||||
s = cStringIO.StringIO()
|
||||
s = StringIO()
|
||||
f = writer(s)
|
||||
f.write("spam")
|
||||
f.write("spam")
|
||||
|
@ -294,16 +295,16 @@ class UTF16Test(ReadTest):
|
|||
# check whether there is exactly one BOM in it
|
||||
self.assert_(d == self.spamle or d == self.spambe)
|
||||
# try to read it back
|
||||
s = cStringIO.StringIO(d)
|
||||
s = StringIO(d)
|
||||
f = reader(s)
|
||||
self.assertEquals(f.read(), "spamspam")
|
||||
|
||||
def test_badbom(self):
|
||||
s = cStringIO.StringIO(b"\xff\xff")
|
||||
s = StringIO(b"\xff\xff")
|
||||
f = codecs.getreader(self.encoding)(s)
|
||||
self.assertRaises(UnicodeError, f.read)
|
||||
|
||||
s = cStringIO.StringIO(b"\xff\xff\xff\xff")
|
||||
s = StringIO(b"\xff\xff\xff\xff")
|
||||
f = codecs.getreader(self.encoding)(s)
|
||||
self.assertRaises(UnicodeError, f.read)
|
||||
|
||||
|
@ -492,7 +493,7 @@ class EscapeDecodeTest(unittest.TestCase):
|
|||
|
||||
class RecodingTest(unittest.TestCase):
|
||||
def test_recoding(self):
|
||||
f = cStringIO.StringIO()
|
||||
f = StringIO()
|
||||
f2 = codecs.EncodedFile(f, "unicode_internal", "utf-8")
|
||||
f2.write("a")
|
||||
f2.close()
|
||||
|
@ -856,7 +857,7 @@ class IDNACodecTest(unittest.TestCase):
|
|||
self.assertEquals("pyth\xf6n.org.".encode("idna"), "xn--pythn-mua.org.")
|
||||
|
||||
def test_stream(self):
|
||||
r = codecs.getreader("idna")(cStringIO.StringIO(b"abc"))
|
||||
r = codecs.getreader("idna")(StringIO(b"abc"))
|
||||
r.read(3)
|
||||
self.assertEquals(r.read(), "")
|
||||
|
||||
|
@ -964,7 +965,7 @@ class StreamReaderTest(unittest.TestCase):
|
|||
|
||||
def setUp(self):
|
||||
self.reader = codecs.getreader('utf-8')
|
||||
self.stream = cStringIO.StringIO(b'\xed\x95\x9c\n\xea\xb8\x80')
|
||||
self.stream = StringIO(b'\xed\x95\x9c\n\xea\xb8\x80')
|
||||
|
||||
def test_readlines(self):
|
||||
f = self.reader(self.stream)
|
||||
|
@ -973,11 +974,11 @@ class StreamReaderTest(unittest.TestCase):
|
|||
class EncodedFileTest(unittest.TestCase):
|
||||
|
||||
def test_basic(self):
|
||||
f = cStringIO.StringIO(b'\xed\x95\x9c\n\xea\xb8\x80')
|
||||
f = StringIO(b'\xed\x95\x9c\n\xea\xb8\x80')
|
||||
ef = codecs.EncodedFile(f, 'utf-16-le', 'utf-8')
|
||||
self.assertEquals(ef.read(), b'\\\xd5\n\x00\x00\xae')
|
||||
|
||||
f = cStringIO.StringIO()
|
||||
f = StringIO()
|
||||
ef = codecs.EncodedFile(f, 'utf-8', 'latin1')
|
||||
ef.write(b'\xc3\xbc')
|
||||
self.assertEquals(f.getvalue(), b'\xfc')
|
||||
|
@ -986,14 +987,14 @@ class Str2StrTest(unittest.TestCase):
|
|||
|
||||
def test_read(self):
|
||||
sin = "\x80".encode("base64_codec")
|
||||
reader = codecs.getreader("base64_codec")(cStringIO.StringIO(sin))
|
||||
reader = codecs.getreader("base64_codec")(StringIO(sin))
|
||||
sout = reader.read()
|
||||
self.assertEqual(sout, "\x80")
|
||||
self.assert_(isinstance(sout, str))
|
||||
|
||||
def test_readline(self):
|
||||
sin = "\x80".encode("base64_codec")
|
||||
reader = codecs.getreader("base64_codec")(cStringIO.StringIO(sin))
|
||||
reader = codecs.getreader("base64_codec")(StringIO(sin))
|
||||
sout = reader.readline()
|
||||
self.assertEqual(sout, "\x80")
|
||||
self.assert_(isinstance(sout, str))
|
||||
|
@ -1250,7 +1251,7 @@ class BasicUnicodeTest(unittest.TestCase, MixInCheckStateHandling):
|
|||
continue
|
||||
if encoding in broken_unicode_with_streams:
|
||||
continue
|
||||
reader = codecs.getreader(encoding)(cStringIO.StringIO(s.encode(encoding)))
|
||||
reader = codecs.getreader(encoding)(StringIO(s.encode(encoding)))
|
||||
for t in range(5):
|
||||
# Test that calling seek resets the internal codec state and buffers
|
||||
reader.seek(0, 0)
|
||||
|
@ -1327,12 +1328,12 @@ class CharmapTest(unittest.TestCase):
|
|||
|
||||
class WithStmtTest(unittest.TestCase):
|
||||
def test_encodedfile(self):
|
||||
f = cStringIO.StringIO(b"\xc3\xbc")
|
||||
f = StringIO(b"\xc3\xbc")
|
||||
with codecs.EncodedFile(f, "latin-1", "utf-8") as ef:
|
||||
self.assertEquals(ef.read(), b"\xfc")
|
||||
|
||||
def test_streamreaderwriter(self):
|
||||
f = cStringIO.StringIO(b"\xc3\xbc")
|
||||
f = StringIO(b"\xc3\xbc")
|
||||
info = codecs.lookup("utf-8")
|
||||
with codecs.StreamReaderWriter(f, info.streamreader,
|
||||
info.streamwriter, 'strict') as srw:
|
||||
|
|
Loading…
Reference in New Issue