Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_StringIO.py @ 28646:193b8aec00ce legacy-trunk

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Changes from Jonathan Riehl to allow his pgen extension (PEP 269) to work. This includes some more code that used to be part of pgen in the main parser; I'm okay with that. I'll see if the Windows build needs work next.
author Guido van Rossum <guido@python.org>
date Thu, 17 Apr 2003 14:55:42 +0000
parents aa0d2d51fbde
children 0ea5991f6edb
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
14688
d6ffb6d7fe8a New test cases for the StringIO module
Martin v. Löwis <martin@v.loewis.de>
parents:
diff changeset
1 # Tests StringIO and cStringIO
d6ffb6d7fe8a New test cases for the StringIO module
Martin v. Löwis <martin@v.loewis.de>
parents:
diff changeset
2
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
3 import unittest
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
4 import StringIO
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
5 import cStringIO
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
6 import types
24364
aa0d2d51fbde Get rid of relative imports in all unittests. Now anything that
Barry Warsaw <barry@python.org>
parents: 23432
diff changeset
7 from test import test_support
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
8
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
9
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
10 class TestGenericStringIO(unittest.TestCase):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
11 # use a class variable MODULE to define which module is being tested
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
12
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
13 # Line of data to test as string
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
14 _line = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!'
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
15
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
16 # Constructor to use for the test data (._line is passed to this
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
17 # constructor)
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
18 constructor = str
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
19
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
20 def setUp(self):
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
21 self._line = self.constructor(self._line)
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
22 self._lines = self.constructor((self._line + '\n') * 5)
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
23 self._fp = self.MODULE.StringIO(self._lines)
14688
d6ffb6d7fe8a New test cases for the StringIO module
Martin v. Löwis <martin@v.loewis.de>
parents:
diff changeset
24
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
25 def test_reads(self):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
26 eq = self.assertEqual
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
27 eq(self._fp.read(10), self._line[:10])
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
28 eq(self._fp.readline(), self._line[10:] + '\n')
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
29 eq(len(self._fp.readlines(60)), 2)
16546
60a2e99c863b In O_writelines: Replace use of string.joinfields with "".join.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 15175
diff changeset
30
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
31 def test_writes(self):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
32 f = self.MODULE.StringIO()
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
33 f.write(self._line[:6])
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
34 f.seek(3)
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
35 f.write(self._line[20:26])
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
36 f.write(self._line[52])
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
37 self.assertEqual(f.getvalue(), 'abcuvwxyz!')
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
38
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
39 def test_writelines(self):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
40 f = self.MODULE.StringIO()
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
41 f.writelines([self._line[0], self._line[1], self._line[2]])
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
42 f.seek(0)
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
43 self.assertEqual(f.getvalue(), 'abc')
16546
60a2e99c863b In O_writelines: Replace use of string.joinfields with "".join.
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 15175
diff changeset
44
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
45 def test_truncate(self):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
46 eq = self.assertEqual
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
47 f = self.MODULE.StringIO()
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
48 f.write(self._lines)
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
49 f.seek(10)
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
50 f.truncate()
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
51 eq(f.getvalue(), 'abcdefghij')
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
52 f.seek(0)
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
53 f.truncate(5)
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
54 eq(f.getvalue(), 'abcde')
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
55 f.close()
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
56 self.assertRaises(ValueError, f.write, 'frobnitz')
14905
45967ff1b6dc Added some tests for the truncate() method; one is commented out because
Fred Drake <fdrake@acm.org>
parents: 14688
diff changeset
57
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
58 def test_iterator(self):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
59 eq = self.assertEqual
20096
19abc339d3fc test_iterator(): Don't do a type comparison to see if it's an
Barry Warsaw <barry@python.org>
parents: 20060
diff changeset
60 unless = self.failUnless
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
61 it = iter(self._fp)
20096
19abc339d3fc test_iterator(): Don't do a type comparison to see if it's an
Barry Warsaw <barry@python.org>
parents: 20060
diff changeset
62 # Does this object support the iteration protocol?
19abc339d3fc test_iterator(): Don't do a type comparison to see if it's an
Barry Warsaw <barry@python.org>
parents: 20060
diff changeset
63 unless(hasattr(it, '__iter__'))
19abc339d3fc test_iterator(): Don't do a type comparison to see if it's an
Barry Warsaw <barry@python.org>
parents: 20060
diff changeset
64 unless(hasattr(it, 'next'))
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
65 i = 0
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
66 for line in self._fp:
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
67 eq(line, self._line + '\n')
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
68 i += 1
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
69 eq(i, 5)
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
70
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
71 class TestStringIO(TestGenericStringIO):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
72 MODULE = StringIO
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
73
21656
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
74 def test_unicode(self):
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
75
23432
56240bcda88c Make StringIO work in --disable-unicode builds...
Michael W. Hudson <mwh@python.net>
parents: 21958
diff changeset
76 if not test_support.have_unicode: return
56240bcda88c Make StringIO work in --disable-unicode builds...
Michael W. Hudson <mwh@python.net>
parents: 21958
diff changeset
77
21656
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
78 # The StringIO module also supports concatenating Unicode
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
79 # snippets to larger Unicode strings. This is tested by this
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
80 # method. Note that cStringIO does not support this extension.
21958
5ed782bdbb2a Whitespace normalization.
Tim Peters <tim.peters@gmail.com>
parents: 21656
diff changeset
81
21656
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
82 f = self.MODULE.StringIO()
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
83 f.write(self._line[:6])
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
84 f.seek(3)
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
85 f.write(unicode(self._line[20:26]))
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
86 f.write(unicode(self._line[52]))
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
87 s = f.getvalue()
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
88 self.assertEqual(s, unicode('abcuvwxyz!'))
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
89 self.assertEqual(type(s), types.UnicodeType)
6b0e16cb65d5 Restore Python 2.1 StringIO.py behaviour: support concatenating
Marc-André Lemburg <mal@egenix.com>
parents: 21320
diff changeset
90
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
91 class TestcStringIO(TestGenericStringIO):
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
92 MODULE = cStringIO
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
93
21320
60df2bf854ac A workaround for the missing buffer() builtin in jython.
Finn Bock <bckfnn@worldonline.dk>
parents: 20096
diff changeset
94 import sys
60df2bf854ac A workaround for the missing buffer() builtin in jython.
Finn Bock <bckfnn@worldonline.dk>
parents: 20096
diff changeset
95 if sys.platform.startswith('java'):
60df2bf854ac A workaround for the missing buffer() builtin in jython.
Finn Bock <bckfnn@worldonline.dk>
parents: 20096
diff changeset
96 # Jython doesn't have a buffer object, so we just do a useless
60df2bf854ac A workaround for the missing buffer() builtin in jython.
Finn Bock <bckfnn@worldonline.dk>
parents: 20096
diff changeset
97 # fake of the buffer tests.
60df2bf854ac A workaround for the missing buffer() builtin in jython.
Finn Bock <bckfnn@worldonline.dk>
parents: 20096
diff changeset
98 buffer = str
60df2bf854ac A workaround for the missing buffer() builtin in jython.
Finn Bock <bckfnn@worldonline.dk>
parents: 20096
diff changeset
99
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
100 class TestBufferStringIO(TestStringIO):
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
101 constructor = buffer
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
102
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
103 class TestBuffercStringIO(TestcStringIO):
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
104 constructor = buffer
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
105
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
106
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
107 def test_main():
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
108 test_support.run_unittest(TestStringIO)
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
109 test_support.run_unittest(TestcStringIO)
20060
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
110 test_support.run_unittest(TestBufferStringIO)
ccec02bfe9a0 StringIO patch #462596: let's [c]StringIO accept read buffers on
Marc-André Lemburg <mal@egenix.com>
parents: 20030
diff changeset
111 test_support.run_unittest(TestBuffercStringIO)
20030
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
112
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
113 if __name__ == '__main__':
6378cb117b73 Converted test_StringIO.py to use unittest, so
Barry Warsaw <barry@python.org>
parents: 16546
diff changeset
114 test_main()