225 lines
6.9 KiB
Python
225 lines
6.9 KiB
Python
# Tests StringIO and cStringIO
|
|
|
|
import unittest
|
|
import StringIO
|
|
import cStringIO
|
|
import types
|
|
import array
|
|
import sys
|
|
from test import test_support
|
|
|
|
|
|
class TestGenericStringIO(unittest.TestCase):
|
|
# use a class variable MODULE to define which module is being tested
|
|
|
|
# Line of data to test as string
|
|
_line = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!'
|
|
|
|
# Constructor to use for the test data (._line is passed to this
|
|
# constructor)
|
|
constructor = str
|
|
|
|
def setUp(self):
|
|
self._lines = self.constructor((self._line + '\n') * 5)
|
|
self._fp = self.MODULE.StringIO(self._lines)
|
|
|
|
def test_reads(self):
|
|
eq = self.assertEqual
|
|
self.assertRaises(TypeError, self._fp.seek)
|
|
eq(self._fp.read(10), self._line[:10])
|
|
eq(self._fp.read(0), '')
|
|
eq(self._fp.readline(0), '')
|
|
eq(self._fp.readline(), self._line[10:] + '\n')
|
|
eq(len(self._fp.readlines(60)), 2)
|
|
self._fp.seek(0)
|
|
eq(self._fp.readline(-1), self._line + '\n')
|
|
|
|
def test_writes(self):
|
|
f = self.MODULE.StringIO()
|
|
self.assertRaises(TypeError, f.seek)
|
|
f.write(self._line[:6])
|
|
f.seek(3)
|
|
f.write(self._line[20:26])
|
|
f.write(self._line[52])
|
|
self.assertEqual(f.getvalue(), 'abcuvwxyz!')
|
|
|
|
def test_writelines(self):
|
|
f = self.MODULE.StringIO()
|
|
f.writelines([self._line[0], self._line[1], self._line[2]])
|
|
f.seek(0)
|
|
self.assertEqual(f.getvalue(), 'abc')
|
|
|
|
def test_writelines_error(self):
|
|
def errorGen():
|
|
yield 'a'
|
|
raise KeyboardInterrupt()
|
|
f = self.MODULE.StringIO()
|
|
self.assertRaises(KeyboardInterrupt, f.writelines, errorGen())
|
|
|
|
def test_truncate(self):
|
|
eq = self.assertEqual
|
|
f = self.MODULE.StringIO()
|
|
f.write(self._lines)
|
|
f.seek(10)
|
|
f.truncate()
|
|
eq(f.getvalue(), 'abcdefghij')
|
|
f.truncate(5)
|
|
eq(f.getvalue(), 'abcde')
|
|
f.write('xyz')
|
|
eq(f.getvalue(), 'abcdexyz')
|
|
self.assertRaises(IOError, f.truncate, -1)
|
|
f.close()
|
|
self.assertRaises(ValueError, f.write, 'frobnitz')
|
|
|
|
def test_closed_flag(self):
|
|
f = self.MODULE.StringIO()
|
|
self.assertEqual(f.closed, False)
|
|
f.close()
|
|
self.assertEqual(f.closed, True)
|
|
f = self.MODULE.StringIO("abc")
|
|
self.assertEqual(f.closed, False)
|
|
f.close()
|
|
self.assertEqual(f.closed, True)
|
|
|
|
def test_isatty(self):
|
|
f = self.MODULE.StringIO()
|
|
self.assertRaises(TypeError, f.isatty, None)
|
|
self.assertEqual(f.isatty(), False)
|
|
f.close()
|
|
self.assertRaises(ValueError, f.isatty)
|
|
|
|
def test_iterator(self):
|
|
eq = self.assertEqual
|
|
unless = self.assertTrue
|
|
eq(iter(self._fp), self._fp)
|
|
# Does this object support the iteration protocol?
|
|
unless(hasattr(self._fp, '__iter__'))
|
|
unless(hasattr(self._fp, 'next'))
|
|
i = 0
|
|
for line in self._fp:
|
|
eq(line, self._line + '\n')
|
|
i += 1
|
|
eq(i, 5)
|
|
self._fp.close()
|
|
self.assertRaises(ValueError, self._fp.next)
|
|
|
|
def test_getvalue(self):
|
|
self._fp.close()
|
|
self.assertRaises(ValueError, self._fp.getvalue)
|
|
|
|
@test_support.bigmemtest(test_support._2G + 2**26, memuse=2.001)
|
|
def test_reads_from_large_stream(self, size):
|
|
linesize = 2**26 # 64 MiB
|
|
lines = ['x' * (linesize - 1) + '\n'] * (size // linesize) + \
|
|
['y' * (size % linesize)]
|
|
f = self.MODULE.StringIO(''.join(lines))
|
|
for i, expected in enumerate(lines):
|
|
line = f.read(len(expected))
|
|
self.assertEqual(len(line), len(expected))
|
|
self.assertEqual(line, expected)
|
|
self.assertEqual(f.read(), '')
|
|
f.seek(0)
|
|
for i, expected in enumerate(lines):
|
|
line = f.readline()
|
|
self.assertEqual(len(line), len(expected))
|
|
self.assertEqual(line, expected)
|
|
self.assertEqual(f.readline(), '')
|
|
f.seek(0)
|
|
self.assertEqual(f.readlines(), lines)
|
|
self.assertEqual(f.readlines(), [])
|
|
f.seek(0)
|
|
self.assertEqual(f.readlines(size), lines)
|
|
self.assertEqual(f.readlines(), [])
|
|
|
|
# In worst case cStringIO requires 2 + 1 + 1/2 + 1/2**2 + ... = 4
|
|
# bytes per input character.
|
|
@test_support.bigmemtest(test_support._2G, memuse=4)
|
|
def test_writes_to_large_stream(self, size):
|
|
s = 'x' * 2**26 # 64 MiB
|
|
f = self.MODULE.StringIO()
|
|
n = size
|
|
while n > len(s):
|
|
f.write(s)
|
|
n -= len(s)
|
|
s = None
|
|
f.write('x' * n)
|
|
self.assertEqual(len(f.getvalue()), size)
|
|
|
|
|
|
class TestStringIO(TestGenericStringIO):
|
|
MODULE = StringIO
|
|
|
|
def test_unicode(self):
|
|
|
|
if not test_support.have_unicode: return
|
|
|
|
# The StringIO module also supports concatenating Unicode
|
|
# snippets to larger Unicode strings. This is tested by this
|
|
# method. Note that cStringIO does not support this extension.
|
|
|
|
f = self.MODULE.StringIO()
|
|
f.write(self._line[:6])
|
|
f.seek(3)
|
|
f.write(unicode(self._line[20:26]))
|
|
f.write(unicode(self._line[52]))
|
|
s = f.getvalue()
|
|
self.assertEqual(s, unicode('abcuvwxyz!'))
|
|
self.assertEqual(type(s), types.UnicodeType)
|
|
|
|
class TestcStringIO(TestGenericStringIO):
|
|
MODULE = cStringIO
|
|
|
|
def test_array_support(self):
|
|
# Issue #1730114: cStringIO should accept array objects
|
|
a = array.array('B', [0,1,2])
|
|
f = self.MODULE.StringIO(a)
|
|
self.assertEqual(f.getvalue(), '\x00\x01\x02')
|
|
|
|
def test_unicode(self):
|
|
|
|
if not test_support.have_unicode: return
|
|
|
|
# The cStringIO module converts Unicode strings to character
|
|
# strings when writing them to cStringIO objects.
|
|
# Check that this works.
|
|
|
|
f = self.MODULE.StringIO()
|
|
f.write(u'abcde')
|
|
s = f.getvalue()
|
|
self.assertEqual(s, 'abcde')
|
|
self.assertEqual(type(s), str)
|
|
|
|
f = self.MODULE.StringIO(u'abcde')
|
|
s = f.getvalue()
|
|
self.assertEqual(s, 'abcde')
|
|
self.assertEqual(type(s), str)
|
|
|
|
self.assertRaises(UnicodeEncodeError, self.MODULE.StringIO, u'\xf4')
|
|
|
|
|
|
import sys
|
|
if sys.platform.startswith('java'):
|
|
# Jython doesn't have a buffer object, so we just do a useless
|
|
# fake of the buffer tests.
|
|
buffer = str
|
|
|
|
class TestBufferStringIO(TestStringIO):
|
|
constructor = buffer
|
|
|
|
class TestBuffercStringIO(TestcStringIO):
|
|
constructor = buffer
|
|
|
|
class TestMemoryviewcStringIO(TestcStringIO):
|
|
constructor = memoryview
|
|
|
|
|
|
def test_main():
|
|
test_support.run_unittest(TestStringIO, TestcStringIO)
|
|
with test_support.check_py3k_warnings(("buffer.. not supported",
|
|
DeprecationWarning)):
|
|
test_support.run_unittest(TestBufferStringIO, TestBuffercStringIO)
|
|
test_support.run_unittest(TestMemoryviewcStringIO)
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|