]>
Commit | Line | Data |
---|---|---|
4710c53d | 1 | # Tests that work for both bytes and buffer objects.\r |
2 | # See PEP 3137.\r | |
3 | \r | |
4 | import struct\r | |
5 | import sys\r | |
6 | \r | |
7 | class MixinBytesBufferCommonTests(object):\r | |
8 | """Tests that work for both bytes and buffer objects.\r | |
9 | See PEP 3137.\r | |
10 | """\r | |
11 | \r | |
12 | def marshal(self, x):\r | |
13 | """Convert x into the appropriate type for these tests."""\r | |
14 | raise RuntimeError('test class must provide a marshal method')\r | |
15 | \r | |
16 | def test_islower(self):\r | |
17 | self.assertFalse(self.marshal(b'').islower())\r | |
18 | self.assertTrue(self.marshal(b'a').islower())\r | |
19 | self.assertFalse(self.marshal(b'A').islower())\r | |
20 | self.assertFalse(self.marshal(b'\n').islower())\r | |
21 | self.assertTrue(self.marshal(b'abc').islower())\r | |
22 | self.assertFalse(self.marshal(b'aBc').islower())\r | |
23 | self.assertTrue(self.marshal(b'abc\n').islower())\r | |
24 | self.assertRaises(TypeError, self.marshal(b'abc').islower, 42)\r | |
25 | \r | |
26 | def test_isupper(self):\r | |
27 | self.assertFalse(self.marshal(b'').isupper())\r | |
28 | self.assertFalse(self.marshal(b'a').isupper())\r | |
29 | self.assertTrue(self.marshal(b'A').isupper())\r | |
30 | self.assertFalse(self.marshal(b'\n').isupper())\r | |
31 | self.assertTrue(self.marshal(b'ABC').isupper())\r | |
32 | self.assertFalse(self.marshal(b'AbC').isupper())\r | |
33 | self.assertTrue(self.marshal(b'ABC\n').isupper())\r | |
34 | self.assertRaises(TypeError, self.marshal(b'abc').isupper, 42)\r | |
35 | \r | |
36 | def test_istitle(self):\r | |
37 | self.assertFalse(self.marshal(b'').istitle())\r | |
38 | self.assertFalse(self.marshal(b'a').istitle())\r | |
39 | self.assertTrue(self.marshal(b'A').istitle())\r | |
40 | self.assertFalse(self.marshal(b'\n').istitle())\r | |
41 | self.assertTrue(self.marshal(b'A Titlecased Line').istitle())\r | |
42 | self.assertTrue(self.marshal(b'A\nTitlecased Line').istitle())\r | |
43 | self.assertTrue(self.marshal(b'A Titlecased, Line').istitle())\r | |
44 | self.assertFalse(self.marshal(b'Not a capitalized String').istitle())\r | |
45 | self.assertFalse(self.marshal(b'Not\ta Titlecase String').istitle())\r | |
46 | self.assertFalse(self.marshal(b'Not--a Titlecase String').istitle())\r | |
47 | self.assertFalse(self.marshal(b'NOT').istitle())\r | |
48 | self.assertRaises(TypeError, self.marshal(b'abc').istitle, 42)\r | |
49 | \r | |
50 | def test_isspace(self):\r | |
51 | self.assertFalse(self.marshal(b'').isspace())\r | |
52 | self.assertFalse(self.marshal(b'a').isspace())\r | |
53 | self.assertTrue(self.marshal(b' ').isspace())\r | |
54 | self.assertTrue(self.marshal(b'\t').isspace())\r | |
55 | self.assertTrue(self.marshal(b'\r').isspace())\r | |
56 | self.assertTrue(self.marshal(b'\n').isspace())\r | |
57 | self.assertTrue(self.marshal(b' \t\r\n').isspace())\r | |
58 | self.assertFalse(self.marshal(b' \t\r\na').isspace())\r | |
59 | self.assertRaises(TypeError, self.marshal(b'abc').isspace, 42)\r | |
60 | \r | |
61 | def test_isalpha(self):\r | |
62 | self.assertFalse(self.marshal(b'').isalpha())\r | |
63 | self.assertTrue(self.marshal(b'a').isalpha())\r | |
64 | self.assertTrue(self.marshal(b'A').isalpha())\r | |
65 | self.assertFalse(self.marshal(b'\n').isalpha())\r | |
66 | self.assertTrue(self.marshal(b'abc').isalpha())\r | |
67 | self.assertFalse(self.marshal(b'aBc123').isalpha())\r | |
68 | self.assertFalse(self.marshal(b'abc\n').isalpha())\r | |
69 | self.assertRaises(TypeError, self.marshal(b'abc').isalpha, 42)\r | |
70 | \r | |
71 | def test_isalnum(self):\r | |
72 | self.assertFalse(self.marshal(b'').isalnum())\r | |
73 | self.assertTrue(self.marshal(b'a').isalnum())\r | |
74 | self.assertTrue(self.marshal(b'A').isalnum())\r | |
75 | self.assertFalse(self.marshal(b'\n').isalnum())\r | |
76 | self.assertTrue(self.marshal(b'123abc456').isalnum())\r | |
77 | self.assertTrue(self.marshal(b'a1b3c').isalnum())\r | |
78 | self.assertFalse(self.marshal(b'aBc000 ').isalnum())\r | |
79 | self.assertFalse(self.marshal(b'abc\n').isalnum())\r | |
80 | self.assertRaises(TypeError, self.marshal(b'abc').isalnum, 42)\r | |
81 | \r | |
82 | def test_isdigit(self):\r | |
83 | self.assertFalse(self.marshal(b'').isdigit())\r | |
84 | self.assertFalse(self.marshal(b'a').isdigit())\r | |
85 | self.assertTrue(self.marshal(b'0').isdigit())\r | |
86 | self.assertTrue(self.marshal(b'0123456789').isdigit())\r | |
87 | self.assertFalse(self.marshal(b'0123456789a').isdigit())\r | |
88 | \r | |
89 | self.assertRaises(TypeError, self.marshal(b'abc').isdigit, 42)\r | |
90 | \r | |
91 | def test_lower(self):\r | |
92 | self.assertEqual(b'hello', self.marshal(b'HeLLo').lower())\r | |
93 | self.assertEqual(b'hello', self.marshal(b'hello').lower())\r | |
94 | self.assertRaises(TypeError, self.marshal(b'hello').lower, 42)\r | |
95 | \r | |
96 | def test_upper(self):\r | |
97 | self.assertEqual(b'HELLO', self.marshal(b'HeLLo').upper())\r | |
98 | self.assertEqual(b'HELLO', self.marshal(b'HELLO').upper())\r | |
99 | self.assertRaises(TypeError, self.marshal(b'hello').upper, 42)\r | |
100 | \r | |
101 | def test_capitalize(self):\r | |
102 | self.assertEqual(b' hello ', self.marshal(b' hello ').capitalize())\r | |
103 | self.assertEqual(b'Hello ', self.marshal(b'Hello ').capitalize())\r | |
104 | self.assertEqual(b'Hello ', self.marshal(b'hello ').capitalize())\r | |
105 | self.assertEqual(b'Aaaa', self.marshal(b'aaaa').capitalize())\r | |
106 | self.assertEqual(b'Aaaa', self.marshal(b'AaAa').capitalize())\r | |
107 | \r | |
108 | self.assertRaises(TypeError, self.marshal(b'hello').capitalize, 42)\r | |
109 | \r | |
110 | def test_ljust(self):\r | |
111 | self.assertEqual(b'abc ', self.marshal(b'abc').ljust(10))\r | |
112 | self.assertEqual(b'abc ', self.marshal(b'abc').ljust(6))\r | |
113 | self.assertEqual(b'abc', self.marshal(b'abc').ljust(3))\r | |
114 | self.assertEqual(b'abc', self.marshal(b'abc').ljust(2))\r | |
115 | self.assertEqual(b'abc*******', self.marshal(b'abc').ljust(10, '*'))\r | |
116 | self.assertRaises(TypeError, self.marshal(b'abc').ljust)\r | |
117 | \r | |
118 | def test_rjust(self):\r | |
119 | self.assertEqual(b' abc', self.marshal(b'abc').rjust(10))\r | |
120 | self.assertEqual(b' abc', self.marshal(b'abc').rjust(6))\r | |
121 | self.assertEqual(b'abc', self.marshal(b'abc').rjust(3))\r | |
122 | self.assertEqual(b'abc', self.marshal(b'abc').rjust(2))\r | |
123 | self.assertEqual(b'*******abc', self.marshal(b'abc').rjust(10, '*'))\r | |
124 | self.assertRaises(TypeError, self.marshal(b'abc').rjust)\r | |
125 | \r | |
126 | def test_center(self):\r | |
127 | self.assertEqual(b' abc ', self.marshal(b'abc').center(10))\r | |
128 | self.assertEqual(b' abc ', self.marshal(b'abc').center(6))\r | |
129 | self.assertEqual(b'abc', self.marshal(b'abc').center(3))\r | |
130 | self.assertEqual(b'abc', self.marshal(b'abc').center(2))\r | |
131 | self.assertEqual(b'***abc****', self.marshal(b'abc').center(10, '*'))\r | |
132 | self.assertRaises(TypeError, self.marshal(b'abc').center)\r | |
133 | \r | |
134 | def test_swapcase(self):\r | |
135 | self.assertEqual(b'hEllO CoMPuTErS',\r | |
136 | self.marshal(b'HeLLo cOmpUteRs').swapcase())\r | |
137 | \r | |
138 | self.assertRaises(TypeError, self.marshal(b'hello').swapcase, 42)\r | |
139 | \r | |
140 | def test_zfill(self):\r | |
141 | self.assertEqual(b'123', self.marshal(b'123').zfill(2))\r | |
142 | self.assertEqual(b'123', self.marshal(b'123').zfill(3))\r | |
143 | self.assertEqual(b'0123', self.marshal(b'123').zfill(4))\r | |
144 | self.assertEqual(b'+123', self.marshal(b'+123').zfill(3))\r | |
145 | self.assertEqual(b'+123', self.marshal(b'+123').zfill(4))\r | |
146 | self.assertEqual(b'+0123', self.marshal(b'+123').zfill(5))\r | |
147 | self.assertEqual(b'-123', self.marshal(b'-123').zfill(3))\r | |
148 | self.assertEqual(b'-123', self.marshal(b'-123').zfill(4))\r | |
149 | self.assertEqual(b'-0123', self.marshal(b'-123').zfill(5))\r | |
150 | self.assertEqual(b'000', self.marshal(b'').zfill(3))\r | |
151 | self.assertEqual(b'34', self.marshal(b'34').zfill(1))\r | |
152 | self.assertEqual(b'0034', self.marshal(b'34').zfill(4))\r | |
153 | \r | |
154 | self.assertRaises(TypeError, self.marshal(b'123').zfill)\r | |
155 | \r | |
156 | def test_expandtabs(self):\r | |
157 | self.assertEqual(b'abc\rab def\ng hi',\r | |
158 | self.marshal(b'abc\rab\tdef\ng\thi').expandtabs())\r | |
159 | self.assertEqual(b'abc\rab def\ng hi',\r | |
160 | self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8))\r | |
161 | self.assertEqual(b'abc\rab def\ng hi',\r | |
162 | self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(4))\r | |
163 | self.assertEqual(b'abc\r\nab def\ng hi',\r | |
164 | self.marshal(b'abc\r\nab\tdef\ng\thi').expandtabs(4))\r | |
165 | self.assertEqual(b'abc\rab def\ng hi',\r | |
166 | self.marshal(b'abc\rab\tdef\ng\thi').expandtabs())\r | |
167 | self.assertEqual(b'abc\rab def\ng hi',\r | |
168 | self.marshal(b'abc\rab\tdef\ng\thi').expandtabs(8))\r | |
169 | self.assertEqual(b'abc\r\nab\r\ndef\ng\r\nhi',\r | |
170 | self.marshal(b'abc\r\nab\r\ndef\ng\r\nhi').expandtabs(4))\r | |
171 | self.assertEqual(b' a\n b', self.marshal(b' \ta\n\tb').expandtabs(1))\r | |
172 | \r | |
173 | self.assertRaises(TypeError, self.marshal(b'hello').expandtabs, 42, 42)\r | |
174 | # This test is only valid when sizeof(int) == sizeof(void*) == 4.\r | |
175 | if sys.maxint < (1 << 32) and struct.calcsize('P') == 4:\r | |
176 | self.assertRaises(OverflowError,\r | |
177 | self.marshal(b'\ta\n\tb').expandtabs, sys.maxint)\r | |
178 | \r | |
179 | def test_title(self):\r | |
180 | self.assertEqual(b' Hello ', self.marshal(b' hello ').title())\r | |
181 | self.assertEqual(b'Hello ', self.marshal(b'hello ').title())\r | |
182 | self.assertEqual(b'Hello ', self.marshal(b'Hello ').title())\r | |
183 | self.assertEqual(b'Format This As Title String',\r | |
184 | self.marshal(b'fOrMaT thIs aS titLe String').title())\r | |
185 | self.assertEqual(b'Format,This-As*Title;String',\r | |
186 | self.marshal(b'fOrMaT,thIs-aS*titLe;String').title())\r | |
187 | self.assertEqual(b'Getint', self.marshal(b'getInt').title())\r | |
188 | self.assertRaises(TypeError, self.marshal(b'hello').title, 42)\r | |
189 | \r | |
190 | def test_splitlines(self):\r | |
191 | self.assertEqual([b'abc', b'def', b'', b'ghi'],\r | |
192 | self.marshal(b'abc\ndef\n\rghi').splitlines())\r | |
193 | self.assertEqual([b'abc', b'def', b'', b'ghi'],\r | |
194 | self.marshal(b'abc\ndef\n\r\nghi').splitlines())\r | |
195 | self.assertEqual([b'abc', b'def', b'ghi'],\r | |
196 | self.marshal(b'abc\ndef\r\nghi').splitlines())\r | |
197 | self.assertEqual([b'abc', b'def', b'ghi'],\r | |
198 | self.marshal(b'abc\ndef\r\nghi\n').splitlines())\r | |
199 | self.assertEqual([b'abc', b'def', b'ghi', b''],\r | |
200 | self.marshal(b'abc\ndef\r\nghi\n\r').splitlines())\r | |
201 | self.assertEqual([b'', b'abc', b'def', b'ghi', b''],\r | |
202 | self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines())\r | |
203 | self.assertEqual([b'\n', b'abc\n', b'def\r\n', b'ghi\n', b'\r'],\r | |
204 | self.marshal(b'\nabc\ndef\r\nghi\n\r').splitlines(1))\r | |
205 | \r | |
206 | self.assertRaises(TypeError, self.marshal(b'abc').splitlines, 42, 42)\r |