blob: 1d889ddb3237df488b55d0e2c1a6aacf4b3f24fb [file] [log] [blame]
Tim Peters9ea17ac2001-02-02 05:57:15 +00001/*
2 * C Extension module to test Python interpreter C APIs.
3 *
4 * The 'test_*' functions exported by this module are run as part of the
5 * standard Python regression test, via Lib/test/test_capi.py.
6 */
7
8#include "Python.h"
Martin v. Löwis0347a9a2006-10-27 07:06:52 +00009#include <float.h>
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000010#include "structmember.h"
Benjamin Peterson001e4a62009-12-13 21:27:53 +000011#include "datetime.h"
Tim Peters9ea17ac2001-02-02 05:57:15 +000012
Mark Hammond8d98d2c2003-04-19 15:41:53 +000013#ifdef WITH_THREAD
14#include "pythread.h"
15#endif /* WITH_THREAD */
Tim Peters9ea17ac2001-02-02 05:57:15 +000016static PyObject *TestError; /* set to exception object in init */
17
Tim Peters91621db2001-06-12 20:10:01 +000018/* Raise TestError with test_name + ": " + msg, and return NULL. */
19
20static PyObject *
21raiseTestError(const char* test_name, const char* msg)
22{
23 char buf[2048];
24
25 if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
26 PyErr_SetString(TestError, "internal error msg too large");
27 else {
Tim Peters885d4572001-11-28 20:27:42 +000028 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
Tim Peters91621db2001-06-12 20:10:01 +000029 PyErr_SetString(TestError, buf);
30 }
31 return NULL;
32}
33
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000034/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
Tim Peters9ea17ac2001-02-02 05:57:15 +000035
36 The ones derived from autoconf on the UNIX-like OSes can be relied
37 upon (in the absence of sloppy cross-compiling), but the Windows
38 platforms have these hardcoded. Better safe than sorry.
39*/
40static PyObject*
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000041sizeof_error(const char* fatname, const char* typname,
Tim Peters9ea17ac2001-02-02 05:57:15 +000042 int expected, int got)
43{
44 char buf[1024];
Tim Petersbaefd9e2003-01-28 20:37:45 +000045 PyOS_snprintf(buf, sizeof(buf),
Tim Peters885d4572001-11-28 20:27:42 +000046 "%.200s #define == %d but sizeof(%.200s) == %d",
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000047 fatname, expected, typname, got);
Tim Peters9ea17ac2001-02-02 05:57:15 +000048 PyErr_SetString(TestError, buf);
49 return (PyObject*)NULL;
50}
51
52static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000053test_config(PyObject *self)
Tim Peters9ea17ac2001-02-02 05:57:15 +000054{
Tim Peters9ea17ac2001-02-02 05:57:15 +000055#define CHECK_SIZEOF(FATNAME, TYPE) \
56 if (FATNAME != sizeof(TYPE)) \
57 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
58
Tim Peters208efe52001-06-26 22:40:47 +000059 CHECK_SIZEOF(SIZEOF_SHORT, short);
Tim Peters9ea17ac2001-02-02 05:57:15 +000060 CHECK_SIZEOF(SIZEOF_INT, int);
61 CHECK_SIZEOF(SIZEOF_LONG, long);
62 CHECK_SIZEOF(SIZEOF_VOID_P, void*);
63 CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
64#ifdef HAVE_LONG_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +000065 CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
Tim Peters9ea17ac2001-02-02 05:57:15 +000066#endif
67
68#undef CHECK_SIZEOF
69
70 Py_INCREF(Py_None);
71 return Py_None;
72}
73
Tim Peters5c4d5bf2001-02-12 22:13:26 +000074static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000075test_list_api(PyObject *self)
Tim Peters5c4d5bf2001-02-12 22:13:26 +000076{
77 PyObject* list;
78 int i;
Tim Peters5c4d5bf2001-02-12 22:13:26 +000079
80 /* SF bug 132008: PyList_Reverse segfaults */
81#define NLIST 30
82 list = PyList_New(NLIST);
83 if (list == (PyObject*)NULL)
84 return (PyObject*)NULL;
85 /* list = range(NLIST) */
86 for (i = 0; i < NLIST; ++i) {
87 PyObject* anint = PyInt_FromLong(i);
88 if (anint == (PyObject*)NULL) {
89 Py_DECREF(list);
90 return (PyObject*)NULL;
91 }
92 PyList_SET_ITEM(list, i, anint);
93 }
94 /* list.reverse(), via PyList_Reverse() */
95 i = PyList_Reverse(list); /* should not blow up! */
96 if (i != 0) {
97 Py_DECREF(list);
98 return (PyObject*)NULL;
99 }
100 /* Check that list == range(29, -1, -1) now */
101 for (i = 0; i < NLIST; ++i) {
102 PyObject* anint = PyList_GET_ITEM(list, i);
103 if (PyInt_AS_LONG(anint) != NLIST-1-i) {
104 PyErr_SetString(TestError,
105 "test_list_api: reverse screwed up");
106 Py_DECREF(list);
107 return (PyObject*)NULL;
108 }
109 }
110 Py_DECREF(list);
111#undef NLIST
112
113 Py_INCREF(Py_None);
114 return Py_None;
115}
116
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000117static int
118test_dict_inner(int count)
119{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000120 Py_ssize_t pos = 0, iterations = 0;
121 int i;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000122 PyObject *dict = PyDict_New();
123 PyObject *v, *k;
124
125 if (dict == NULL)
126 return -1;
127
128 for (i = 0; i < count; i++) {
129 v = PyInt_FromLong(i);
130 PyDict_SetItem(dict, v, v);
131 Py_DECREF(v);
132 }
133
134 while (PyDict_Next(dict, &pos, &k, &v)) {
135 PyObject *o;
136 iterations++;
137
138 i = PyInt_AS_LONG(v) + 1;
139 o = PyInt_FromLong(i);
140 if (o == NULL)
141 return -1;
142 if (PyDict_SetItem(dict, k, o) < 0) {
143 Py_DECREF(o);
144 return -1;
145 }
146 Py_DECREF(o);
147 }
148
149 Py_DECREF(dict);
150
151 if (iterations != count) {
152 PyErr_SetString(
153 TestError,
154 "test_dict_iteration: dict iteration went wrong ");
155 return -1;
156 } else {
157 return 0;
158 }
159}
160
161static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +0000162test_dict_iteration(PyObject* self)
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000163{
164 int i;
165
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000166 for (i = 0; i < 200; i++) {
167 if (test_dict_inner(i) < 0) {
168 return NULL;
169 }
170 }
171
172 Py_INCREF(Py_None);
173 return Py_None;
174}
175
Tim Peters91621db2001-06-12 20:10:01 +0000176
Nick Coghlan180e4002008-12-30 01:18:48 +0000177/* Issue #4701: Check that PyObject_Hash implicitly calls
178 * PyType_Ready if it hasn't already been called
179 */
180static PyTypeObject _HashInheritanceTester_Type = {
Hirokazu Yamamoto52c1e3c2008-12-31 05:24:37 +0000181 PyObject_HEAD_INIT(NULL)
Nick Coghlan180e4002008-12-30 01:18:48 +0000182 0, /* Number of items for varobject */
183 "hashinheritancetester", /* Name of this type */
184 sizeof(PyObject), /* Basic object size */
185 0, /* Item size for varobject */
186 (destructor)PyObject_Del, /* tp_dealloc */
187 0, /* tp_print */
188 0, /* tp_getattr */
189 0, /* tp_setattr */
190 0, /* tp_compare */
191 0, /* tp_repr */
192 0, /* tp_as_number */
193 0, /* tp_as_sequence */
194 0, /* tp_as_mapping */
195 0, /* tp_hash */
196 0, /* tp_call */
197 0, /* tp_str */
198 PyObject_GenericGetAttr, /* tp_getattro */
199 0, /* tp_setattro */
200 0, /* tp_as_buffer */
201 Py_TPFLAGS_DEFAULT, /* tp_flags */
202 0, /* tp_doc */
203 0, /* tp_traverse */
204 0, /* tp_clear */
205 0, /* tp_richcompare */
206 0, /* tp_weaklistoffset */
207 0, /* tp_iter */
208 0, /* tp_iternext */
209 0, /* tp_methods */
210 0, /* tp_members */
211 0, /* tp_getset */
212 0, /* tp_base */
213 0, /* tp_dict */
214 0, /* tp_descr_get */
215 0, /* tp_descr_set */
216 0, /* tp_dictoffset */
217 0, /* tp_init */
218 0, /* tp_alloc */
219 PyType_GenericNew, /* tp_new */
220};
221
222static PyObject*
223test_lazy_hash_inheritance(PyObject* self)
224{
225 PyTypeObject *type;
226 PyObject *obj;
227 long hash;
228
229 type = &_HashInheritanceTester_Type;
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000230
Benjamin Petersonc6c1f962009-05-05 23:00:48 +0000231 if (type->tp_dict != NULL)
232 /* The type has already been initialized. This probably means
233 -R is being used. */
234 Py_RETURN_NONE;
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000235
236
Nick Coghlan180e4002008-12-30 01:18:48 +0000237 obj = PyObject_New(PyObject, type);
238 if (obj == NULL) {
239 PyErr_Clear();
240 PyErr_SetString(
241 TestError,
242 "test_lazy_hash_inheritance: failed to create object");
243 return NULL;
244 }
245
246 if (type->tp_dict != NULL) {
247 PyErr_SetString(
248 TestError,
249 "test_lazy_hash_inheritance: type initialised too soon");
250 Py_DECREF(obj);
251 return NULL;
252 }
253
254 hash = PyObject_Hash(obj);
255 if ((hash == -1) && PyErr_Occurred()) {
256 PyErr_Clear();
257 PyErr_SetString(
258 TestError,
259 "test_lazy_hash_inheritance: could not hash object");
260 Py_DECREF(obj);
261 return NULL;
262 }
263
264 if (type->tp_dict == NULL) {
265 PyErr_SetString(
266 TestError,
267 "test_lazy_hash_inheritance: type not initialised by hash()");
268 Py_DECREF(obj);
269 return NULL;
270 }
271
272 if (type->tp_hash != PyType_Type.tp_hash) {
273 PyErr_SetString(
274 TestError,
275 "test_lazy_hash_inheritance: unexpected hash function");
276 Py_DECREF(obj);
277 return NULL;
278 }
279
Benjamin Petersonc6c1f962009-05-05 23:00:48 +0000280 Py_DECREF(obj);
Benjamin Petersona98c8e12009-05-05 21:09:21 +0000281
Nick Coghlan180e4002008-12-30 01:18:48 +0000282 Py_RETURN_NONE;
283}
284
285
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000286/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
Tim Petersff70d3c2001-06-14 01:11:03 +0000287 PyLong_{As, From}{Unsigned,}LongLong().
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000288
289 Note that the meat of the test is contained in testcapi_long.h.
290 This is revolting, but delicate code duplication is worse: "almost
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000291 exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000292 dependence on type names makes it impossible to use a parameterized
293 function. A giant macro would be even worse than this. A C++ template
294 would be perfect.
295
296 The "report an error" functions are deliberately not part of the #include
297 file: if the test fails, you can set a breakpoint in the appropriate
298 error function directly, and crawl back from there in the debugger.
299*/
300
301#define UNBIND(X) Py_DECREF(X); (X) = NULL
302
303static PyObject *
304raise_test_long_error(const char* msg)
305{
306 return raiseTestError("test_long_api", msg);
307}
308
309#define TESTNAME test_long_api_inner
310#define TYPENAME long
311#define F_S_TO_PY PyLong_FromLong
312#define F_PY_TO_S PyLong_AsLong
313#define F_U_TO_PY PyLong_FromUnsignedLong
314#define F_PY_TO_U PyLong_AsUnsignedLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000315
316#include "testcapi_long.h"
317
318static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000319test_long_api(PyObject* self)
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000320{
Tim Peters83c9edc2001-06-16 08:10:13 +0000321 return TESTNAME(raise_test_long_error);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000322}
323
324#undef TESTNAME
325#undef TYPENAME
326#undef F_S_TO_PY
327#undef F_PY_TO_S
328#undef F_U_TO_PY
329#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000330
331#ifdef HAVE_LONG_LONG
Tim Peters91621db2001-06-12 20:10:01 +0000332
333static PyObject *
Tim Petersd1a7da62001-06-13 00:35:57 +0000334raise_test_longlong_error(const char* msg)
335{
336 return raiseTestError("test_longlong_api", msg);
337}
338
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000339#define TESTNAME test_longlong_api_inner
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000340#define TYPENAME PY_LONG_LONG
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000341#define F_S_TO_PY PyLong_FromLongLong
342#define F_PY_TO_S PyLong_AsLongLong
343#define F_U_TO_PY PyLong_FromUnsignedLongLong
344#define F_PY_TO_U PyLong_AsUnsignedLongLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000345
346#include "testcapi_long.h"
Tim Petersd1a7da62001-06-13 00:35:57 +0000347
348static PyObject *
Skip Montanaro9582c142006-04-18 01:01:41 +0000349test_longlong_api(PyObject* self, PyObject *args)
Tim Peters91621db2001-06-12 20:10:01 +0000350{
Tim Peters83c9edc2001-06-16 08:10:13 +0000351 return TESTNAME(raise_test_longlong_error);
Tim Peters91621db2001-06-12 20:10:01 +0000352}
353
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000354#undef TESTNAME
355#undef TYPENAME
356#undef F_S_TO_PY
357#undef F_PY_TO_S
358#undef F_U_TO_PY
359#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000360
Mark Dickinsone31d3002009-12-21 11:21:25 +0000361/* Test the PyLong_AsLongAndOverflow API. General conversion to PY_LONG
362 is tested by test_long_api_inner. This test will concentrate on proper
363 handling of overflow.
364*/
365
366static PyObject *
367test_long_and_overflow(PyObject *self)
368{
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000369 PyObject *num, *one, *temp;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000370 long value;
371 int overflow;
372
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000373 /* Test that overflow is set properly for a large value. */
374 /* num is a number larger than LONG_MAX even on 64-bit platforms */
Mark Dickinsone31d3002009-12-21 11:21:25 +0000375 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
376 if (num == NULL)
377 return NULL;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000378 overflow = 1234;
379 value = PyLong_AsLongAndOverflow(num, &overflow);
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000380 Py_DECREF(num);
381 if (value == -1 && PyErr_Occurred())
382 return NULL;
383 if (value != -1)
384 return raiseTestError("test_long_and_overflow",
385 "return value was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000386 if (overflow != 1)
387 return raiseTestError("test_long_and_overflow",
388 "overflow was not set to 1");
389
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000390 /* Same again, with num = LONG_MAX + 1 */
391 num = PyLong_FromLong(LONG_MAX);
392 if (num == NULL)
393 return NULL;
394 one = PyLong_FromLong(1L);
395 if (one == NULL) {
396 Py_DECREF(num);
397 return NULL;
398 }
399 temp = PyNumber_Add(num, one);
400 Py_DECREF(one);
401 Py_DECREF(num);
402 num = temp;
403 if (num == NULL)
404 return NULL;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000405 overflow = 0;
406 value = PyLong_AsLongAndOverflow(num, &overflow);
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000407 Py_DECREF(num);
408 if (value == -1 && PyErr_Occurred())
409 return NULL;
410 if (value != -1)
411 return raiseTestError("test_long_and_overflow",
412 "return value was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000413 if (overflow != 1)
414 return raiseTestError("test_long_and_overflow",
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000415 "overflow was not set to 1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000416
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000417 /* Test that overflow is set properly for a large negative value. */
418 /* num is a number smaller than LONG_MIN even on 64-bit platforms */
Mark Dickinsone31d3002009-12-21 11:21:25 +0000419 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
420 if (num == NULL)
421 return NULL;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000422 overflow = 1234;
423 value = PyLong_AsLongAndOverflow(num, &overflow);
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000424 Py_DECREF(num);
425 if (value == -1 && PyErr_Occurred())
426 return NULL;
427 if (value != -1)
428 return raiseTestError("test_long_and_overflow",
429 "return value was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000430 if (overflow != -1)
431 return raiseTestError("test_long_and_overflow",
432 "overflow was not set to -1");
433
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000434 /* Same again, with num = LONG_MIN - 1 */
435 num = PyLong_FromLong(LONG_MIN);
436 if (num == NULL)
437 return NULL;
438 one = PyLong_FromLong(1L);
439 if (one == NULL) {
440 Py_DECREF(num);
441 return NULL;
442 }
443 temp = PyNumber_Subtract(num, one);
444 Py_DECREF(one);
445 Py_DECREF(num);
446 num = temp;
447 if (num == NULL)
448 return NULL;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000449 overflow = 0;
450 value = PyLong_AsLongAndOverflow(num, &overflow);
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000451 Py_DECREF(num);
452 if (value == -1 && PyErr_Occurred())
453 return NULL;
454 if (value != -1)
455 return raiseTestError("test_long_and_overflow",
456 "return value was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000457 if (overflow != -1)
458 return raiseTestError("test_long_and_overflow",
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000459 "overflow was not set to -1");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000460
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000461 /* Test that overflow is cleared properly for small values. */
Mark Dickinsone31d3002009-12-21 11:21:25 +0000462 num = PyLong_FromString("FF", NULL, 16);
463 if (num == NULL)
464 return NULL;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000465 overflow = 1234;
466 value = PyLong_AsLongAndOverflow(num, &overflow);
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000467 Py_DECREF(num);
468 if (value == -1 && PyErr_Occurred())
469 return NULL;
470 if (value != 0xFF)
471 return raiseTestError("test_long_and_overflow",
472 "expected return value 0xFF");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000473 if (overflow != 0)
474 return raiseTestError("test_long_and_overflow",
475 "overflow was not cleared");
476
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000477 num = PyLong_FromString("-FF", NULL, 16);
478 if (num == NULL)
479 return NULL;
Mark Dickinsone31d3002009-12-21 11:21:25 +0000480 overflow = 0;
481 value = PyLong_AsLongAndOverflow(num, &overflow);
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000482 Py_DECREF(num);
483 if (value == -1 && PyErr_Occurred())
484 return NULL;
485 if (value != -0xFF)
486 return raiseTestError("test_long_and_overflow",
487 "expected return value 0xFF");
Mark Dickinsone31d3002009-12-21 11:21:25 +0000488 if (overflow != 0)
489 return raiseTestError("test_long_and_overflow",
490 "overflow was set incorrectly");
491
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000492 num = PyLong_FromLong(LONG_MAX);
493 if (num == NULL)
494 return NULL;
495 overflow = 1234;
496 value = PyLong_AsLongAndOverflow(num, &overflow);
Mark Dickinsoned02b3f2009-12-21 11:31:54 +0000497 Py_DECREF(num);
Mark Dickinsonbd335bf2009-12-21 12:15:48 +0000498 if (value == -1 && PyErr_Occurred())
499 return NULL;
500 if (value != LONG_MAX)
501 return raiseTestError("test_long_and_overflow",
502 "expected return value LONG_MAX");
503 if (overflow != 0)
504 return raiseTestError("test_long_and_overflow",
505 "overflow was not cleared");
506
507 num = PyLong_FromLong(LONG_MIN);
508 if (num == NULL)
509 return NULL;
510 overflow = 0;
511 value = PyLong_AsLongAndOverflow(num, &overflow);
512 Py_DECREF(num);
513 if (value == -1 && PyErr_Occurred())
514 return NULL;
515 if (value != LONG_MIN)
516 return raiseTestError("test_long_and_overflow",
517 "expected return value LONG_MIN");
518 if (overflow != 0)
519 return raiseTestError("test_long_and_overflow",
520 "overflow was not cleared");
Mark Dickinsoned02b3f2009-12-21 11:31:54 +0000521
Mark Dickinsone31d3002009-12-21 11:21:25 +0000522 Py_INCREF(Py_None);
523 return Py_None;
524}
525
Mark Dickinsona36507c2010-01-30 10:08:33 +0000526/* Test the PyLong_AsLongLongAndOverflow API. General conversion to
527 PY_LONG_LONG is tested by test_long_api_inner. This test will
528 concentrate on proper handling of overflow.
529*/
530
531static PyObject *
532test_long_long_and_overflow(PyObject *self)
533{
534 PyObject *num, *one, *temp;
535 PY_LONG_LONG value;
536 int overflow;
537
538 /* Test that overflow is set properly for a large value. */
539 /* num is a number larger than PY_LLONG_MAX on a typical machine. */
540 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
541 if (num == NULL)
542 return NULL;
543 overflow = 1234;
544 value = PyLong_AsLongLongAndOverflow(num, &overflow);
545 Py_DECREF(num);
546 if (value == -1 && PyErr_Occurred())
547 return NULL;
548 if (value != -1)
549 return raiseTestError("test_long_long_and_overflow",
550 "return value was not set to -1");
551 if (overflow != 1)
552 return raiseTestError("test_long_long_and_overflow",
553 "overflow was not set to 1");
554
555 /* Same again, with num = PY_LLONG_MAX + 1 */
556 num = PyLong_FromLongLong(PY_LLONG_MAX);
557 if (num == NULL)
558 return NULL;
559 one = PyLong_FromLong(1L);
560 if (one == NULL) {
561 Py_DECREF(num);
562 return NULL;
563 }
564 temp = PyNumber_Add(num, one);
565 Py_DECREF(one);
566 Py_DECREF(num);
567 num = temp;
568 if (num == NULL)
569 return NULL;
570 overflow = 0;
571 value = PyLong_AsLongLongAndOverflow(num, &overflow);
572 Py_DECREF(num);
573 if (value == -1 && PyErr_Occurred())
574 return NULL;
575 if (value != -1)
576 return raiseTestError("test_long_long_and_overflow",
577 "return value was not set to -1");
578 if (overflow != 1)
579 return raiseTestError("test_long_long_and_overflow",
580 "overflow was not set to 1");
581
582 /* Test that overflow is set properly for a large negative value. */
583 /* num is a number smaller than PY_LLONG_MIN on a typical platform */
584 num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
585 if (num == NULL)
586 return NULL;
587 overflow = 1234;
588 value = PyLong_AsLongLongAndOverflow(num, &overflow);
589 Py_DECREF(num);
590 if (value == -1 && PyErr_Occurred())
591 return NULL;
592 if (value != -1)
593 return raiseTestError("test_long_long_and_overflow",
594 "return value was not set to -1");
595 if (overflow != -1)
596 return raiseTestError("test_long_long_and_overflow",
597 "overflow was not set to -1");
598
599 /* Same again, with num = PY_LLONG_MIN - 1 */
600 num = PyLong_FromLongLong(PY_LLONG_MIN);
601 if (num == NULL)
602 return NULL;
603 one = PyLong_FromLong(1L);
604 if (one == NULL) {
605 Py_DECREF(num);
606 return NULL;
607 }
608 temp = PyNumber_Subtract(num, one);
609 Py_DECREF(one);
610 Py_DECREF(num);
611 num = temp;
612 if (num == NULL)
613 return NULL;
614 overflow = 0;
615 value = PyLong_AsLongLongAndOverflow(num, &overflow);
616 Py_DECREF(num);
617 if (value == -1 && PyErr_Occurred())
618 return NULL;
619 if (value != -1)
620 return raiseTestError("test_long_long_and_overflow",
621 "return value was not set to -1");
622 if (overflow != -1)
623 return raiseTestError("test_long_long_and_overflow",
624 "overflow was not set to -1");
625
626 /* Test that overflow is cleared properly for small values. */
627 num = PyLong_FromString("FF", NULL, 16);
628 if (num == NULL)
629 return NULL;
630 overflow = 1234;
631 value = PyLong_AsLongLongAndOverflow(num, &overflow);
632 Py_DECREF(num);
633 if (value == -1 && PyErr_Occurred())
634 return NULL;
635 if (value != 0xFF)
636 return raiseTestError("test_long_long_and_overflow",
637 "expected return value 0xFF");
638 if (overflow != 0)
639 return raiseTestError("test_long_long_and_overflow",
640 "overflow was not cleared");
641
642 num = PyLong_FromString("-FF", NULL, 16);
643 if (num == NULL)
644 return NULL;
645 overflow = 0;
646 value = PyLong_AsLongLongAndOverflow(num, &overflow);
647 Py_DECREF(num);
648 if (value == -1 && PyErr_Occurred())
649 return NULL;
650 if (value != -0xFF)
651 return raiseTestError("test_long_long_and_overflow",
652 "expected return value 0xFF");
653 if (overflow != 0)
654 return raiseTestError("test_long_long_and_overflow",
655 "overflow was set incorrectly");
656
657 num = PyLong_FromLongLong(PY_LLONG_MAX);
658 if (num == NULL)
659 return NULL;
660 overflow = 1234;
661 value = PyLong_AsLongLongAndOverflow(num, &overflow);
662 Py_DECREF(num);
663 if (value == -1 && PyErr_Occurred())
664 return NULL;
665 if (value != PY_LLONG_MAX)
666 return raiseTestError("test_long_long_and_overflow",
667 "expected return value PY_LLONG_MAX");
668 if (overflow != 0)
669 return raiseTestError("test_long_long_and_overflow",
670 "overflow was not cleared");
671
672 num = PyLong_FromLongLong(PY_LLONG_MIN);
673 if (num == NULL)
674 return NULL;
675 overflow = 0;
676 value = PyLong_AsLongLongAndOverflow(num, &overflow);
677 Py_DECREF(num);
678 if (value == -1 && PyErr_Occurred())
679 return NULL;
680 if (value != PY_LLONG_MIN)
681 return raiseTestError("test_long_long_and_overflow",
682 "expected return value PY_LLONG_MIN");
683 if (overflow != 0)
684 return raiseTestError("test_long_long_and_overflow",
685 "overflow was not cleared");
686
687 Py_INCREF(Py_None);
688 return Py_None;
689}
690
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000691/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000692 for both long and int arguments. The test may leak a little memory if
693 it fails.
694*/
695static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000696test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000697{
698 PyObject *tuple, *num;
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000699 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000700
Tim Petersd38b1c72001-09-30 05:09:37 +0000701 tuple = PyTuple_New(1);
702 if (tuple == NULL)
703 return NULL;
704
705 num = PyLong_FromLong(42);
706 if (num == NULL)
707 return NULL;
708
709 PyTuple_SET_ITEM(tuple, 0, num);
710
711 value = -1;
712 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
713 return NULL;
714 if (value != 42)
715 return raiseTestError("test_L_code",
716 "L code returned wrong value for long 42");
717
718 Py_DECREF(num);
719 num = PyInt_FromLong(42);
720 if (num == NULL)
721 return NULL;
722
723 PyTuple_SET_ITEM(tuple, 0, num);
724
725 value = -1;
726 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
727 return NULL;
728 if (value != 42)
729 return raiseTestError("test_L_code",
730 "L code returned wrong value for int 42");
731
732 Py_DECREF(tuple);
733 Py_INCREF(Py_None);
734 return Py_None;
735}
736
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000737#endif /* ifdef HAVE_LONG_LONG */
738
Georg Brandl5f135782006-07-26 08:03:10 +0000739/* Test tuple argument processing */
740static PyObject *
741getargs_tuple(PyObject *self, PyObject *args)
742{
743 int a, b, c;
744 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
745 return NULL;
746 return Py_BuildValue("iii", a, b, c);
747}
748
Christian Heimesea837932008-02-26 17:23:51 +0000749/* test PyArg_ParseTupleAndKeywords */
750static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
751{
752 static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
753 static char *fmt="(ii)i|(i(ii))(iii)i";
754 int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
755
756 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
757 &int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
758 &int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
759 return NULL;
760 return Py_BuildValue("iiiiiiiiii",
761 int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
762 int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
763}
764
Thomas Heller3457e4b2003-04-24 16:14:27 +0000765/* Functions to call PyArg_ParseTuple with integer format codes,
766 and return the result.
767*/
Thomas Hellera4ea6032003-04-17 18:55:45 +0000768static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000769getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000770{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000771 unsigned char value;
772 if (!PyArg_ParseTuple(args, "b", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000773 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000774 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000775}
776
Thomas Heller3457e4b2003-04-24 16:14:27 +0000777static PyObject *
778getargs_B(PyObject *self, PyObject *args)
779{
780 unsigned char value;
781 if (!PyArg_ParseTuple(args, "B", &value))
782 return NULL;
783 return PyLong_FromUnsignedLong((unsigned long)value);
784}
785
786static PyObject *
Mark Dickinsonb5e348b2009-12-20 15:57:56 +0000787getargs_h(PyObject *self, PyObject *args)
788{
789 short value;
790 if (!PyArg_ParseTuple(args, "h", &value))
791 return NULL;
792 return PyLong_FromLong((long)value);
793}
794
795static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000796getargs_H(PyObject *self, PyObject *args)
797{
798 unsigned short value;
799 if (!PyArg_ParseTuple(args, "H", &value))
800 return NULL;
801 return PyLong_FromUnsignedLong((unsigned long)value);
802}
803
804static PyObject *
805getargs_I(PyObject *self, PyObject *args)
806{
807 unsigned int value;
808 if (!PyArg_ParseTuple(args, "I", &value))
809 return NULL;
810 return PyLong_FromUnsignedLong((unsigned long)value);
811}
812
813static PyObject *
814getargs_k(PyObject *self, PyObject *args)
815{
816 unsigned long value;
817 if (!PyArg_ParseTuple(args, "k", &value))
818 return NULL;
819 return PyLong_FromUnsignedLong(value);
820}
821
822static PyObject *
823getargs_i(PyObject *self, PyObject *args)
824{
825 int value;
826 if (!PyArg_ParseTuple(args, "i", &value))
827 return NULL;
828 return PyLong_FromLong((long)value);
829}
830
Thomas Hellera4ea6032003-04-17 18:55:45 +0000831static PyObject *
832getargs_l(PyObject *self, PyObject *args)
833{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000834 long value;
835 if (!PyArg_ParseTuple(args, "l", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000836 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000837 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000838}
839
Georg Brandl7f573f72006-04-13 07:59:30 +0000840static PyObject *
841getargs_n(PyObject *self, PyObject *args)
842{
843 Py_ssize_t value;
844 if (!PyArg_ParseTuple(args, "n", &value))
845 return NULL;
846 return PyInt_FromSsize_t(value);
847}
848
Thomas Hellera4ea6032003-04-17 18:55:45 +0000849#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +0000850static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000851getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000852{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000853 PY_LONG_LONG value;
854 if (!PyArg_ParseTuple(args, "L", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000855 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000856 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000857}
858
Thomas Hellera4ea6032003-04-17 18:55:45 +0000859static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000860getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000861{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000862 unsigned PY_LONG_LONG value;
863 if (!PyArg_ParseTuple(args, "K", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000864 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000865 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000866}
867#endif
868
869/* This function not only tests the 'k' getargs code, but also the
870 PyInt_AsUnsignedLongMask() and PyInt_AsUnsignedLongMask() functions. */
871static PyObject *
872test_k_code(PyObject *self)
873{
874 PyObject *tuple, *num;
875 unsigned long value;
876
877 tuple = PyTuple_New(1);
878 if (tuple == NULL)
879 return NULL;
880
Neal Norwitz699cbb72003-04-22 01:28:57 +0000881 /* a number larger than ULONG_MAX even on 64-bit platforms */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000882 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
883 if (num == NULL)
884 return NULL;
885
886 value = PyInt_AsUnsignedLongMask(num);
Neal Norwitz699cbb72003-04-22 01:28:57 +0000887 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000888 return raiseTestError("test_k_code",
889 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
890
891 PyTuple_SET_ITEM(tuple, 0, num);
892
Neal Norwitzb183a252006-04-10 01:03:32 +0000893 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000894 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
895 return NULL;
Neal Norwitz699cbb72003-04-22 01:28:57 +0000896 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000897 return raiseTestError("test_k_code",
898 "k code returned wrong value for long 0xFFF...FFF");
899
900 Py_DECREF(num);
901 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
902 if (num == NULL)
903 return NULL;
904
905 value = PyInt_AsUnsignedLongMask(num);
906 if (value != (unsigned long)-0x42)
907 return raiseTestError("test_k_code",
908 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
909
910 PyTuple_SET_ITEM(tuple, 0, num);
911
Neal Norwitzb183a252006-04-10 01:03:32 +0000912 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000913 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
914 return NULL;
915 if (value != (unsigned long)-0x42)
916 return raiseTestError("test_k_code",
917 "k code returned wrong value for long -0xFFF..000042");
918
919 Py_DECREF(tuple);
920 Py_INCREF(Py_None);
921 return Py_None;
922}
923
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000924#ifdef Py_USING_UNICODE
925
Benjamin Petersone098c4a2008-12-23 20:12:33 +0000926static volatile int x;
Benjamin Petersonb6a53b52008-12-23 20:09:28 +0000927
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000928/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
929 of an error.
930*/
931static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000932test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000933{
934 PyObject *tuple, *obj;
935 Py_UNICODE *value;
936 int len;
937
Amaury Forgeot d'Arc07d539d2008-10-14 21:47:22 +0000938 /* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
939 /* Just use the macro and check that it compiles */
Benjamin Petersonb6a53b52008-12-23 20:09:28 +0000940 x = Py_UNICODE_ISSPACE(25);
Amaury Forgeot d'Arc07d539d2008-10-14 21:47:22 +0000941
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000942 tuple = PyTuple_New(1);
943 if (tuple == NULL)
944 return NULL;
945
946 obj = PyUnicode_Decode("test", strlen("test"),
947 "ascii", NULL);
948 if (obj == NULL)
949 return NULL;
950
951 PyTuple_SET_ITEM(tuple, 0, obj);
952
953 value = 0;
954 if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
955 return NULL;
956 if (value != PyUnicode_AS_UNICODE(obj))
957 return raiseTestError("test_u_code",
958 "u code returned wrong value for u'test'");
959 value = 0;
960 if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
961 return NULL;
962 if (value != PyUnicode_AS_UNICODE(obj) ||
963 len != PyUnicode_GET_SIZE(obj))
964 return raiseTestError("test_u_code",
965 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000966
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000967 Py_DECREF(tuple);
968 Py_INCREF(Py_None);
969 return Py_None;
970}
971
Tim Peters8931ff12006-05-13 23:28:20 +0000972static PyObject *
Mark Dickinson6b265f12009-03-18 16:07:26 +0000973test_widechar(PyObject *self)
974{
975#if defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
976 const wchar_t wtext[2] = {(wchar_t)0x10ABCDu};
977 size_t wtextlen = 1;
978#else
979 const wchar_t wtext[3] = {(wchar_t)0xDBEAu, (wchar_t)0xDFCDu};
980 size_t wtextlen = 2;
981#endif
982 PyObject *wide, *utf8;
983
984 wide = PyUnicode_FromWideChar(wtext, wtextlen);
985 if (wide == NULL)
986 return NULL;
987
988 utf8 = PyUnicode_FromString("\xf4\x8a\xaf\x8d");
989 if (utf8 == NULL) {
990 Py_DECREF(wide);
991 return NULL;
992 }
993
994 if (PyUnicode_GET_SIZE(wide) != PyUnicode_GET_SIZE(utf8)) {
995 Py_DECREF(wide);
996 Py_DECREF(utf8);
997 return raiseTestError("test_widechar",
998 "wide string and utf8 string have different length");
999 }
1000 if (PyUnicode_Compare(wide, utf8)) {
1001 Py_DECREF(wide);
1002 Py_DECREF(utf8);
1003 if (PyErr_Occurred())
1004 return NULL;
1005 return raiseTestError("test_widechar",
1006 "wide string and utf8 string are differents");
1007 }
1008
1009 Py_DECREF(wide);
1010 Py_DECREF(utf8);
1011 Py_RETURN_NONE;
1012}
1013
1014static PyObject *
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001015test_empty_argparse(PyObject *self)
1016{
1017 /* Test that formats can begin with '|'. See issue #4720. */
1018 PyObject *tuple, *dict = NULL;
1019 static char *kwlist[] = {NULL};
1020 int result;
1021 tuple = PyTuple_New(0);
1022 if (!tuple)
1023 return NULL;
1024 if ((result = PyArg_ParseTuple(tuple, "|:test_empty_argparse")) < 0)
1025 goto done;
1026 dict = PyDict_New();
1027 if (!dict)
1028 goto done;
1029 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
1030 done:
1031 Py_DECREF(tuple);
1032 Py_XDECREF(dict);
1033 if (result < 0)
1034 return NULL;
1035 else {
1036 Py_RETURN_NONE;
1037 }
1038}
1039
1040static PyObject *
Tim Peters8931ff12006-05-13 23:28:20 +00001041codec_incrementalencoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001042{
1043 const char *encoding, *errors = NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001044 if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001045 &encoding, &errors))
1046 return NULL;
1047 return PyCodec_IncrementalEncoder(encoding, errors);
1048}
1049
Tim Peters8931ff12006-05-13 23:28:20 +00001050static PyObject *
1051codec_incrementaldecoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001052{
1053 const char *encoding, *errors = NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001054 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001055 &encoding, &errors))
1056 return NULL;
1057 return PyCodec_IncrementalDecoder(encoding, errors);
1058}
1059
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001060#endif
1061
Tim Peters5b8132f2003-01-31 15:52:05 +00001062/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +00001063static PyObject *
1064test_long_numbits(PyObject *self)
1065{
Tim Peters5b8132f2003-01-31 15:52:05 +00001066 struct triple {
Tim Petersbaefd9e2003-01-28 20:37:45 +00001067 long input;
Tim Peters5b8132f2003-01-31 15:52:05 +00001068 size_t nbits;
1069 int sign;
1070 } testcases[] = {{0, 0, 0},
1071 {1L, 1, 1},
1072 {-1L, 1, -1},
1073 {2L, 2, 1},
1074 {-2L, 2, -1},
1075 {3L, 2, 1},
1076 {-3L, 2, -1},
1077 {4L, 3, 1},
1078 {-4L, 3, -1},
1079 {0x7fffL, 15, 1}, /* one Python long digit */
1080 {-0x7fffL, 15, -1},
1081 {0xffffL, 16, 1},
1082 {-0xffffL, 16, -1},
1083 {0xfffffffL, 28, 1},
1084 {-0xfffffffL, 28, -1}};
Tim Petersbaefd9e2003-01-28 20:37:45 +00001085 int i;
1086
Tim Peters5b8132f2003-01-31 15:52:05 +00001087 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
1088 PyObject *plong = PyLong_FromLong(testcases[i].input);
Tim Petersbaefd9e2003-01-28 20:37:45 +00001089 size_t nbits = _PyLong_NumBits(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +00001090 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +00001091
1092 Py_DECREF(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +00001093 if (nbits != testcases[i].nbits)
Tim Petersbaefd9e2003-01-28 20:37:45 +00001094 return raiseTestError("test_long_numbits",
Tim Peters5b8132f2003-01-31 15:52:05 +00001095 "wrong result for _PyLong_NumBits");
1096 if (sign != testcases[i].sign)
1097 return raiseTestError("test_long_numbits",
1098 "wrong result for _PyLong_Sign");
Tim Petersbaefd9e2003-01-28 20:37:45 +00001099 }
1100 Py_INCREF(Py_None);
1101 return Py_None;
1102}
1103
Neal Norwitz4a53dad2006-03-15 05:43:10 +00001104/* Example passing NULLs to PyObject_Str(NULL) and PyObject_Unicode(NULL). */
1105
1106static PyObject *
1107test_null_strings(PyObject *self)
1108{
1109 PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Unicode(NULL);
1110 PyObject *tuple = PyTuple_Pack(2, o1, o2);
1111 Py_XDECREF(o1);
1112 Py_XDECREF(o2);
1113 return tuple;
1114}
1115
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001116static PyObject *
1117raise_exception(PyObject *self, PyObject *args)
1118{
1119 PyObject *exc;
1120 PyObject *exc_args, *v;
1121 int num_args, i;
1122
1123 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
1124 &exc, &num_args))
1125 return NULL;
Benjamin Peterson37346b22008-08-23 20:27:43 +00001126 if (!PyExceptionClass_Check(exc)) {
1127 PyErr_Format(PyExc_TypeError, "an exception class is required");
1128 return NULL;
1129 }
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001130
1131 exc_args = PyTuple_New(num_args);
1132 if (exc_args == NULL)
1133 return NULL;
1134 for (i = 0; i < num_args; ++i) {
1135 v = PyInt_FromLong(i);
1136 if (v == NULL) {
1137 Py_DECREF(exc_args);
1138 return NULL;
1139 }
1140 PyTuple_SET_ITEM(exc_args, i, v);
1141 }
1142 PyErr_SetObject(exc, exc_args);
Michael W. Hudsonf02bcee2003-08-15 13:03:30 +00001143 Py_DECREF(exc_args);
Jeremy Hyltonede049b2001-09-26 20:01:13 +00001144 return NULL;
1145}
Tim Peters91621db2001-06-12 20:10:01 +00001146
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001147static PyObject *
1148test_datetime_capi(PyObject *self, PyObject *args) {
1149 if (PyDateTimeAPI) {
1150 PyErr_SetString(PyExc_AssertionError,
1151 "PyDateTime_CAPI somehow initialized");
1152 return NULL;
1153 }
1154 PyDateTime_IMPORT;
1155 if (PyDateTimeAPI)
1156 Py_RETURN_NONE;
1157 else
1158 return NULL;
1159}
1160
Benjamin Petersona04ae012010-01-30 23:26:05 +00001161
1162#ifdef WITH_THREAD
1163
Tim Peters59b96c12006-03-21 03:58:41 +00001164/* test_thread_state spawns a thread of its own, and that thread releases
1165 * `thread_done` when it's finished. The driver code has to know when the
1166 * thread finishes, because the thread uses a PyObject (the callable) that
1167 * may go away when the driver finishes. The former lack of this explicit
1168 * synchronization caused rare segfaults, so rare that they were seen only
1169 * on a Mac buildbot (although they were possible on any box).
1170 */
1171static PyThread_type_lock thread_done = NULL;
1172
Benjamin Peterson37346b22008-08-23 20:27:43 +00001173static int
Tim Peters59b96c12006-03-21 03:58:41 +00001174_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001175{
1176 PyObject *rc;
Benjamin Peterson37346b22008-08-23 20:27:43 +00001177 int success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001178 PyGILState_STATE s = PyGILState_Ensure();
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001179 rc = PyObject_CallFunction((PyObject *)callable, "");
Benjamin Peterson37346b22008-08-23 20:27:43 +00001180 success = (rc != NULL);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001181 Py_XDECREF(rc);
1182 PyGILState_Release(s);
Benjamin Peterson37346b22008-08-23 20:27:43 +00001183 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001184}
1185
Tim Peters59b96c12006-03-21 03:58:41 +00001186/* Same thing, but releases `thread_done` when it returns. This variant
1187 * should be called only from threads spawned by test_thread_state().
1188 */
1189static void
1190_make_call_from_thread(void *callable)
1191{
1192 _make_call(callable);
1193 PyThread_release_lock(thread_done);
1194}
1195
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001196static PyObject *
1197test_thread_state(PyObject *self, PyObject *args)
1198{
1199 PyObject *fn;
Benjamin Peterson37346b22008-08-23 20:27:43 +00001200 int success = 1;
Tim Peters59b96c12006-03-21 03:58:41 +00001201
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001202 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
1203 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001204
Benjamin Peterson37346b22008-08-23 20:27:43 +00001205 if (!PyCallable_Check(fn)) {
1206 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
1207 fn->ob_type->tp_name);
1208 return NULL;
1209 }
1210
Tim Peters59b96c12006-03-21 03:58:41 +00001211 /* Ensure Python is set up for threading */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001212 PyEval_InitThreads();
Tim Peters59b96c12006-03-21 03:58:41 +00001213 thread_done = PyThread_allocate_lock();
1214 if (thread_done == NULL)
1215 return PyErr_NoMemory();
1216 PyThread_acquire_lock(thread_done, 1);
1217
1218 /* Start a new thread with our callback. */
1219 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001220 /* Make the callback with the thread lock held by this thread */
Benjamin Peterson37346b22008-08-23 20:27:43 +00001221 success &= _make_call(fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001222 /* Do it all again, but this time with the thread-lock released */
1223 Py_BEGIN_ALLOW_THREADS
Benjamin Peterson37346b22008-08-23 20:27:43 +00001224 success &= _make_call(fn);
Tim Peters59b96c12006-03-21 03:58:41 +00001225 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001226 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001227
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001228 /* And once more with and without a thread
Tim Peters59b96c12006-03-21 03:58:41 +00001229 XXX - should use a lock and work out exactly what we are trying
1230 to test <wink>
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001231 */
1232 Py_BEGIN_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001233 PyThread_start_new_thread(_make_call_from_thread, fn);
Benjamin Peterson37346b22008-08-23 20:27:43 +00001234 success &= _make_call(fn);
Tim Peters59b96c12006-03-21 03:58:41 +00001235 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001236 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001237
Neal Norwitz97a57222006-10-28 21:17:51 +00001238 /* Release lock we acquired above. This is required on HP-UX. */
1239 PyThread_release_lock(thread_done);
1240
Tim Peters59b96c12006-03-21 03:58:41 +00001241 PyThread_free_lock(thread_done);
Benjamin Peterson37346b22008-08-23 20:27:43 +00001242 if (!success)
1243 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001244 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001245}
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001246
1247/* test Py_AddPendingCalls using threads */
1248static int _pending_callback(void *arg)
1249{
1250 /* we assume the argument is callable object to which we own a reference */
1251 PyObject *callable = (PyObject *)arg;
1252 PyObject *r = PyObject_CallObject(callable, NULL);
1253 Py_DECREF(callable);
1254 Py_XDECREF(r);
1255 return r != NULL ? 0 : -1;
1256}
1257
1258/* The following requests n callbacks to _pending_callback. It can be
1259 * run from any python thread.
1260 */
1261PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
1262{
1263 PyObject *callable;
1264 int r;
1265 if (PyArg_ParseTuple(arg, "O", &callable) == 0)
1266 return NULL;
1267
1268 /* create the reference for the callbackwhile we hold the lock */
1269 Py_INCREF(callable);
1270
1271 Py_BEGIN_ALLOW_THREADS
1272 r = Py_AddPendingCall(&_pending_callback, callable);
1273 Py_END_ALLOW_THREADS
1274
1275 if (r<0) {
1276 Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
1277 Py_INCREF(Py_False);
1278 return Py_False;
1279 }
1280 Py_INCREF(Py_True);
1281 return Py_True;
1282}
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001283#endif
1284
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001285/* Some tests of PyString_FromFormat(). This needs more tests. */
Tim Peters8931ff12006-05-13 23:28:20 +00001286static PyObject *
1287test_string_from_format(PyObject *self, PyObject *args)
1288{
1289 PyObject *result;
1290 char *msg;
1291
1292#define CHECK_1_FORMAT(FORMAT, TYPE) \
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001293 result = PyString_FromFormat(FORMAT, (TYPE)1); \
Tim Peters8931ff12006-05-13 23:28:20 +00001294 if (result == NULL) \
1295 return NULL; \
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001296 if (strcmp(PyString_AsString(result), "1")) { \
Tim Peters8931ff12006-05-13 23:28:20 +00001297 msg = FORMAT " failed at 1"; \
1298 goto Fail; \
1299 } \
1300 Py_DECREF(result)
1301
1302 CHECK_1_FORMAT("%d", int);
1303 CHECK_1_FORMAT("%ld", long);
1304 /* The z width modifier was added in Python 2.5. */
1305 CHECK_1_FORMAT("%zd", Py_ssize_t);
1306
1307 /* The u type code was added in Python 2.5. */
1308 CHECK_1_FORMAT("%u", unsigned int);
1309 CHECK_1_FORMAT("%lu", unsigned long);
1310 CHECK_1_FORMAT("%zu", size_t);
1311
Mark Dickinson82864d12009-11-15 16:18:58 +00001312 /* "%lld" and "%llu" support added in Python 2.7. */
1313#ifdef HAVE_LONG_LONG
1314 CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
1315 CHECK_1_FORMAT("%lld", PY_LONG_LONG);
1316#endif
1317
Tim Peters8931ff12006-05-13 23:28:20 +00001318 Py_RETURN_NONE;
1319
1320 Fail:
1321 Py_XDECREF(result);
1322 return raiseTestError("test_string_from_format", msg);
1323
1324#undef CHECK_1_FORMAT
1325}
1326
Georg Brandle9462c72006-08-04 18:03:37 +00001327/* This is here to provide a docstring for test_descr. */
1328static PyObject *
1329test_with_docstring(PyObject *self)
1330{
1331 Py_RETURN_NONE;
1332}
1333
Brett Cannon141534e2008-04-28 03:23:50 +00001334/* To test the format of tracebacks as printed out. */
1335static PyObject *
Brett Cannon8dc43032008-04-28 04:50:06 +00001336traceback_print(PyObject *self, PyObject *args)
Brett Cannon141534e2008-04-28 03:23:50 +00001337{
1338 PyObject *file;
1339 PyObject *traceback;
1340 int result;
1341
Brett Cannon8dc43032008-04-28 04:50:06 +00001342 if (!PyArg_ParseTuple(args, "OO:traceback_print",
Brett Cannon141534e2008-04-28 03:23:50 +00001343 &traceback, &file))
1344 return NULL;
1345
1346 result = PyTraceBack_Print(traceback, file);
1347 if (result < 0)
1348 return NULL;
1349 Py_RETURN_NONE;
1350}
1351
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00001352/* To test that the result of PyCode_NewEmpty has the right members. */
1353static PyObject *
1354code_newempty(PyObject *self, PyObject *args)
1355{
1356 const char *filename;
1357 const char *funcname;
1358 int firstlineno;
1359
1360 if (!PyArg_ParseTuple(args, "ssi:code_newempty",
1361 &filename, &funcname, &firstlineno))
1362 return NULL;
1363
1364 return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
1365}
1366
Georg Brandl740cdc32009-12-28 08:34:58 +00001367/* Test PyErr_NewExceptionWithDoc (also exercise PyErr_NewException).
1368 Run via Lib/test/test_exceptions.py */
1369static PyObject *
1370make_exception_with_doc(PyObject *self, PyObject *args, PyObject *kwargs)
1371{
1372 char *name;
1373 char *doc = NULL;
1374 PyObject *base = NULL;
1375 PyObject *dict = NULL;
1376
1377 static char *kwlist[] = {"name", "doc", "base", "dict", NULL};
1378
1379 if (!PyArg_ParseTupleAndKeywords(args, kwargs,
1380 "s|sOO:make_exception_with_doc", kwlist,
1381 &name, &doc, &base, &dict))
1382 return NULL;
1383
1384 return PyErr_NewExceptionWithDoc(name, doc, base, dict);
1385}
1386
Tim Peters9ea17ac2001-02-02 05:57:15 +00001387static PyMethodDef TestMethods[] = {
Fred Drakeacee69f2002-04-01 14:28:58 +00001388 {"raise_exception", raise_exception, METH_VARARGS},
1389 {"test_config", (PyCFunction)test_config, METH_NOARGS},
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001390 {"test_datetime_capi", test_datetime_capi, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001391 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
1392 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
Nick Coghlan180e4002008-12-30 01:18:48 +00001393 {"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001394 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
Mark Dickinsonbd335bf2009-12-21 12:15:48 +00001395 {"test_long_and_overflow", (PyCFunction)test_long_and_overflow,
1396 METH_NOARGS},
Tim Petersbaefd9e2003-01-28 20:37:45 +00001397 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
Thomas Hellera4ea6032003-04-17 18:55:45 +00001398 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001399 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Neal Norwitz4a53dad2006-03-15 05:43:10 +00001400 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
Tim Peters8931ff12006-05-13 23:28:20 +00001401 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
Georg Brandle9462c72006-08-04 18:03:37 +00001402 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
1403 PyDoc_STR("This is a pretty normal docstring.")},
Thomas Heller3457e4b2003-04-24 16:14:27 +00001404
Georg Brandl5f135782006-07-26 08:03:10 +00001405 {"getargs_tuple", getargs_tuple, METH_VARARGS},
Christian Heimesea837932008-02-26 17:23:51 +00001406 {"getargs_keywords", (PyCFunction)getargs_keywords,
1407 METH_VARARGS|METH_KEYWORDS},
Georg Brandl7f573f72006-04-13 07:59:30 +00001408 {"getargs_b", getargs_b, METH_VARARGS},
1409 {"getargs_B", getargs_B, METH_VARARGS},
Mark Dickinsonb5e348b2009-12-20 15:57:56 +00001410 {"getargs_h", getargs_h, METH_VARARGS},
Georg Brandl7f573f72006-04-13 07:59:30 +00001411 {"getargs_H", getargs_H, METH_VARARGS},
1412 {"getargs_I", getargs_I, METH_VARARGS},
1413 {"getargs_k", getargs_k, METH_VARARGS},
1414 {"getargs_i", getargs_i, METH_VARARGS},
1415 {"getargs_l", getargs_l, METH_VARARGS},
1416 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001417#ifdef HAVE_LONG_LONG
Georg Brandl7f573f72006-04-13 07:59:30 +00001418 {"getargs_L", getargs_L, METH_VARARGS},
1419 {"getargs_K", getargs_K, METH_VARARGS},
1420 {"test_longlong_api", test_longlong_api, METH_NOARGS},
Mark Dickinsona36507c2010-01-30 10:08:33 +00001421 {"test_long_long_and_overflow",
1422 (PyCFunction)test_long_long_and_overflow, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001423 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001424 {"codec_incrementalencoder",
1425 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
1426 {"codec_incrementaldecoder",
1427 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001428#endif
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001429#ifdef Py_USING_UNICODE
Fred Drakeacee69f2002-04-01 14:28:58 +00001430 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
Mark Dickinson6b265f12009-03-18 16:07:26 +00001431 {"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001432#endif
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001433#ifdef WITH_THREAD
Georg Brandl7f573f72006-04-13 07:59:30 +00001434 {"_test_thread_state", test_thread_state, METH_VARARGS},
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001435 {"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001436#endif
Brett Cannon8dc43032008-04-28 04:50:06 +00001437 {"traceback_print", traceback_print, METH_VARARGS},
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00001438 {"code_newempty", code_newempty, METH_VARARGS},
Georg Brandl740cdc32009-12-28 08:34:58 +00001439 {"make_exception_with_doc", (PyCFunction)make_exception_with_doc,
1440 METH_VARARGS | METH_KEYWORDS},
Tim Peters9ea17ac2001-02-02 05:57:15 +00001441 {NULL, NULL} /* sentinel */
1442};
1443
Thomas Hellera4ea6032003-04-17 18:55:45 +00001444#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
1445
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001446typedef struct {
Georg Brandl32a3fb52008-01-21 21:23:15 +00001447 char bool_member;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001448 char byte_member;
1449 unsigned char ubyte_member;
1450 short short_member;
1451 unsigned short ushort_member;
1452 int int_member;
1453 unsigned int uint_member;
1454 long long_member;
1455 unsigned long ulong_member;
1456 float float_member;
1457 double double_member;
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001458#ifdef HAVE_LONG_LONG
1459 PY_LONG_LONG longlong_member;
1460 unsigned PY_LONG_LONG ulonglong_member;
1461#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001462} all_structmembers;
1463
1464typedef struct {
1465 PyObject_HEAD
1466 all_structmembers structmembers;
1467} test_structmembers;
1468
1469static struct PyMemberDef test_members[] = {
Georg Brandl32a3fb52008-01-21 21:23:15 +00001470 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001471 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
1472 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
1473 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
1474 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
1475 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
1476 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
1477 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
1478 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
1479 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
1480 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001481#ifdef HAVE_LONG_LONG
1482 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
1483 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
1484#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001485 {NULL}
1486};
1487
1488
Georg Brandlc09b94e2008-01-21 21:28:32 +00001489static PyObject *
1490test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
1491{
1492 static char *keywords[] = {
1493 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
1494 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
1495 "T_FLOAT", "T_DOUBLE",
1496#ifdef HAVE_LONG_LONG
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001497 "T_LONGLONG", "T_ULONGLONG",
Georg Brandlc09b94e2008-01-21 21:28:32 +00001498#endif
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001499 NULL};
Georg Brandlc09b94e2008-01-21 21:28:32 +00001500 static char *fmt = "|bbBhHiIlkfd"
1501#ifdef HAVE_LONG_LONG
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001502 "LK"
Georg Brandlc09b94e2008-01-21 21:28:32 +00001503#endif
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001504 ;
Georg Brandlc09b94e2008-01-21 21:28:32 +00001505 test_structmembers *ob;
1506 ob = PyObject_New(test_structmembers, type);
1507 if (ob == NULL)
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001508 return NULL;
1509 memset(&ob->structmembers, 0, sizeof(all_structmembers));
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001510 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
Georg Brandlc09b94e2008-01-21 21:28:32 +00001511 &ob->structmembers.bool_member,
1512 &ob->structmembers.byte_member,
1513 &ob->structmembers.ubyte_member,
1514 &ob->structmembers.short_member,
1515 &ob->structmembers.ushort_member,
1516 &ob->structmembers.int_member,
1517 &ob->structmembers.uint_member,
1518 &ob->structmembers.long_member,
1519 &ob->structmembers.ulong_member,
1520 &ob->structmembers.float_member,
1521 &ob->structmembers.double_member
1522#ifdef HAVE_LONG_LONG
1523 , &ob->structmembers.longlong_member,
1524 &ob->structmembers.ulonglong_member
1525#endif
1526 )) {
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001527 Py_DECREF(ob);
1528 return NULL;
Georg Brandlc09b94e2008-01-21 21:28:32 +00001529 }
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001530 return (PyObject *)ob;
1531}
1532
Georg Brandlc09b94e2008-01-21 21:28:32 +00001533static void
1534test_structmembers_free(PyObject *ob)
1535{
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001536 PyObject_FREE(ob);
1537}
1538
1539static PyTypeObject test_structmembersType = {
Martin v. Löwis68192102007-07-21 06:55:02 +00001540 PyVarObject_HEAD_INIT(NULL, 0)
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001541 "test_structmembersType",
1542 sizeof(test_structmembers), /* tp_basicsize */
1543 0, /* tp_itemsize */
1544 test_structmembers_free, /* destructor tp_dealloc */
1545 0, /* tp_print */
1546 0, /* tp_getattr */
1547 0, /* tp_setattr */
1548 0, /* tp_compare */
1549 0, /* tp_repr */
1550 0, /* tp_as_number */
1551 0, /* tp_as_sequence */
1552 0, /* tp_as_mapping */
1553 0, /* tp_hash */
1554 0, /* tp_call */
1555 0, /* tp_str */
Georg Brandlc09b94e2008-01-21 21:28:32 +00001556 PyObject_GenericGetAttr, /* tp_getattro */
1557 PyObject_GenericSetAttr, /* tp_setattro */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001558 0, /* tp_as_buffer */
1559 0, /* tp_flags */
1560 "Type containing all structmember types",
1561 0, /* traverseproc tp_traverse */
1562 0, /* tp_clear */
1563 0, /* tp_richcompare */
1564 0, /* tp_weaklistoffset */
1565 0, /* tp_iter */
1566 0, /* tp_iternext */
1567 0, /* tp_methods */
Georg Brandlc09b94e2008-01-21 21:28:32 +00001568 test_members, /* tp_members */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001569 0,
1570 0,
1571 0,
1572 0,
1573 0,
1574 0,
1575 0,
1576 0,
Georg Brandlc09b94e2008-01-21 21:28:32 +00001577 test_structmembers_new, /* tp_new */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001578};
1579
1580
Mark Hammond62b1ab12002-07-23 06:31:15 +00001581PyMODINIT_FUNC
Tim Petersd66595f2001-02-04 03:09:53 +00001582init_testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00001583{
Fred Drakeacee69f2002-04-01 14:28:58 +00001584 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001585
Tim Petersd66595f2001-02-04 03:09:53 +00001586 m = Py_InitModule("_testcapi", TestMethods);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001587 if (m == NULL)
1588 return;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001589
Hirokazu Yamamoto52c1e3c2008-12-31 05:24:37 +00001590 Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
Hirokazu Yamamoto3cda1db2008-12-31 05:47:19 +00001591
Christian Heimese93237d2007-12-19 02:37:44 +00001592 Py_TYPE(&test_structmembersType)=&PyType_Type;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001593 Py_INCREF(&test_structmembersType);
1594 PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
1595
1596 PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
1597 PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001598 PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001599 PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
1600 PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001601 PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001602 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
1603 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001604 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001605 PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001606 PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
1607 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
1608 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
1609 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
1610 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
1611 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001612 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
1613 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
1614 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
Georg Brandl635af322006-04-13 07:29:18 +00001615 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001616 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
Robert Schuppenies47629022008-07-10 17:13:55 +00001617 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyInt_FromSsize_t(sizeof(PyGC_Head)));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001618
Tim Petersd66595f2001-02-04 03:09:53 +00001619 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
Fred Drakeacee69f2002-04-01 14:28:58 +00001620 Py_INCREF(TestError);
1621 PyModule_AddObject(m, "error", TestError);
Tim Peters9ea17ac2001-02-02 05:57:15 +00001622}