blob: 79eb6399f41a0f5935aaf31e87a506665f8b77e8 [file] [log] [blame]
Thomas Hellera4ea6032003-04-17 18:55:45 +00001import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +00002from test import support
Larry Hastings83a9f482012-03-20 20:06:16 +00003from _testcapi import getargs_keywords, getargs_keyword_only
Thomas Hellera4ea6032003-04-17 18:55:45 +00004
5"""
6> How about the following counterproposal. This also changes some of
7> the other format codes to be a little more regular.
8>
9> Code C type Range check
10>
11> b unsigned char 0..UCHAR_MAX
Guido van Rossumbbb931b2003-04-18 00:13:53 +000012> h signed short SHRT_MIN..SHRT_MAX
Thomas Hellera4ea6032003-04-17 18:55:45 +000013> B unsigned char none **
14> H unsigned short none **
15> k * unsigned long none
16> I * unsigned int 0..UINT_MAX
17
18
19> i int INT_MIN..INT_MAX
20> l long LONG_MIN..LONG_MAX
21
22> K * unsigned long long none
23> L long long LLONG_MIN..LLONG_MAX
24
25> Notes:
26>
27> * New format codes.
28>
29> ** Changed from previous "range-and-a-half" to "none"; the
30> range-and-a-half checking wasn't particularly useful.
31
32Plus a C API or two, e.g. PyInt_AsLongMask() ->
33unsigned long and PyInt_AsLongLongMask() -> unsigned
34long long (if that exists).
35"""
36
37LARGE = 0x7FFFFFFF
Guido van Rossume2a383d2007-01-15 16:59:06 +000038VERY_LARGE = 0xFF0000121212121212121242
Thomas Hellera4ea6032003-04-17 18:55:45 +000039
40from _testcapi import UCHAR_MAX, USHRT_MAX, UINT_MAX, ULONG_MAX, INT_MAX, \
Mark Dickinson1554b182009-12-20 16:03:30 +000041 INT_MIN, LONG_MIN, LONG_MAX, PY_SSIZE_T_MIN, PY_SSIZE_T_MAX, \
42 SHRT_MIN, SHRT_MAX
Thomas Hellera4ea6032003-04-17 18:55:45 +000043
Thomas Hellera4ea6032003-04-17 18:55:45 +000044# fake, they are not defined in Python's header files
45LLONG_MAX = 2**63-1
46LLONG_MIN = -2**63
47ULLONG_MAX = 2**64-1
48
Thomas Hellera4ea6032003-04-17 18:55:45 +000049class Int:
50 def __int__(self):
51 return 99
52
Serhiy Storchaka31a65542013-12-11 21:07:54 +020053class IntSubclass(int):
54 def __int__(self):
55 return 99
56
57class BadInt:
58 def __int__(self):
59 return 1.0
60
61class BadInt2:
62 def __int__(self):
63 return True
64
65class BadInt3(int):
66 def __int__(self):
67 return True
68
69
Thomas Hellera4ea6032003-04-17 18:55:45 +000070class Unsigned_TestCase(unittest.TestCase):
71 def test_b(self):
Thomas Heller54aa5782003-04-24 16:15:29 +000072 from _testcapi import getargs_b
Thomas Hellera4ea6032003-04-17 18:55:45 +000073 # b returns 'unsigned char', and does range checking (0 ... UCHAR_MAX)
Thomas Woutersc9471232006-04-15 09:15:11 +000074 self.assertRaises(TypeError, getargs_b, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000075 self.assertEqual(99, getargs_b(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +020076 self.assertEqual(0, getargs_b(IntSubclass()))
77 self.assertRaises(TypeError, getargs_b, BadInt())
78 with self.assertWarns(DeprecationWarning):
79 self.assertEqual(1, getargs_b(BadInt2()))
80 self.assertEqual(0, getargs_b(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +000081
Thomas Heller54aa5782003-04-24 16:15:29 +000082 self.assertRaises(OverflowError, getargs_b, -1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000083 self.assertEqual(0, getargs_b(0))
84 self.assertEqual(UCHAR_MAX, getargs_b(UCHAR_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +000085 self.assertRaises(OverflowError, getargs_b, UCHAR_MAX + 1)
Thomas Hellera4ea6032003-04-17 18:55:45 +000086
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000087 self.assertEqual(42, getargs_b(42))
Thomas Heller54aa5782003-04-24 16:15:29 +000088 self.assertRaises(OverflowError, getargs_b, VERY_LARGE)
Thomas Hellera4ea6032003-04-17 18:55:45 +000089
Thomas Hellera4ea6032003-04-17 18:55:45 +000090 def test_B(self):
Thomas Heller54aa5782003-04-24 16:15:29 +000091 from _testcapi import getargs_B
Thomas Hellera4ea6032003-04-17 18:55:45 +000092 # B returns 'unsigned char', no range checking
Thomas Woutersc9471232006-04-15 09:15:11 +000093 self.assertRaises(TypeError, getargs_B, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000094 self.assertEqual(99, getargs_B(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +020095 self.assertEqual(0, getargs_B(IntSubclass()))
96 self.assertRaises(TypeError, getargs_B, BadInt())
97 with self.assertWarns(DeprecationWarning):
98 self.assertEqual(1, getargs_B(BadInt2()))
99 self.assertEqual(0, getargs_B(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000100
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000101 self.assertEqual(UCHAR_MAX, getargs_B(-1))
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000102 self.assertEqual(0, getargs_B(0))
103 self.assertEqual(UCHAR_MAX, getargs_B(UCHAR_MAX))
104 self.assertEqual(0, getargs_B(UCHAR_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000105
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000106 self.assertEqual(42, getargs_B(42))
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000107 self.assertEqual(UCHAR_MAX & VERY_LARGE, getargs_B(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000108
109 def test_H(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000110 from _testcapi import getargs_H
Thomas Hellera4ea6032003-04-17 18:55:45 +0000111 # H returns 'unsigned short', no range checking
Thomas Woutersc9471232006-04-15 09:15:11 +0000112 self.assertRaises(TypeError, getargs_H, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000113 self.assertEqual(99, getargs_H(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200114 self.assertEqual(0, getargs_H(IntSubclass()))
115 self.assertRaises(TypeError, getargs_H, BadInt())
116 with self.assertWarns(DeprecationWarning):
117 self.assertEqual(1, getargs_H(BadInt2()))
118 self.assertEqual(0, getargs_H(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000119
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000120 self.assertEqual(USHRT_MAX, getargs_H(-1))
121 self.assertEqual(0, getargs_H(0))
122 self.assertEqual(USHRT_MAX, getargs_H(USHRT_MAX))
123 self.assertEqual(0, getargs_H(USHRT_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000124
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000125 self.assertEqual(42, getargs_H(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000126
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000127 self.assertEqual(VERY_LARGE & USHRT_MAX, getargs_H(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000128
129 def test_I(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000130 from _testcapi import getargs_I
Thomas Hellera4ea6032003-04-17 18:55:45 +0000131 # I returns 'unsigned int', no range checking
Thomas Woutersc9471232006-04-15 09:15:11 +0000132 self.assertRaises(TypeError, getargs_I, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000133 self.assertEqual(99, getargs_I(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200134 self.assertEqual(0, getargs_I(IntSubclass()))
135 self.assertRaises(TypeError, getargs_I, BadInt())
136 with self.assertWarns(DeprecationWarning):
137 self.assertEqual(1, getargs_I(BadInt2()))
138 self.assertEqual(0, getargs_I(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000139
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000140 self.assertEqual(UINT_MAX, getargs_I(-1))
141 self.assertEqual(0, getargs_I(0))
142 self.assertEqual(UINT_MAX, getargs_I(UINT_MAX))
143 self.assertEqual(0, getargs_I(UINT_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000144
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000145 self.assertEqual(42, getargs_I(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000146
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000147 self.assertEqual(VERY_LARGE & UINT_MAX, getargs_I(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000148
149 def test_k(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000150 from _testcapi import getargs_k
Thomas Hellera4ea6032003-04-17 18:55:45 +0000151 # k returns 'unsigned long', no range checking
152 # it does not accept float, or instances with __int__
Thomas Heller54aa5782003-04-24 16:15:29 +0000153 self.assertRaises(TypeError, getargs_k, 3.14)
Thomas Heller54aa5782003-04-24 16:15:29 +0000154 self.assertRaises(TypeError, getargs_k, Int())
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200155 self.assertEqual(0, getargs_k(IntSubclass()))
156 self.assertRaises(TypeError, getargs_k, BadInt())
157 self.assertRaises(TypeError, getargs_k, BadInt2())
158 self.assertEqual(0, getargs_k(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000159
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000160 self.assertEqual(ULONG_MAX, getargs_k(-1))
161 self.assertEqual(0, getargs_k(0))
162 self.assertEqual(ULONG_MAX, getargs_k(ULONG_MAX))
163 self.assertEqual(0, getargs_k(ULONG_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000164
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000165 self.assertEqual(42, getargs_k(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000166
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000167 self.assertEqual(VERY_LARGE & ULONG_MAX, getargs_k(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000168
169class Signed_TestCase(unittest.TestCase):
Mark Dickinson1554b182009-12-20 16:03:30 +0000170 def test_h(self):
171 from _testcapi import getargs_h
172 # h returns 'short', and does range checking (SHRT_MIN ... SHRT_MAX)
173 self.assertRaises(TypeError, getargs_h, 3.14)
174 self.assertEqual(99, getargs_h(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200175 self.assertEqual(0, getargs_h(IntSubclass()))
176 self.assertRaises(TypeError, getargs_h, BadInt())
177 with self.assertWarns(DeprecationWarning):
178 self.assertEqual(1, getargs_h(BadInt2()))
179 self.assertEqual(0, getargs_h(BadInt3()))
Mark Dickinson1554b182009-12-20 16:03:30 +0000180
181 self.assertRaises(OverflowError, getargs_h, SHRT_MIN-1)
182 self.assertEqual(SHRT_MIN, getargs_h(SHRT_MIN))
183 self.assertEqual(SHRT_MAX, getargs_h(SHRT_MAX))
184 self.assertRaises(OverflowError, getargs_h, SHRT_MAX+1)
185
186 self.assertEqual(42, getargs_h(42))
187 self.assertRaises(OverflowError, getargs_h, VERY_LARGE)
188
Thomas Hellera4ea6032003-04-17 18:55:45 +0000189 def test_i(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000190 from _testcapi import getargs_i
Thomas Hellera4ea6032003-04-17 18:55:45 +0000191 # i returns 'int', and does range checking (INT_MIN ... INT_MAX)
Thomas Woutersc9471232006-04-15 09:15:11 +0000192 self.assertRaises(TypeError, getargs_i, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000193 self.assertEqual(99, getargs_i(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200194 self.assertEqual(0, getargs_i(IntSubclass()))
195 self.assertRaises(TypeError, getargs_i, BadInt())
196 with self.assertWarns(DeprecationWarning):
197 self.assertEqual(1, getargs_i(BadInt2()))
198 self.assertEqual(0, getargs_i(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000199
Thomas Heller54aa5782003-04-24 16:15:29 +0000200 self.assertRaises(OverflowError, getargs_i, INT_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000201 self.assertEqual(INT_MIN, getargs_i(INT_MIN))
202 self.assertEqual(INT_MAX, getargs_i(INT_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +0000203 self.assertRaises(OverflowError, getargs_i, INT_MAX+1)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000204
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000205 self.assertEqual(42, getargs_i(42))
Thomas Heller54aa5782003-04-24 16:15:29 +0000206 self.assertRaises(OverflowError, getargs_i, VERY_LARGE)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000207
208 def test_l(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000209 from _testcapi import getargs_l
Thomas Hellera4ea6032003-04-17 18:55:45 +0000210 # l returns 'long', and does range checking (LONG_MIN ... LONG_MAX)
Thomas Woutersc9471232006-04-15 09:15:11 +0000211 self.assertRaises(TypeError, getargs_l, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000212 self.assertEqual(99, getargs_l(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200213 self.assertEqual(0, getargs_l(IntSubclass()))
214 self.assertRaises(TypeError, getargs_l, BadInt())
215 with self.assertWarns(DeprecationWarning):
216 self.assertEqual(1, getargs_l(BadInt2()))
217 self.assertEqual(0, getargs_l(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000218
Thomas Heller54aa5782003-04-24 16:15:29 +0000219 self.assertRaises(OverflowError, getargs_l, LONG_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000220 self.assertEqual(LONG_MIN, getargs_l(LONG_MIN))
221 self.assertEqual(LONG_MAX, getargs_l(LONG_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +0000222 self.assertRaises(OverflowError, getargs_l, LONG_MAX+1)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000223
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000224 self.assertEqual(42, getargs_l(42))
Thomas Heller54aa5782003-04-24 16:15:29 +0000225 self.assertRaises(OverflowError, getargs_l, VERY_LARGE)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000226
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000227 def test_n(self):
228 from _testcapi import getargs_n
229 # n returns 'Py_ssize_t', and does range checking
230 # (PY_SSIZE_T_MIN ... PY_SSIZE_T_MAX)
Thomas Woutersd8073282006-04-21 11:36:13 +0000231 self.assertRaises(TypeError, getargs_n, 3.14)
Trent Nelson35133582008-04-22 19:02:40 +0000232 self.assertRaises(TypeError, getargs_n, Int())
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200233 self.assertEqual(0, getargs_n(IntSubclass()))
234 self.assertRaises(TypeError, getargs_n, BadInt())
235 self.assertRaises(TypeError, getargs_n, BadInt2())
236 self.assertEqual(0, getargs_n(BadInt3()))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000237
238 self.assertRaises(OverflowError, getargs_n, PY_SSIZE_T_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000239 self.assertEqual(PY_SSIZE_T_MIN, getargs_n(PY_SSIZE_T_MIN))
240 self.assertEqual(PY_SSIZE_T_MAX, getargs_n(PY_SSIZE_T_MAX))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000241 self.assertRaises(OverflowError, getargs_n, PY_SSIZE_T_MAX+1)
242
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000243 self.assertEqual(42, getargs_n(42))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000244 self.assertRaises(OverflowError, getargs_n, VERY_LARGE)
245
Thomas Hellera4ea6032003-04-17 18:55:45 +0000246
247class LongLong_TestCase(unittest.TestCase):
248 def test_L(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000249 from _testcapi import getargs_L
Mark Dickinsonde604012010-01-01 19:27:32 +0000250 # L returns 'long long', and does range checking (LLONG_MIN
251 # ... LLONG_MAX)
Mark Dickinsonc7301312010-06-10 16:05:10 +0000252 self.assertRaises(TypeError, getargs_L, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000253 self.assertRaises(TypeError, getargs_L, "Hello")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000254 self.assertEqual(99, getargs_L(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200255 self.assertEqual(0, getargs_L(IntSubclass()))
256 self.assertRaises(TypeError, getargs_L, BadInt())
257 with self.assertWarns(DeprecationWarning):
258 self.assertEqual(1, getargs_L(BadInt2()))
259 self.assertEqual(0, getargs_L(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000260
Thomas Heller54aa5782003-04-24 16:15:29 +0000261 self.assertRaises(OverflowError, getargs_L, LLONG_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000262 self.assertEqual(LLONG_MIN, getargs_L(LLONG_MIN))
263 self.assertEqual(LLONG_MAX, getargs_L(LLONG_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +0000264 self.assertRaises(OverflowError, getargs_L, LLONG_MAX+1)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000265
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000266 self.assertEqual(42, getargs_L(42))
Thomas Heller54aa5782003-04-24 16:15:29 +0000267 self.assertRaises(OverflowError, getargs_L, VERY_LARGE)
Tim Peters0eadaac2003-04-24 16:02:54 +0000268
Thomas Hellera4ea6032003-04-17 18:55:45 +0000269 def test_K(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000270 from _testcapi import getargs_K
Thomas Hellera4ea6032003-04-17 18:55:45 +0000271 # K return 'unsigned long long', no range checking
Thomas Heller54aa5782003-04-24 16:15:29 +0000272 self.assertRaises(TypeError, getargs_K, 3.14)
Thomas Heller54aa5782003-04-24 16:15:29 +0000273 self.assertRaises(TypeError, getargs_K, Int())
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200274 self.assertEqual(0, getargs_K(IntSubclass()))
275 self.assertRaises(TypeError, getargs_K, BadInt())
276 self.assertRaises(TypeError, getargs_K, BadInt2())
277 self.assertEqual(0, getargs_K(BadInt3()))
278
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000279 self.assertEqual(ULLONG_MAX, getargs_K(ULLONG_MAX))
280 self.assertEqual(0, getargs_K(0))
281 self.assertEqual(0, getargs_K(ULLONG_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000282
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000283 self.assertEqual(42, getargs_K(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000284
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000285 self.assertEqual(VERY_LARGE & ULLONG_MAX, getargs_K(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000286
Larry Hastingsfaf91e72012-05-05 16:54:29 -0700287class Paradox:
288 "This statement is false."
289 def __bool__(self):
290 raise NotImplementedError
291
292class Boolean_TestCase(unittest.TestCase):
293 def test_p(self):
294 from _testcapi import getargs_p
295 self.assertEqual(0, getargs_p(False))
296 self.assertEqual(0, getargs_p(None))
297 self.assertEqual(0, getargs_p(0))
298 self.assertEqual(0, getargs_p(0.0))
299 self.assertEqual(0, getargs_p(0j))
300 self.assertEqual(0, getargs_p(''))
301 self.assertEqual(0, getargs_p(()))
302 self.assertEqual(0, getargs_p([]))
303 self.assertEqual(0, getargs_p({}))
304
305 self.assertEqual(1, getargs_p(True))
306 self.assertEqual(1, getargs_p(1))
307 self.assertEqual(1, getargs_p(1.0))
308 self.assertEqual(1, getargs_p(1j))
309 self.assertEqual(1, getargs_p('x'))
310 self.assertEqual(1, getargs_p((1,)))
311 self.assertEqual(1, getargs_p([1]))
312 self.assertEqual(1, getargs_p({1:2}))
313 self.assertEqual(1, getargs_p(unittest.TestCase))
314
315 self.assertRaises(NotImplementedError, getargs_p, Paradox())
316
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000317
318class Tuple_TestCase(unittest.TestCase):
319 def test_tuple(self):
320 from _testcapi import getargs_tuple
321
322 ret = getargs_tuple(1, (2, 3))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000323 self.assertEqual(ret, (1,2,3))
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000324
325 # make sure invalid tuple arguments are handled correctly
326 class seq:
327 def __len__(self):
328 return 2
329 def __getitem__(self, n):
330 raise ValueError
331 self.assertRaises(TypeError, getargs_tuple, 1, seq())
332
Christian Heimes380f7f22008-02-28 11:19:05 +0000333class Keywords_TestCase(unittest.TestCase):
334 def test_positional_args(self):
335 # using all positional args
Ezio Melottib3aedd42010-11-20 19:04:17 +0000336 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000337 getargs_keywords((1,2), 3, (4,(5,6)), (7,8,9), 10),
338 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
339 )
Victor Stinner93b55132010-05-19 00:54:06 +0000340
Christian Heimes380f7f22008-02-28 11:19:05 +0000341 def test_mixed_args(self):
342 # positional and keyword args
Ezio Melottib3aedd42010-11-20 19:04:17 +0000343 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000344 getargs_keywords((1,2), 3, (4,(5,6)), arg4=(7,8,9), arg5=10),
345 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
346 )
Victor Stinner93b55132010-05-19 00:54:06 +0000347
Christian Heimes380f7f22008-02-28 11:19:05 +0000348 def test_keyword_args(self):
349 # all keywords
Ezio Melottib3aedd42010-11-20 19:04:17 +0000350 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000351 getargs_keywords(arg1=(1,2), arg2=3, arg3=(4,(5,6)), arg4=(7,8,9), arg5=10),
352 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
353 )
Victor Stinner93b55132010-05-19 00:54:06 +0000354
Christian Heimes380f7f22008-02-28 11:19:05 +0000355 def test_optional_args(self):
356 # missing optional keyword args, skipping tuples
Ezio Melottib3aedd42010-11-20 19:04:17 +0000357 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000358 getargs_keywords(arg1=(1,2), arg2=3, arg5=10),
359 (1, 2, 3, -1, -1, -1, -1, -1, -1, 10)
360 )
Victor Stinner93b55132010-05-19 00:54:06 +0000361
Christian Heimes380f7f22008-02-28 11:19:05 +0000362 def test_required_args(self):
363 # required arg missing
364 try:
365 getargs_keywords(arg1=(1,2))
366 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000367 self.assertEqual(str(err), "Required argument 'arg2' (pos 2) not found")
Christian Heimes380f7f22008-02-28 11:19:05 +0000368 else:
369 self.fail('TypeError should have been raised')
Victor Stinner93b55132010-05-19 00:54:06 +0000370
Christian Heimes380f7f22008-02-28 11:19:05 +0000371 def test_too_many_args(self):
372 try:
373 getargs_keywords((1,2),3,(4,(5,6)),(7,8,9),10,111)
374 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000375 self.assertEqual(str(err), "function takes at most 5 arguments (6 given)")
Christian Heimes380f7f22008-02-28 11:19:05 +0000376 else:
377 self.fail('TypeError should have been raised')
Victor Stinner93b55132010-05-19 00:54:06 +0000378
Christian Heimes380f7f22008-02-28 11:19:05 +0000379 def test_invalid_keyword(self):
380 # extraneous keyword arg
381 try:
382 getargs_keywords((1,2),3,arg5=10,arg666=666)
383 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000384 self.assertEqual(str(err), "'arg666' is an invalid keyword argument for this function")
Christian Heimes380f7f22008-02-28 11:19:05 +0000385 else:
386 self.fail('TypeError should have been raised')
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000387
Victor Stinner93b55132010-05-19 00:54:06 +0000388 def test_surrogate_keyword(self):
389 try:
390 getargs_keywords((1,2), 3, (4,(5,6)), (7,8,9), **{'\uDC80': 10})
391 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000392 self.assertEqual(str(err), "'\udc80' is an invalid keyword argument for this function")
Victor Stinner93b55132010-05-19 00:54:06 +0000393 else:
394 self.fail('TypeError should have been raised')
395
Larry Hastings83a9f482012-03-20 20:06:16 +0000396class KeywordOnly_TestCase(unittest.TestCase):
397 def test_positional_args(self):
398 # using all possible positional args
399 self.assertEqual(
400 getargs_keyword_only(1, 2),
401 (1, 2, -1)
402 )
403
404 def test_mixed_args(self):
405 # positional and keyword args
406 self.assertEqual(
407 getargs_keyword_only(1, 2, keyword_only=3),
408 (1, 2, 3)
409 )
410
411 def test_keyword_args(self):
412 # all keywords
413 self.assertEqual(
414 getargs_keyword_only(required=1, optional=2, keyword_only=3),
415 (1, 2, 3)
416 )
417
418 def test_optional_args(self):
419 # missing optional keyword args, skipping tuples
420 self.assertEqual(
421 getargs_keyword_only(required=1, optional=2),
422 (1, 2, -1)
423 )
424 self.assertEqual(
425 getargs_keyword_only(required=1, keyword_only=3),
426 (1, -1, 3)
427 )
428
429 def test_required_args(self):
430 self.assertEqual(
431 getargs_keyword_only(1),
432 (1, -1, -1)
433 )
434 self.assertEqual(
435 getargs_keyword_only(required=1),
436 (1, -1, -1)
437 )
438 # required arg missing
439 with self.assertRaisesRegex(TypeError,
440 "Required argument 'required' \(pos 1\) not found"):
441 getargs_keyword_only(optional=2)
442
443 with self.assertRaisesRegex(TypeError,
444 "Required argument 'required' \(pos 1\) not found"):
445 getargs_keyword_only(keyword_only=3)
446
447 def test_too_many_args(self):
448 with self.assertRaisesRegex(TypeError,
449 "Function takes at most 2 positional arguments \(3 given\)"):
450 getargs_keyword_only(1, 2, 3)
451
452 with self.assertRaisesRegex(TypeError,
453 "function takes at most 3 arguments \(4 given\)"):
454 getargs_keyword_only(1, 2, 3, keyword_only=5)
455
456 def test_invalid_keyword(self):
457 # extraneous keyword arg
458 with self.assertRaisesRegex(TypeError,
459 "'monster' is an invalid keyword argument for this function"):
460 getargs_keyword_only(1, 2, monster=666)
461
462 def test_surrogate_keyword(self):
463 with self.assertRaisesRegex(TypeError,
464 "'\udc80' is an invalid keyword argument for this function"):
465 getargs_keyword_only(1, 2, **{'\uDC80': 10})
466
Victor Stinner06e49dd2010-06-13 18:21:50 +0000467class Bytes_TestCase(unittest.TestCase):
Eli Bendersky906b88f2011-07-29 07:05:08 +0300468 def test_c(self):
469 from _testcapi import getargs_c
470 self.assertRaises(TypeError, getargs_c, b'abc') # len > 1
471 self.assertEqual(getargs_c(b'a'), b'a')
472 self.assertEqual(getargs_c(bytearray(b'a')), b'a')
473 self.assertRaises(TypeError, getargs_c, memoryview(b'a'))
474 self.assertRaises(TypeError, getargs_c, 's')
475 self.assertRaises(TypeError, getargs_c, None)
476
Victor Stinner06e49dd2010-06-13 18:21:50 +0000477 def test_s(self):
478 from _testcapi import getargs_s
479 self.assertEqual(getargs_s('abc\xe9'), b'abc\xc3\xa9')
480 self.assertRaises(TypeError, getargs_s, 'nul:\0')
481 self.assertRaises(TypeError, getargs_s, b'bytes')
482 self.assertRaises(TypeError, getargs_s, bytearray(b'bytearray'))
483 self.assertRaises(TypeError, getargs_s, memoryview(b'memoryview'))
484 self.assertRaises(TypeError, getargs_s, None)
485
486 def test_s_star(self):
487 from _testcapi import getargs_s_star
488 self.assertEqual(getargs_s_star('abc\xe9'), b'abc\xc3\xa9')
489 self.assertEqual(getargs_s_star('nul:\0'), b'nul:\0')
490 self.assertEqual(getargs_s_star(b'bytes'), b'bytes')
491 self.assertEqual(getargs_s_star(bytearray(b'bytearray')), b'bytearray')
492 self.assertEqual(getargs_s_star(memoryview(b'memoryview')), b'memoryview')
493 self.assertRaises(TypeError, getargs_s_star, None)
494
495 def test_s_hash(self):
496 from _testcapi import getargs_s_hash
497 self.assertEqual(getargs_s_hash('abc\xe9'), b'abc\xc3\xa9')
498 self.assertEqual(getargs_s_hash('nul:\0'), b'nul:\0')
499 self.assertEqual(getargs_s_hash(b'bytes'), b'bytes')
500 self.assertRaises(TypeError, getargs_s_hash, bytearray(b'bytearray'))
501 self.assertRaises(TypeError, getargs_s_hash, memoryview(b'memoryview'))
502 self.assertRaises(TypeError, getargs_s_hash, None)
503
504 def test_z(self):
505 from _testcapi import getargs_z
506 self.assertEqual(getargs_z('abc\xe9'), b'abc\xc3\xa9')
507 self.assertRaises(TypeError, getargs_z, 'nul:\0')
Victor Stinner4aae1eb2010-06-24 22:08:25 +0000508 self.assertRaises(TypeError, getargs_z, b'bytes')
Victor Stinner06e49dd2010-06-13 18:21:50 +0000509 self.assertRaises(TypeError, getargs_z, bytearray(b'bytearray'))
510 self.assertRaises(TypeError, getargs_z, memoryview(b'memoryview'))
511 self.assertIsNone(getargs_z(None))
512
513 def test_z_star(self):
514 from _testcapi import getargs_z_star
515 self.assertEqual(getargs_z_star('abc\xe9'), b'abc\xc3\xa9')
516 self.assertEqual(getargs_z_star('nul:\0'), b'nul:\0')
517 self.assertEqual(getargs_z_star(b'bytes'), b'bytes')
518 self.assertEqual(getargs_z_star(bytearray(b'bytearray')), b'bytearray')
519 self.assertEqual(getargs_z_star(memoryview(b'memoryview')), b'memoryview')
520 self.assertIsNone(getargs_z_star(None))
521
522 def test_z_hash(self):
523 from _testcapi import getargs_z_hash
524 self.assertEqual(getargs_z_hash('abc\xe9'), b'abc\xc3\xa9')
525 self.assertEqual(getargs_z_hash('nul:\0'), b'nul:\0')
526 self.assertEqual(getargs_z_hash(b'bytes'), b'bytes')
527 self.assertRaises(TypeError, getargs_z_hash, bytearray(b'bytearray'))
528 self.assertRaises(TypeError, getargs_z_hash, memoryview(b'memoryview'))
529 self.assertIsNone(getargs_z_hash(None))
530
531 def test_y(self):
532 from _testcapi import getargs_y
533 self.assertRaises(TypeError, getargs_y, 'abc\xe9')
534 self.assertEqual(getargs_y(b'bytes'), b'bytes')
535 self.assertRaises(TypeError, getargs_y, b'nul:\0')
536 self.assertRaises(TypeError, getargs_y, bytearray(b'bytearray'))
537 self.assertRaises(TypeError, getargs_y, memoryview(b'memoryview'))
538 self.assertRaises(TypeError, getargs_y, None)
539
540 def test_y_star(self):
541 from _testcapi import getargs_y_star
542 self.assertRaises(TypeError, getargs_y_star, 'abc\xe9')
543 self.assertEqual(getargs_y_star(b'bytes'), b'bytes')
544 self.assertEqual(getargs_y_star(b'nul:\0'), b'nul:\0')
545 self.assertEqual(getargs_y_star(bytearray(b'bytearray')), b'bytearray')
546 self.assertEqual(getargs_y_star(memoryview(b'memoryview')), b'memoryview')
547 self.assertRaises(TypeError, getargs_y_star, None)
548
549 def test_y_hash(self):
550 from _testcapi import getargs_y_hash
551 self.assertRaises(TypeError, getargs_y_hash, 'abc\xe9')
552 self.assertEqual(getargs_y_hash(b'bytes'), b'bytes')
553 self.assertEqual(getargs_y_hash(b'nul:\0'), b'nul:\0')
554 self.assertRaises(TypeError, getargs_y_hash, bytearray(b'bytearray'))
555 self.assertRaises(TypeError, getargs_y_hash, memoryview(b'memoryview'))
556 self.assertRaises(TypeError, getargs_y_hash, None)
557
Victor Stinner25e8ec42010-06-25 00:02:38 +0000558 def test_w_star(self):
559 # getargs_w_star() modifies first and last byte
560 from _testcapi import getargs_w_star
561 self.assertRaises(TypeError, getargs_w_star, 'abc\xe9')
562 self.assertRaises(TypeError, getargs_w_star, b'bytes')
563 self.assertRaises(TypeError, getargs_w_star, b'nul:\0')
Antoine Pitrouad62b032011-01-18 18:57:52 +0000564 self.assertRaises(TypeError, getargs_w_star, memoryview(b'bytes'))
Victor Stinner25e8ec42010-06-25 00:02:38 +0000565 self.assertEqual(getargs_w_star(bytearray(b'bytearray')), b'[ytearra]')
Antoine Pitrouad62b032011-01-18 18:57:52 +0000566 self.assertEqual(getargs_w_star(memoryview(bytearray(b'memoryview'))),
567 b'[emoryvie]')
Victor Stinner25e8ec42010-06-25 00:02:38 +0000568 self.assertRaises(TypeError, getargs_w_star, None)
569
Victor Stinner06e49dd2010-06-13 18:21:50 +0000570
571class Unicode_TestCase(unittest.TestCase):
572 def test_u(self):
573 from _testcapi import getargs_u
574 self.assertEqual(getargs_u('abc\xe9'), 'abc\xe9')
575 self.assertRaises(TypeError, getargs_u, 'nul:\0')
576 self.assertRaises(TypeError, getargs_u, b'bytes')
577 self.assertRaises(TypeError, getargs_u, bytearray(b'bytearray'))
578 self.assertRaises(TypeError, getargs_u, memoryview(b'memoryview'))
579 self.assertRaises(TypeError, getargs_u, None)
580
581 def test_u_hash(self):
582 from _testcapi import getargs_u_hash
583 self.assertEqual(getargs_u_hash('abc\xe9'), 'abc\xe9')
584 self.assertEqual(getargs_u_hash('nul:\0'), 'nul:\0')
585 self.assertRaises(TypeError, getargs_u_hash, b'bytes')
586 self.assertRaises(TypeError, getargs_u_hash, bytearray(b'bytearray'))
587 self.assertRaises(TypeError, getargs_u_hash, memoryview(b'memoryview'))
588 self.assertRaises(TypeError, getargs_u_hash, None)
589
590 def test_Z(self):
591 from _testcapi import getargs_Z
592 self.assertEqual(getargs_Z('abc\xe9'), 'abc\xe9')
593 self.assertRaises(TypeError, getargs_Z, 'nul:\0')
594 self.assertRaises(TypeError, getargs_Z, b'bytes')
595 self.assertRaises(TypeError, getargs_Z, bytearray(b'bytearray'))
596 self.assertRaises(TypeError, getargs_Z, memoryview(b'memoryview'))
597 self.assertIsNone(getargs_Z(None))
598
599 def test_Z_hash(self):
600 from _testcapi import getargs_Z_hash
601 self.assertEqual(getargs_Z_hash('abc\xe9'), 'abc\xe9')
602 self.assertEqual(getargs_Z_hash('nul:\0'), 'nul:\0')
603 self.assertRaises(TypeError, getargs_Z_hash, b'bytes')
604 self.assertRaises(TypeError, getargs_Z_hash, bytearray(b'bytearray'))
605 self.assertRaises(TypeError, getargs_Z_hash, memoryview(b'memoryview'))
606 self.assertIsNone(getargs_Z_hash(None))
607
608
Thomas Hellera4ea6032003-04-17 18:55:45 +0000609def test_main():
Victor Stinner06e49dd2010-06-13 18:21:50 +0000610 tests = [
611 Signed_TestCase,
612 Unsigned_TestCase,
Larry Hastingsfaf91e72012-05-05 16:54:29 -0700613 Boolean_TestCase,
Victor Stinner06e49dd2010-06-13 18:21:50 +0000614 Tuple_TestCase,
615 Keywords_TestCase,
Larry Hastings83a9f482012-03-20 20:06:16 +0000616 KeywordOnly_TestCase,
Victor Stinner06e49dd2010-06-13 18:21:50 +0000617 Bytes_TestCase,
618 Unicode_TestCase,
619 ]
Thomas Hellera4ea6032003-04-17 18:55:45 +0000620 try:
Thomas Heller54aa5782003-04-24 16:15:29 +0000621 from _testcapi import getargs_L, getargs_K
622 except ImportError:
Thomas Hellera4ea6032003-04-17 18:55:45 +0000623 pass # PY_LONG_LONG not available
624 else:
Walter Dörwald21d3a322003-05-01 17:45:56 +0000625 tests.append(LongLong_TestCase)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000626 support.run_unittest(*tests)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000627
628if __name__ == "__main__":
629 test_main()