summaryrefslogtreecommitdiff
path: root/lib/python2.7/test/test_StringIO.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/python2.7/test/test_StringIO.py')
-rw-r--r--lib/python2.7/test/test_StringIO.py224
1 files changed, 0 insertions, 224 deletions
diff --git a/lib/python2.7/test/test_StringIO.py b/lib/python2.7/test/test_StringIO.py
deleted file mode 100644
index 42f307a..0000000
--- a/lib/python2.7/test/test_StringIO.py
+++ /dev/null
@@ -1,224 +0,0 @@
-# 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()
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-21 21:33:26 +0000