blob: fe64ee1669249d33e3fc6756dbf217bffe4c0441 [file] [log] [blame]
Thomas Hellera4ea6032003-04-17 18:55:45 +00001import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +00002from test import support
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +02003# Skip this test if the _testcapi module isn't available.
4support.import_module('_testcapi')
Larry Hastings83a9f482012-03-20 20:06:16 +00005from _testcapi import getargs_keywords, getargs_keyword_only
Thomas Hellera4ea6032003-04-17 18:55:45 +00006
7"""
8> How about the following counterproposal. This also changes some of
9> the other format codes to be a little more regular.
10>
11> Code C type Range check
12>
13> b unsigned char 0..UCHAR_MAX
Guido van Rossumbbb931b2003-04-18 00:13:53 +000014> h signed short SHRT_MIN..SHRT_MAX
Thomas Hellera4ea6032003-04-17 18:55:45 +000015> B unsigned char none **
16> H unsigned short none **
17> k * unsigned long none
18> I * unsigned int 0..UINT_MAX
19
20
21> i int INT_MIN..INT_MAX
22> l long LONG_MIN..LONG_MAX
23
24> K * unsigned long long none
25> L long long LLONG_MIN..LLONG_MAX
26
27> Notes:
28>
29> * New format codes.
30>
31> ** Changed from previous "range-and-a-half" to "none"; the
32> range-and-a-half checking wasn't particularly useful.
33
34Plus a C API or two, e.g. PyInt_AsLongMask() ->
35unsigned long and PyInt_AsLongLongMask() -> unsigned
36long long (if that exists).
37"""
38
39LARGE = 0x7FFFFFFF
Guido van Rossume2a383d2007-01-15 16:59:06 +000040VERY_LARGE = 0xFF0000121212121212121242
Thomas Hellera4ea6032003-04-17 18:55:45 +000041
42from _testcapi import UCHAR_MAX, USHRT_MAX, UINT_MAX, ULONG_MAX, INT_MAX, \
Mark Dickinson1554b182009-12-20 16:03:30 +000043 INT_MIN, LONG_MIN, LONG_MAX, PY_SSIZE_T_MIN, PY_SSIZE_T_MAX, \
44 SHRT_MIN, SHRT_MAX
Thomas Hellera4ea6032003-04-17 18:55:45 +000045
Thomas Hellera4ea6032003-04-17 18:55:45 +000046# fake, they are not defined in Python's header files
47LLONG_MAX = 2**63-1
48LLONG_MIN = -2**63
49ULLONG_MAX = 2**64-1
50
Thomas Hellera4ea6032003-04-17 18:55:45 +000051class Int:
52 def __int__(self):
53 return 99
54
Serhiy Storchaka31a65542013-12-11 21:07:54 +020055class IntSubclass(int):
56 def __int__(self):
57 return 99
58
59class BadInt:
60 def __int__(self):
61 return 1.0
62
63class BadInt2:
64 def __int__(self):
65 return True
66
67class BadInt3(int):
68 def __int__(self):
69 return True
70
71
Thomas Hellera4ea6032003-04-17 18:55:45 +000072class Unsigned_TestCase(unittest.TestCase):
73 def test_b(self):
Thomas Heller54aa5782003-04-24 16:15:29 +000074 from _testcapi import getargs_b
Thomas Hellera4ea6032003-04-17 18:55:45 +000075 # b returns 'unsigned char', and does range checking (0 ... UCHAR_MAX)
Thomas Woutersc9471232006-04-15 09:15:11 +000076 self.assertRaises(TypeError, getargs_b, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000077 self.assertEqual(99, getargs_b(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +020078 self.assertEqual(0, getargs_b(IntSubclass()))
79 self.assertRaises(TypeError, getargs_b, BadInt())
Serhiy Storchakac836a282013-12-14 21:07:09 +020080 self.assertEqual(1, getargs_b(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +020081 self.assertEqual(0, getargs_b(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +000082
Thomas Heller54aa5782003-04-24 16:15:29 +000083 self.assertRaises(OverflowError, getargs_b, -1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000084 self.assertEqual(0, getargs_b(0))
85 self.assertEqual(UCHAR_MAX, getargs_b(UCHAR_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +000086 self.assertRaises(OverflowError, getargs_b, UCHAR_MAX + 1)
Thomas Hellera4ea6032003-04-17 18:55:45 +000087
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000088 self.assertEqual(42, getargs_b(42))
Thomas Heller54aa5782003-04-24 16:15:29 +000089 self.assertRaises(OverflowError, getargs_b, VERY_LARGE)
Thomas Hellera4ea6032003-04-17 18:55:45 +000090
Thomas Hellera4ea6032003-04-17 18:55:45 +000091 def test_B(self):
Thomas Heller54aa5782003-04-24 16:15:29 +000092 from _testcapi import getargs_B
Thomas Hellera4ea6032003-04-17 18:55:45 +000093 # B returns 'unsigned char', no range checking
Thomas Woutersc9471232006-04-15 09:15:11 +000094 self.assertRaises(TypeError, getargs_B, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000095 self.assertEqual(99, getargs_B(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +020096 self.assertEqual(0, getargs_B(IntSubclass()))
97 self.assertRaises(TypeError, getargs_B, BadInt())
Serhiy Storchakac836a282013-12-14 21:07:09 +020098 self.assertEqual(1, getargs_B(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +020099 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())
Serhiy Storchakac836a282013-12-14 21:07:09 +0200116 self.assertEqual(1, getargs_H(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200117 self.assertEqual(0, getargs_H(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000118
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000119 self.assertEqual(USHRT_MAX, getargs_H(-1))
120 self.assertEqual(0, getargs_H(0))
121 self.assertEqual(USHRT_MAX, getargs_H(USHRT_MAX))
122 self.assertEqual(0, getargs_H(USHRT_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000123
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000124 self.assertEqual(42, getargs_H(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000125
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000126 self.assertEqual(VERY_LARGE & USHRT_MAX, getargs_H(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000127
128 def test_I(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000129 from _testcapi import getargs_I
Thomas Hellera4ea6032003-04-17 18:55:45 +0000130 # I returns 'unsigned int', no range checking
Thomas Woutersc9471232006-04-15 09:15:11 +0000131 self.assertRaises(TypeError, getargs_I, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000132 self.assertEqual(99, getargs_I(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200133 self.assertEqual(0, getargs_I(IntSubclass()))
134 self.assertRaises(TypeError, getargs_I, BadInt())
Serhiy Storchakac836a282013-12-14 21:07:09 +0200135 self.assertEqual(1, getargs_I(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200136 self.assertEqual(0, getargs_I(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000137
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000138 self.assertEqual(UINT_MAX, getargs_I(-1))
139 self.assertEqual(0, getargs_I(0))
140 self.assertEqual(UINT_MAX, getargs_I(UINT_MAX))
141 self.assertEqual(0, getargs_I(UINT_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000142
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000143 self.assertEqual(42, getargs_I(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000144
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000145 self.assertEqual(VERY_LARGE & UINT_MAX, getargs_I(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000146
147 def test_k(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000148 from _testcapi import getargs_k
Thomas Hellera4ea6032003-04-17 18:55:45 +0000149 # k returns 'unsigned long', no range checking
150 # it does not accept float, or instances with __int__
Thomas Heller54aa5782003-04-24 16:15:29 +0000151 self.assertRaises(TypeError, getargs_k, 3.14)
Thomas Heller54aa5782003-04-24 16:15:29 +0000152 self.assertRaises(TypeError, getargs_k, Int())
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200153 self.assertEqual(0, getargs_k(IntSubclass()))
154 self.assertRaises(TypeError, getargs_k, BadInt())
155 self.assertRaises(TypeError, getargs_k, BadInt2())
156 self.assertEqual(0, getargs_k(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000157
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000158 self.assertEqual(ULONG_MAX, getargs_k(-1))
159 self.assertEqual(0, getargs_k(0))
160 self.assertEqual(ULONG_MAX, getargs_k(ULONG_MAX))
161 self.assertEqual(0, getargs_k(ULONG_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000162
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000163 self.assertEqual(42, getargs_k(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000164
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000165 self.assertEqual(VERY_LARGE & ULONG_MAX, getargs_k(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000166
167class Signed_TestCase(unittest.TestCase):
Mark Dickinson1554b182009-12-20 16:03:30 +0000168 def test_h(self):
169 from _testcapi import getargs_h
170 # h returns 'short', and does range checking (SHRT_MIN ... SHRT_MAX)
171 self.assertRaises(TypeError, getargs_h, 3.14)
172 self.assertEqual(99, getargs_h(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200173 self.assertEqual(0, getargs_h(IntSubclass()))
174 self.assertRaises(TypeError, getargs_h, BadInt())
Serhiy Storchakac836a282013-12-14 21:07:09 +0200175 self.assertEqual(1, getargs_h(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200176 self.assertEqual(0, getargs_h(BadInt3()))
Mark Dickinson1554b182009-12-20 16:03:30 +0000177
178 self.assertRaises(OverflowError, getargs_h, SHRT_MIN-1)
179 self.assertEqual(SHRT_MIN, getargs_h(SHRT_MIN))
180 self.assertEqual(SHRT_MAX, getargs_h(SHRT_MAX))
181 self.assertRaises(OverflowError, getargs_h, SHRT_MAX+1)
182
183 self.assertEqual(42, getargs_h(42))
184 self.assertRaises(OverflowError, getargs_h, VERY_LARGE)
185
Thomas Hellera4ea6032003-04-17 18:55:45 +0000186 def test_i(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000187 from _testcapi import getargs_i
Thomas Hellera4ea6032003-04-17 18:55:45 +0000188 # i returns 'int', and does range checking (INT_MIN ... INT_MAX)
Thomas Woutersc9471232006-04-15 09:15:11 +0000189 self.assertRaises(TypeError, getargs_i, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000190 self.assertEqual(99, getargs_i(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200191 self.assertEqual(0, getargs_i(IntSubclass()))
192 self.assertRaises(TypeError, getargs_i, BadInt())
Serhiy Storchakac836a282013-12-14 21:07:09 +0200193 self.assertEqual(1, getargs_i(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200194 self.assertEqual(0, getargs_i(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000195
Thomas Heller54aa5782003-04-24 16:15:29 +0000196 self.assertRaises(OverflowError, getargs_i, INT_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000197 self.assertEqual(INT_MIN, getargs_i(INT_MIN))
198 self.assertEqual(INT_MAX, getargs_i(INT_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +0000199 self.assertRaises(OverflowError, getargs_i, INT_MAX+1)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000200
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000201 self.assertEqual(42, getargs_i(42))
Thomas Heller54aa5782003-04-24 16:15:29 +0000202 self.assertRaises(OverflowError, getargs_i, VERY_LARGE)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000203
204 def test_l(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000205 from _testcapi import getargs_l
Thomas Hellera4ea6032003-04-17 18:55:45 +0000206 # l returns 'long', and does range checking (LONG_MIN ... LONG_MAX)
Thomas Woutersc9471232006-04-15 09:15:11 +0000207 self.assertRaises(TypeError, getargs_l, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000208 self.assertEqual(99, getargs_l(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200209 self.assertEqual(0, getargs_l(IntSubclass()))
210 self.assertRaises(TypeError, getargs_l, BadInt())
Serhiy Storchakac836a282013-12-14 21:07:09 +0200211 self.assertEqual(1, getargs_l(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200212 self.assertEqual(0, getargs_l(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000213
Thomas Heller54aa5782003-04-24 16:15:29 +0000214 self.assertRaises(OverflowError, getargs_l, LONG_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000215 self.assertEqual(LONG_MIN, getargs_l(LONG_MIN))
216 self.assertEqual(LONG_MAX, getargs_l(LONG_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +0000217 self.assertRaises(OverflowError, getargs_l, LONG_MAX+1)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000218
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000219 self.assertEqual(42, getargs_l(42))
Thomas Heller54aa5782003-04-24 16:15:29 +0000220 self.assertRaises(OverflowError, getargs_l, VERY_LARGE)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000221
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000222 def test_n(self):
223 from _testcapi import getargs_n
224 # n returns 'Py_ssize_t', and does range checking
225 # (PY_SSIZE_T_MIN ... PY_SSIZE_T_MAX)
Thomas Woutersd8073282006-04-21 11:36:13 +0000226 self.assertRaises(TypeError, getargs_n, 3.14)
Trent Nelson35133582008-04-22 19:02:40 +0000227 self.assertRaises(TypeError, getargs_n, Int())
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200228 self.assertEqual(0, getargs_n(IntSubclass()))
229 self.assertRaises(TypeError, getargs_n, BadInt())
230 self.assertRaises(TypeError, getargs_n, BadInt2())
231 self.assertEqual(0, getargs_n(BadInt3()))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000232
233 self.assertRaises(OverflowError, getargs_n, PY_SSIZE_T_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000234 self.assertEqual(PY_SSIZE_T_MIN, getargs_n(PY_SSIZE_T_MIN))
235 self.assertEqual(PY_SSIZE_T_MAX, getargs_n(PY_SSIZE_T_MAX))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000236 self.assertRaises(OverflowError, getargs_n, PY_SSIZE_T_MAX+1)
237
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000238 self.assertEqual(42, getargs_n(42))
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000239 self.assertRaises(OverflowError, getargs_n, VERY_LARGE)
240
Thomas Hellera4ea6032003-04-17 18:55:45 +0000241
242class LongLong_TestCase(unittest.TestCase):
243 def test_L(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000244 from _testcapi import getargs_L
Mark Dickinsonde604012010-01-01 19:27:32 +0000245 # L returns 'long long', and does range checking (LLONG_MIN
246 # ... LLONG_MAX)
Mark Dickinsonc7301312010-06-10 16:05:10 +0000247 self.assertRaises(TypeError, getargs_L, 3.14)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000248 self.assertRaises(TypeError, getargs_L, "Hello")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000249 self.assertEqual(99, getargs_L(Int()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200250 self.assertEqual(0, getargs_L(IntSubclass()))
251 self.assertRaises(TypeError, getargs_L, BadInt())
Serhiy Storchakac836a282013-12-14 21:07:09 +0200252 self.assertEqual(1, getargs_L(BadInt2()))
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200253 self.assertEqual(0, getargs_L(BadInt3()))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000254
Thomas Heller54aa5782003-04-24 16:15:29 +0000255 self.assertRaises(OverflowError, getargs_L, LLONG_MIN-1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000256 self.assertEqual(LLONG_MIN, getargs_L(LLONG_MIN))
257 self.assertEqual(LLONG_MAX, getargs_L(LLONG_MAX))
Thomas Heller54aa5782003-04-24 16:15:29 +0000258 self.assertRaises(OverflowError, getargs_L, LLONG_MAX+1)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000259
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000260 self.assertEqual(42, getargs_L(42))
Thomas Heller54aa5782003-04-24 16:15:29 +0000261 self.assertRaises(OverflowError, getargs_L, VERY_LARGE)
Tim Peters0eadaac2003-04-24 16:02:54 +0000262
Thomas Hellera4ea6032003-04-17 18:55:45 +0000263 def test_K(self):
Thomas Heller54aa5782003-04-24 16:15:29 +0000264 from _testcapi import getargs_K
Thomas Hellera4ea6032003-04-17 18:55:45 +0000265 # K return 'unsigned long long', no range checking
Thomas Heller54aa5782003-04-24 16:15:29 +0000266 self.assertRaises(TypeError, getargs_K, 3.14)
Thomas Heller54aa5782003-04-24 16:15:29 +0000267 self.assertRaises(TypeError, getargs_K, Int())
Serhiy Storchaka31a65542013-12-11 21:07:54 +0200268 self.assertEqual(0, getargs_K(IntSubclass()))
269 self.assertRaises(TypeError, getargs_K, BadInt())
270 self.assertRaises(TypeError, getargs_K, BadInt2())
271 self.assertEqual(0, getargs_K(BadInt3()))
272
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000273 self.assertEqual(ULLONG_MAX, getargs_K(ULLONG_MAX))
274 self.assertEqual(0, getargs_K(0))
275 self.assertEqual(0, getargs_K(ULLONG_MAX+1))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000276
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000277 self.assertEqual(42, getargs_K(42))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000278
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000279 self.assertEqual(VERY_LARGE & ULLONG_MAX, getargs_K(VERY_LARGE))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000280
Larry Hastingsfaf91e72012-05-05 16:54:29 -0700281class Paradox:
282 "This statement is false."
283 def __bool__(self):
284 raise NotImplementedError
285
286class Boolean_TestCase(unittest.TestCase):
287 def test_p(self):
288 from _testcapi import getargs_p
289 self.assertEqual(0, getargs_p(False))
290 self.assertEqual(0, getargs_p(None))
291 self.assertEqual(0, getargs_p(0))
292 self.assertEqual(0, getargs_p(0.0))
293 self.assertEqual(0, getargs_p(0j))
294 self.assertEqual(0, getargs_p(''))
295 self.assertEqual(0, getargs_p(()))
296 self.assertEqual(0, getargs_p([]))
297 self.assertEqual(0, getargs_p({}))
298
299 self.assertEqual(1, getargs_p(True))
300 self.assertEqual(1, getargs_p(1))
301 self.assertEqual(1, getargs_p(1.0))
302 self.assertEqual(1, getargs_p(1j))
303 self.assertEqual(1, getargs_p('x'))
304 self.assertEqual(1, getargs_p((1,)))
305 self.assertEqual(1, getargs_p([1]))
306 self.assertEqual(1, getargs_p({1:2}))
307 self.assertEqual(1, getargs_p(unittest.TestCase))
308
309 self.assertRaises(NotImplementedError, getargs_p, Paradox())
310
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000311
312class Tuple_TestCase(unittest.TestCase):
313 def test_tuple(self):
314 from _testcapi import getargs_tuple
315
316 ret = getargs_tuple(1, (2, 3))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000317 self.assertEqual(ret, (1,2,3))
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000318
319 # make sure invalid tuple arguments are handled correctly
320 class seq:
321 def __len__(self):
322 return 2
323 def __getitem__(self, n):
324 raise ValueError
325 self.assertRaises(TypeError, getargs_tuple, 1, seq())
326
Christian Heimes380f7f22008-02-28 11:19:05 +0000327class Keywords_TestCase(unittest.TestCase):
328 def test_positional_args(self):
329 # using all positional args
Ezio Melottib3aedd42010-11-20 19:04:17 +0000330 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000331 getargs_keywords((1,2), 3, (4,(5,6)), (7,8,9), 10),
332 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
333 )
Victor Stinner93b55132010-05-19 00:54:06 +0000334
Christian Heimes380f7f22008-02-28 11:19:05 +0000335 def test_mixed_args(self):
336 # positional and keyword args
Ezio Melottib3aedd42010-11-20 19:04:17 +0000337 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000338 getargs_keywords((1,2), 3, (4,(5,6)), arg4=(7,8,9), arg5=10),
339 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
340 )
Victor Stinner93b55132010-05-19 00:54:06 +0000341
Christian Heimes380f7f22008-02-28 11:19:05 +0000342 def test_keyword_args(self):
343 # all keywords
Ezio Melottib3aedd42010-11-20 19:04:17 +0000344 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000345 getargs_keywords(arg1=(1,2), arg2=3, arg3=(4,(5,6)), arg4=(7,8,9), arg5=10),
346 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
347 )
Victor Stinner93b55132010-05-19 00:54:06 +0000348
Christian Heimes380f7f22008-02-28 11:19:05 +0000349 def test_optional_args(self):
350 # missing optional keyword args, skipping tuples
Ezio Melottib3aedd42010-11-20 19:04:17 +0000351 self.assertEqual(
Christian Heimes380f7f22008-02-28 11:19:05 +0000352 getargs_keywords(arg1=(1,2), arg2=3, arg5=10),
353 (1, 2, 3, -1, -1, -1, -1, -1, -1, 10)
354 )
Victor Stinner93b55132010-05-19 00:54:06 +0000355
Christian Heimes380f7f22008-02-28 11:19:05 +0000356 def test_required_args(self):
357 # required arg missing
358 try:
359 getargs_keywords(arg1=(1,2))
360 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000361 self.assertEqual(str(err), "Required argument 'arg2' (pos 2) not found")
Christian Heimes380f7f22008-02-28 11:19:05 +0000362 else:
363 self.fail('TypeError should have been raised')
Victor Stinner93b55132010-05-19 00:54:06 +0000364
Christian Heimes380f7f22008-02-28 11:19:05 +0000365 def test_too_many_args(self):
366 try:
367 getargs_keywords((1,2),3,(4,(5,6)),(7,8,9),10,111)
368 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000369 self.assertEqual(str(err), "function takes at most 5 arguments (6 given)")
Christian Heimes380f7f22008-02-28 11:19:05 +0000370 else:
371 self.fail('TypeError should have been raised')
Victor Stinner93b55132010-05-19 00:54:06 +0000372
Christian Heimes380f7f22008-02-28 11:19:05 +0000373 def test_invalid_keyword(self):
374 # extraneous keyword arg
375 try:
376 getargs_keywords((1,2),3,arg5=10,arg666=666)
377 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000378 self.assertEqual(str(err), "'arg666' is an invalid keyword argument for this function")
Christian Heimes380f7f22008-02-28 11:19:05 +0000379 else:
380 self.fail('TypeError should have been raised')
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000381
Victor Stinner93b55132010-05-19 00:54:06 +0000382 def test_surrogate_keyword(self):
383 try:
384 getargs_keywords((1,2), 3, (4,(5,6)), (7,8,9), **{'\uDC80': 10})
385 except TypeError as err:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000386 self.assertEqual(str(err), "'\udc80' is an invalid keyword argument for this function")
Victor Stinner93b55132010-05-19 00:54:06 +0000387 else:
388 self.fail('TypeError should have been raised')
389
Larry Hastings83a9f482012-03-20 20:06:16 +0000390class KeywordOnly_TestCase(unittest.TestCase):
391 def test_positional_args(self):
392 # using all possible positional args
393 self.assertEqual(
394 getargs_keyword_only(1, 2),
395 (1, 2, -1)
396 )
397
398 def test_mixed_args(self):
399 # positional and keyword args
400 self.assertEqual(
401 getargs_keyword_only(1, 2, keyword_only=3),
402 (1, 2, 3)
403 )
404
405 def test_keyword_args(self):
406 # all keywords
407 self.assertEqual(
408 getargs_keyword_only(required=1, optional=2, keyword_only=3),
409 (1, 2, 3)
410 )
411
412 def test_optional_args(self):
413 # missing optional keyword args, skipping tuples
414 self.assertEqual(
415 getargs_keyword_only(required=1, optional=2),
416 (1, 2, -1)
417 )
418 self.assertEqual(
419 getargs_keyword_only(required=1, keyword_only=3),
420 (1, -1, 3)
421 )
422
423 def test_required_args(self):
424 self.assertEqual(
425 getargs_keyword_only(1),
426 (1, -1, -1)
427 )
428 self.assertEqual(
429 getargs_keyword_only(required=1),
430 (1, -1, -1)
431 )
432 # required arg missing
433 with self.assertRaisesRegex(TypeError,
434 "Required argument 'required' \(pos 1\) not found"):
435 getargs_keyword_only(optional=2)
436
437 with self.assertRaisesRegex(TypeError,
438 "Required argument 'required' \(pos 1\) not found"):
439 getargs_keyword_only(keyword_only=3)
440
441 def test_too_many_args(self):
442 with self.assertRaisesRegex(TypeError,
443 "Function takes at most 2 positional arguments \(3 given\)"):
444 getargs_keyword_only(1, 2, 3)
445
446 with self.assertRaisesRegex(TypeError,
447 "function takes at most 3 arguments \(4 given\)"):
448 getargs_keyword_only(1, 2, 3, keyword_only=5)
449
450 def test_invalid_keyword(self):
451 # extraneous keyword arg
452 with self.assertRaisesRegex(TypeError,
453 "'monster' is an invalid keyword argument for this function"):
454 getargs_keyword_only(1, 2, monster=666)
455
456 def test_surrogate_keyword(self):
457 with self.assertRaisesRegex(TypeError,
458 "'\udc80' is an invalid keyword argument for this function"):
459 getargs_keyword_only(1, 2, **{'\uDC80': 10})
460
Victor Stinner06e49dd2010-06-13 18:21:50 +0000461class Bytes_TestCase(unittest.TestCase):
Eli Bendersky906b88f2011-07-29 07:05:08 +0300462 def test_c(self):
463 from _testcapi import getargs_c
464 self.assertRaises(TypeError, getargs_c, b'abc') # len > 1
465 self.assertEqual(getargs_c(b'a'), b'a')
466 self.assertEqual(getargs_c(bytearray(b'a')), b'a')
467 self.assertRaises(TypeError, getargs_c, memoryview(b'a'))
468 self.assertRaises(TypeError, getargs_c, 's')
469 self.assertRaises(TypeError, getargs_c, None)
470
Victor Stinner06e49dd2010-06-13 18:21:50 +0000471 def test_s(self):
472 from _testcapi import getargs_s
473 self.assertEqual(getargs_s('abc\xe9'), b'abc\xc3\xa9')
474 self.assertRaises(TypeError, getargs_s, 'nul:\0')
475 self.assertRaises(TypeError, getargs_s, b'bytes')
476 self.assertRaises(TypeError, getargs_s, bytearray(b'bytearray'))
477 self.assertRaises(TypeError, getargs_s, memoryview(b'memoryview'))
478 self.assertRaises(TypeError, getargs_s, None)
479
480 def test_s_star(self):
481 from _testcapi import getargs_s_star
482 self.assertEqual(getargs_s_star('abc\xe9'), b'abc\xc3\xa9')
483 self.assertEqual(getargs_s_star('nul:\0'), b'nul:\0')
484 self.assertEqual(getargs_s_star(b'bytes'), b'bytes')
485 self.assertEqual(getargs_s_star(bytearray(b'bytearray')), b'bytearray')
486 self.assertEqual(getargs_s_star(memoryview(b'memoryview')), b'memoryview')
487 self.assertRaises(TypeError, getargs_s_star, None)
488
489 def test_s_hash(self):
490 from _testcapi import getargs_s_hash
491 self.assertEqual(getargs_s_hash('abc\xe9'), b'abc\xc3\xa9')
492 self.assertEqual(getargs_s_hash('nul:\0'), b'nul:\0')
493 self.assertEqual(getargs_s_hash(b'bytes'), b'bytes')
494 self.assertRaises(TypeError, getargs_s_hash, bytearray(b'bytearray'))
495 self.assertRaises(TypeError, getargs_s_hash, memoryview(b'memoryview'))
496 self.assertRaises(TypeError, getargs_s_hash, None)
497
498 def test_z(self):
499 from _testcapi import getargs_z
500 self.assertEqual(getargs_z('abc\xe9'), b'abc\xc3\xa9')
501 self.assertRaises(TypeError, getargs_z, 'nul:\0')
Victor Stinner4aae1eb2010-06-24 22:08:25 +0000502 self.assertRaises(TypeError, getargs_z, b'bytes')
Victor Stinner06e49dd2010-06-13 18:21:50 +0000503 self.assertRaises(TypeError, getargs_z, bytearray(b'bytearray'))
504 self.assertRaises(TypeError, getargs_z, memoryview(b'memoryview'))
505 self.assertIsNone(getargs_z(None))
506
507 def test_z_star(self):
508 from _testcapi import getargs_z_star
509 self.assertEqual(getargs_z_star('abc\xe9'), b'abc\xc3\xa9')
510 self.assertEqual(getargs_z_star('nul:\0'), b'nul:\0')
511 self.assertEqual(getargs_z_star(b'bytes'), b'bytes')
512 self.assertEqual(getargs_z_star(bytearray(b'bytearray')), b'bytearray')
513 self.assertEqual(getargs_z_star(memoryview(b'memoryview')), b'memoryview')
514 self.assertIsNone(getargs_z_star(None))
515
516 def test_z_hash(self):
517 from _testcapi import getargs_z_hash
518 self.assertEqual(getargs_z_hash('abc\xe9'), b'abc\xc3\xa9')
519 self.assertEqual(getargs_z_hash('nul:\0'), b'nul:\0')
520 self.assertEqual(getargs_z_hash(b'bytes'), b'bytes')
521 self.assertRaises(TypeError, getargs_z_hash, bytearray(b'bytearray'))
522 self.assertRaises(TypeError, getargs_z_hash, memoryview(b'memoryview'))
523 self.assertIsNone(getargs_z_hash(None))
524
525 def test_y(self):
526 from _testcapi import getargs_y
527 self.assertRaises(TypeError, getargs_y, 'abc\xe9')
528 self.assertEqual(getargs_y(b'bytes'), b'bytes')
529 self.assertRaises(TypeError, getargs_y, b'nul:\0')
530 self.assertRaises(TypeError, getargs_y, bytearray(b'bytearray'))
531 self.assertRaises(TypeError, getargs_y, memoryview(b'memoryview'))
532 self.assertRaises(TypeError, getargs_y, None)
533
534 def test_y_star(self):
535 from _testcapi import getargs_y_star
536 self.assertRaises(TypeError, getargs_y_star, 'abc\xe9')
537 self.assertEqual(getargs_y_star(b'bytes'), b'bytes')
538 self.assertEqual(getargs_y_star(b'nul:\0'), b'nul:\0')
539 self.assertEqual(getargs_y_star(bytearray(b'bytearray')), b'bytearray')
540 self.assertEqual(getargs_y_star(memoryview(b'memoryview')), b'memoryview')
541 self.assertRaises(TypeError, getargs_y_star, None)
542
543 def test_y_hash(self):
544 from _testcapi import getargs_y_hash
545 self.assertRaises(TypeError, getargs_y_hash, 'abc\xe9')
546 self.assertEqual(getargs_y_hash(b'bytes'), b'bytes')
547 self.assertEqual(getargs_y_hash(b'nul:\0'), b'nul:\0')
548 self.assertRaises(TypeError, getargs_y_hash, bytearray(b'bytearray'))
549 self.assertRaises(TypeError, getargs_y_hash, memoryview(b'memoryview'))
550 self.assertRaises(TypeError, getargs_y_hash, None)
551
Victor Stinner25e8ec42010-06-25 00:02:38 +0000552 def test_w_star(self):
553 # getargs_w_star() modifies first and last byte
554 from _testcapi import getargs_w_star
555 self.assertRaises(TypeError, getargs_w_star, 'abc\xe9')
556 self.assertRaises(TypeError, getargs_w_star, b'bytes')
557 self.assertRaises(TypeError, getargs_w_star, b'nul:\0')
Antoine Pitrouad62b032011-01-18 18:57:52 +0000558 self.assertRaises(TypeError, getargs_w_star, memoryview(b'bytes'))
Victor Stinner25e8ec42010-06-25 00:02:38 +0000559 self.assertEqual(getargs_w_star(bytearray(b'bytearray')), b'[ytearra]')
Antoine Pitrouad62b032011-01-18 18:57:52 +0000560 self.assertEqual(getargs_w_star(memoryview(bytearray(b'memoryview'))),
561 b'[emoryvie]')
Victor Stinner25e8ec42010-06-25 00:02:38 +0000562 self.assertRaises(TypeError, getargs_w_star, None)
563
Victor Stinner06e49dd2010-06-13 18:21:50 +0000564
565class Unicode_TestCase(unittest.TestCase):
566 def test_u(self):
567 from _testcapi import getargs_u
568 self.assertEqual(getargs_u('abc\xe9'), 'abc\xe9')
569 self.assertRaises(TypeError, getargs_u, 'nul:\0')
570 self.assertRaises(TypeError, getargs_u, b'bytes')
571 self.assertRaises(TypeError, getargs_u, bytearray(b'bytearray'))
572 self.assertRaises(TypeError, getargs_u, memoryview(b'memoryview'))
573 self.assertRaises(TypeError, getargs_u, None)
574
575 def test_u_hash(self):
576 from _testcapi import getargs_u_hash
577 self.assertEqual(getargs_u_hash('abc\xe9'), 'abc\xe9')
578 self.assertEqual(getargs_u_hash('nul:\0'), 'nul:\0')
579 self.assertRaises(TypeError, getargs_u_hash, b'bytes')
580 self.assertRaises(TypeError, getargs_u_hash, bytearray(b'bytearray'))
581 self.assertRaises(TypeError, getargs_u_hash, memoryview(b'memoryview'))
582 self.assertRaises(TypeError, getargs_u_hash, None)
583
584 def test_Z(self):
585 from _testcapi import getargs_Z
586 self.assertEqual(getargs_Z('abc\xe9'), 'abc\xe9')
587 self.assertRaises(TypeError, getargs_Z, 'nul:\0')
588 self.assertRaises(TypeError, getargs_Z, b'bytes')
589 self.assertRaises(TypeError, getargs_Z, bytearray(b'bytearray'))
590 self.assertRaises(TypeError, getargs_Z, memoryview(b'memoryview'))
591 self.assertIsNone(getargs_Z(None))
592
593 def test_Z_hash(self):
594 from _testcapi import getargs_Z_hash
595 self.assertEqual(getargs_Z_hash('abc\xe9'), 'abc\xe9')
596 self.assertEqual(getargs_Z_hash('nul:\0'), 'nul:\0')
597 self.assertRaises(TypeError, getargs_Z_hash, b'bytes')
598 self.assertRaises(TypeError, getargs_Z_hash, bytearray(b'bytearray'))
599 self.assertRaises(TypeError, getargs_Z_hash, memoryview(b'memoryview'))
600 self.assertIsNone(getargs_Z_hash(None))
601
602
Thomas Hellera4ea6032003-04-17 18:55:45 +0000603def test_main():
Victor Stinner06e49dd2010-06-13 18:21:50 +0000604 tests = [
605 Signed_TestCase,
606 Unsigned_TestCase,
Larry Hastingsfaf91e72012-05-05 16:54:29 -0700607 Boolean_TestCase,
Victor Stinner06e49dd2010-06-13 18:21:50 +0000608 Tuple_TestCase,
609 Keywords_TestCase,
Larry Hastings83a9f482012-03-20 20:06:16 +0000610 KeywordOnly_TestCase,
Victor Stinner06e49dd2010-06-13 18:21:50 +0000611 Bytes_TestCase,
612 Unicode_TestCase,
613 ]
Thomas Hellera4ea6032003-04-17 18:55:45 +0000614 try:
Thomas Heller54aa5782003-04-24 16:15:29 +0000615 from _testcapi import getargs_L, getargs_K
616 except ImportError:
Thomas Hellera4ea6032003-04-17 18:55:45 +0000617 pass # PY_LONG_LONG not available
618 else:
Walter Dörwald21d3a322003-05-01 17:45:56 +0000619 tests.append(LongLong_TestCase)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000620 support.run_unittest(*tests)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000621
622if __name__ == "__main__":
623 test_main()