2 from test
import test_support
3 from _testcapi
import getargs_keywords
6 warnings
.filterwarnings("ignore",
7 category
=DeprecationWarning,
8 message
=".*integer argument expected, got float",
10 warnings
.filterwarnings("ignore",
11 category
=DeprecationWarning,
12 message
=".*integer argument expected, got float",
16 > How about the following counterproposal. This also changes some of
17 > the other format codes to be a little more regular.
19 > Code C type Range check
21 > b unsigned char 0..UCHAR_MAX
22 > h signed short SHRT_MIN..SHRT_MAX
23 > B unsigned char none **
24 > H unsigned short none **
25 > k * unsigned long none
26 > I * unsigned int 0..UINT_MAX
29 > i int INT_MIN..INT_MAX
30 > l long LONG_MIN..LONG_MAX
32 > K * unsigned long long none
33 > L long long LLONG_MIN..LLONG_MAX
39 > ** Changed from previous "range-and-a-half" to "none"; the
40 > range-and-a-half checking wasn't particularly useful.
42 Plus a C API or two, e.g. PyInt_AsLongMask() ->
43 unsigned long and PyInt_AsLongLongMask() -> unsigned
44 long long (if that exists).
48 VERY_LARGE
= 0xFF0000121212121212121242L
50 from _testcapi
import UCHAR_MAX
, USHRT_MAX
, UINT_MAX
, ULONG_MAX
, INT_MAX
, \
51 INT_MIN
, LONG_MIN
, LONG_MAX
, PY_SSIZE_T_MIN
, PY_SSIZE_T_MAX
53 # fake, they are not defined in Python's header files
66 class Unsigned_TestCase(unittest
.TestCase
):
68 from _testcapi
import getargs_b
69 # b returns 'unsigned char', and does range checking (0 ... UCHAR_MAX)
70 self
.assertEqual(3, getargs_b(3.14))
71 self
.assertEqual(99, getargs_b(Long()))
72 self
.assertEqual(99, getargs_b(Int()))
74 self
.assertRaises(OverflowError, getargs_b
, -1)
75 self
.assertEqual(0, getargs_b(0))
76 self
.assertEqual(UCHAR_MAX
, getargs_b(UCHAR_MAX
))
77 self
.assertRaises(OverflowError, getargs_b
, UCHAR_MAX
+ 1)
79 self
.assertEqual(42, getargs_b(42))
80 self
.assertEqual(42, getargs_b(42L))
81 self
.assertRaises(OverflowError, getargs_b
, VERY_LARGE
)
84 from _testcapi
import getargs_B
85 # B returns 'unsigned char', no range checking
86 self
.assertEqual(3, getargs_B(3.14))
87 self
.assertEqual(99, getargs_B(Long()))
88 self
.assertEqual(99, getargs_B(Int()))
90 self
.assertEqual(UCHAR_MAX
, getargs_B(-1))
91 self
.assertEqual(UCHAR_MAX
, getargs_B(-1L))
92 self
.assertEqual(0, getargs_B(0))
93 self
.assertEqual(UCHAR_MAX
, getargs_B(UCHAR_MAX
))
94 self
.assertEqual(0, getargs_B(UCHAR_MAX
+1))
96 self
.assertEqual(42, getargs_B(42))
97 self
.assertEqual(42, getargs_B(42L))
98 self
.assertEqual(UCHAR_MAX
& VERY_LARGE
, getargs_B(VERY_LARGE
))
101 from _testcapi
import getargs_H
102 # H returns 'unsigned short', no range checking
103 self
.assertEqual(3, getargs_H(3.14))
104 self
.assertEqual(99, getargs_H(Long()))
105 self
.assertEqual(99, getargs_H(Int()))
107 self
.assertEqual(USHRT_MAX
, getargs_H(-1))
108 self
.assertEqual(0, getargs_H(0))
109 self
.assertEqual(USHRT_MAX
, getargs_H(USHRT_MAX
))
110 self
.assertEqual(0, getargs_H(USHRT_MAX
+1))
112 self
.assertEqual(42, getargs_H(42))
113 self
.assertEqual(42, getargs_H(42L))
115 self
.assertEqual(VERY_LARGE
& USHRT_MAX
, getargs_H(VERY_LARGE
))
118 from _testcapi
import getargs_I
119 # I returns 'unsigned int', no range checking
120 self
.assertEqual(3, getargs_I(3.14))
121 self
.assertEqual(99, getargs_I(Long()))
122 self
.assertEqual(99, getargs_I(Int()))
124 self
.assertEqual(UINT_MAX
, getargs_I(-1))
125 self
.assertEqual(0, getargs_I(0))
126 self
.assertEqual(UINT_MAX
, getargs_I(UINT_MAX
))
127 self
.assertEqual(0, getargs_I(UINT_MAX
+1))
129 self
.assertEqual(42, getargs_I(42))
130 self
.assertEqual(42, getargs_I(42L))
132 self
.assertEqual(VERY_LARGE
& UINT_MAX
, getargs_I(VERY_LARGE
))
135 from _testcapi
import getargs_k
136 # k returns 'unsigned long', no range checking
137 # it does not accept float, or instances with __int__
138 self
.assertRaises(TypeError, getargs_k
, 3.14)
139 self
.assertRaises(TypeError, getargs_k
, Long())
140 self
.assertRaises(TypeError, getargs_k
, Int())
142 self
.assertEqual(ULONG_MAX
, getargs_k(-1))
143 self
.assertEqual(0, getargs_k(0))
144 self
.assertEqual(ULONG_MAX
, getargs_k(ULONG_MAX
))
145 self
.assertEqual(0, getargs_k(ULONG_MAX
+1))
147 self
.assertEqual(42, getargs_k(42))
148 self
.assertEqual(42, getargs_k(42L))
150 self
.assertEqual(VERY_LARGE
& ULONG_MAX
, getargs_k(VERY_LARGE
))
152 class Signed_TestCase(unittest
.TestCase
):
154 from _testcapi
import getargs_i
155 # i returns 'int', and does range checking (INT_MIN ... INT_MAX)
156 self
.assertEqual(3, getargs_i(3.14))
157 self
.assertEqual(99, getargs_i(Long()))
158 self
.assertEqual(99, getargs_i(Int()))
160 self
.assertRaises(OverflowError, getargs_i
, INT_MIN
-1)
161 self
.assertEqual(INT_MIN
, getargs_i(INT_MIN
))
162 self
.assertEqual(INT_MAX
, getargs_i(INT_MAX
))
163 self
.assertRaises(OverflowError, getargs_i
, INT_MAX
+1)
165 self
.assertEqual(42, getargs_i(42))
166 self
.assertEqual(42, getargs_i(42L))
167 self
.assertRaises(OverflowError, getargs_i
, VERY_LARGE
)
170 from _testcapi
import getargs_l
171 # l returns 'long', and does range checking (LONG_MIN ... LONG_MAX)
172 self
.assertEqual(3, getargs_l(3.14))
173 self
.assertEqual(99, getargs_l(Long()))
174 self
.assertEqual(99, getargs_l(Int()))
176 self
.assertRaises(OverflowError, getargs_l
, LONG_MIN
-1)
177 self
.assertEqual(LONG_MIN
, getargs_l(LONG_MIN
))
178 self
.assertEqual(LONG_MAX
, getargs_l(LONG_MAX
))
179 self
.assertRaises(OverflowError, getargs_l
, LONG_MAX
+1)
181 self
.assertEqual(42, getargs_l(42))
182 self
.assertEqual(42, getargs_l(42L))
183 self
.assertRaises(OverflowError, getargs_l
, VERY_LARGE
)
186 from _testcapi
import getargs_n
187 # n returns 'Py_ssize_t', and does range checking
188 # (PY_SSIZE_T_MIN ... PY_SSIZE_T_MAX)
189 self
.assertEqual(3, getargs_n(3.14))
190 self
.assertEqual(99, getargs_n(Long()))
191 self
.assertEqual(99, getargs_n(Int()))
193 self
.assertRaises(OverflowError, getargs_n
, PY_SSIZE_T_MIN
-1)
194 self
.assertEqual(PY_SSIZE_T_MIN
, getargs_n(PY_SSIZE_T_MIN
))
195 self
.assertEqual(PY_SSIZE_T_MAX
, getargs_n(PY_SSIZE_T_MAX
))
196 self
.assertRaises(OverflowError, getargs_n
, PY_SSIZE_T_MAX
+1)
198 self
.assertEqual(42, getargs_n(42))
199 self
.assertEqual(42, getargs_n(42L))
200 self
.assertRaises(OverflowError, getargs_n
, VERY_LARGE
)
203 class LongLong_TestCase(unittest
.TestCase
):
205 from _testcapi
import getargs_L
206 # L returns 'long long', and does range checking (LLONG_MIN ... LLONG_MAX)
207 self
.assertRaises(TypeError, getargs_L
, "Hello")
208 self
.assertEqual(3, getargs_L(3.14))
209 self
.assertEqual(99, getargs_L(Long()))
210 self
.assertEqual(99, getargs_L(Int()))
212 self
.assertRaises(OverflowError, getargs_L
, LLONG_MIN
-1)
213 self
.assertEqual(LLONG_MIN
, getargs_L(LLONG_MIN
))
214 self
.assertEqual(LLONG_MAX
, getargs_L(LLONG_MAX
))
215 self
.assertRaises(OverflowError, getargs_L
, LLONG_MAX
+1)
217 self
.assertEqual(42, getargs_L(42))
218 self
.assertEqual(42, getargs_L(42L))
219 self
.assertRaises(OverflowError, getargs_L
, VERY_LARGE
)
222 from _testcapi
import getargs_K
223 # K return 'unsigned long long', no range checking
224 self
.assertRaises(TypeError, getargs_K
, 3.14)
225 self
.assertRaises(TypeError, getargs_K
, Long())
226 self
.assertRaises(TypeError, getargs_K
, Int())
227 self
.assertEqual(ULLONG_MAX
, getargs_K(ULLONG_MAX
))
228 self
.assertEqual(0, getargs_K(0))
229 self
.assertEqual(0, getargs_K(ULLONG_MAX
+1))
231 self
.assertEqual(42, getargs_K(42))
232 self
.assertEqual(42, getargs_K(42L))
234 self
.assertEqual(VERY_LARGE
& ULLONG_MAX
, getargs_K(VERY_LARGE
))
237 class Tuple_TestCase(unittest
.TestCase
):
238 def test_tuple(self
):
239 from _testcapi
import getargs_tuple
241 ret
= getargs_tuple(1, (2, 3))
242 self
.assertEquals(ret
, (1,2,3))
244 # make sure invalid tuple arguments are handled correctly
248 def __getitem__(self
, n
):
250 self
.assertRaises(TypeError, getargs_tuple
, 1, seq())
252 class Keywords_TestCase(unittest
.TestCase
):
253 def test_positional_args(self
):
254 # using all positional args
256 getargs_keywords((1,2), 3, (4,(5,6)), (7,8,9), 10),
257 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
259 def test_mixed_args(self
):
260 # positional and keyword args
262 getargs_keywords((1,2), 3, (4,(5,6)), arg4
=(7,8,9), arg5
=10),
263 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
265 def test_keyword_args(self
):
268 getargs_keywords(arg1
=(1,2), arg2
=3, arg3
=(4,(5,6)), arg4
=(7,8,9), arg5
=10),
269 (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
271 def test_optional_args(self
):
272 # missing optional keyword args, skipping tuples
274 getargs_keywords(arg1
=(1,2), arg2
=3, arg5
=10),
275 (1, 2, 3, -1, -1, -1, -1, -1, -1, 10)
277 def test_required_args(self
):
278 # required arg missing
280 getargs_keywords(arg1
=(1,2))
281 except TypeError, err
:
282 self
.assertEquals(str(err
), "Required argument 'arg2' (pos 2) not found")
284 self
.fail('TypeError should have been raised')
285 def test_too_many_args(self
):
287 getargs_keywords((1,2),3,(4,(5,6)),(7,8,9),10,111)
288 except TypeError, err
:
289 self
.assertEquals(str(err
), "function takes at most 5 arguments (6 given)")
291 self
.fail('TypeError should have been raised')
292 def test_invalid_keyword(self
):
293 # extraneous keyword arg
295 getargs_keywords((1,2),3,arg5
=10,arg666
=666)
296 except TypeError, err
:
297 self
.assertEquals(str(err
), "'arg666' is an invalid keyword argument for this function")
299 self
.fail('TypeError should have been raised')
302 tests
= [Signed_TestCase
, Unsigned_TestCase
, Tuple_TestCase
, Keywords_TestCase
]
304 from _testcapi
import getargs_L
, getargs_K
306 pass # PY_LONG_LONG not available
308 tests
.append(LongLong_TestCase
)
309 test_support
.run_unittest(*tests
)
311 if __name__
== "__main__":