blob: 7aa57581d8a9646668637e3c6a69639746ac0cb5 [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
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001147#ifdef WITH_THREAD
1148
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001149static PyObject *
1150test_datetime_capi(PyObject *self, PyObject *args) {
1151 if (PyDateTimeAPI) {
1152 PyErr_SetString(PyExc_AssertionError,
1153 "PyDateTime_CAPI somehow initialized");
1154 return NULL;
1155 }
1156 PyDateTime_IMPORT;
1157 if (PyDateTimeAPI)
1158 Py_RETURN_NONE;
1159 else
1160 return NULL;
1161}
1162
Tim Peters59b96c12006-03-21 03:58:41 +00001163/* test_thread_state spawns a thread of its own, and that thread releases
1164 * `thread_done` when it's finished. The driver code has to know when the
1165 * thread finishes, because the thread uses a PyObject (the callable) that
1166 * may go away when the driver finishes. The former lack of this explicit
1167 * synchronization caused rare segfaults, so rare that they were seen only
1168 * on a Mac buildbot (although they were possible on any box).
1169 */
1170static PyThread_type_lock thread_done = NULL;
1171
Benjamin Peterson37346b22008-08-23 20:27:43 +00001172static int
Tim Peters59b96c12006-03-21 03:58:41 +00001173_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001174{
1175 PyObject *rc;
Benjamin Peterson37346b22008-08-23 20:27:43 +00001176 int success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001177 PyGILState_STATE s = PyGILState_Ensure();
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001178 rc = PyObject_CallFunction((PyObject *)callable, "");
Benjamin Peterson37346b22008-08-23 20:27:43 +00001179 success = (rc != NULL);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001180 Py_XDECREF(rc);
1181 PyGILState_Release(s);
Benjamin Peterson37346b22008-08-23 20:27:43 +00001182 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001183}
1184
Tim Peters59b96c12006-03-21 03:58:41 +00001185/* Same thing, but releases `thread_done` when it returns. This variant
1186 * should be called only from threads spawned by test_thread_state().
1187 */
1188static void
1189_make_call_from_thread(void *callable)
1190{
1191 _make_call(callable);
1192 PyThread_release_lock(thread_done);
1193}
1194
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001195static PyObject *
1196test_thread_state(PyObject *self, PyObject *args)
1197{
1198 PyObject *fn;
Benjamin Peterson37346b22008-08-23 20:27:43 +00001199 int success = 1;
Tim Peters59b96c12006-03-21 03:58:41 +00001200
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001201 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
1202 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001203
Benjamin Peterson37346b22008-08-23 20:27:43 +00001204 if (!PyCallable_Check(fn)) {
1205 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
1206 fn->ob_type->tp_name);
1207 return NULL;
1208 }
1209
Tim Peters59b96c12006-03-21 03:58:41 +00001210 /* Ensure Python is set up for threading */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001211 PyEval_InitThreads();
Tim Peters59b96c12006-03-21 03:58:41 +00001212 thread_done = PyThread_allocate_lock();
1213 if (thread_done == NULL)
1214 return PyErr_NoMemory();
1215 PyThread_acquire_lock(thread_done, 1);
1216
1217 /* Start a new thread with our callback. */
1218 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001219 /* Make the callback with the thread lock held by this thread */
Benjamin Peterson37346b22008-08-23 20:27:43 +00001220 success &= _make_call(fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001221 /* Do it all again, but this time with the thread-lock released */
1222 Py_BEGIN_ALLOW_THREADS
Benjamin Peterson37346b22008-08-23 20:27:43 +00001223 success &= _make_call(fn);
Tim Peters59b96c12006-03-21 03:58:41 +00001224 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001225 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001226
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001227 /* And once more with and without a thread
Tim Peters59b96c12006-03-21 03:58:41 +00001228 XXX - should use a lock and work out exactly what we are trying
1229 to test <wink>
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001230 */
1231 Py_BEGIN_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001232 PyThread_start_new_thread(_make_call_from_thread, fn);
Benjamin Peterson37346b22008-08-23 20:27:43 +00001233 success &= _make_call(fn);
Tim Peters59b96c12006-03-21 03:58:41 +00001234 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001235 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +00001236
Neal Norwitz97a57222006-10-28 21:17:51 +00001237 /* Release lock we acquired above. This is required on HP-UX. */
1238 PyThread_release_lock(thread_done);
1239
Tim Peters59b96c12006-03-21 03:58:41 +00001240 PyThread_free_lock(thread_done);
Benjamin Peterson37346b22008-08-23 20:27:43 +00001241 if (!success)
1242 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +00001243 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001244}
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001245
1246/* test Py_AddPendingCalls using threads */
1247static int _pending_callback(void *arg)
1248{
1249 /* we assume the argument is callable object to which we own a reference */
1250 PyObject *callable = (PyObject *)arg;
1251 PyObject *r = PyObject_CallObject(callable, NULL);
1252 Py_DECREF(callable);
1253 Py_XDECREF(r);
1254 return r != NULL ? 0 : -1;
1255}
1256
1257/* The following requests n callbacks to _pending_callback. It can be
1258 * run from any python thread.
1259 */
1260PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
1261{
1262 PyObject *callable;
1263 int r;
1264 if (PyArg_ParseTuple(arg, "O", &callable) == 0)
1265 return NULL;
1266
1267 /* create the reference for the callbackwhile we hold the lock */
1268 Py_INCREF(callable);
1269
1270 Py_BEGIN_ALLOW_THREADS
1271 r = Py_AddPendingCall(&_pending_callback, callable);
1272 Py_END_ALLOW_THREADS
1273
1274 if (r<0) {
1275 Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
1276 Py_INCREF(Py_False);
1277 return Py_False;
1278 }
1279 Py_INCREF(Py_True);
1280 return Py_True;
1281}
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001282#endif
1283
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001284/* Some tests of PyString_FromFormat(). This needs more tests. */
Tim Peters8931ff12006-05-13 23:28:20 +00001285static PyObject *
1286test_string_from_format(PyObject *self, PyObject *args)
1287{
1288 PyObject *result;
1289 char *msg;
1290
1291#define CHECK_1_FORMAT(FORMAT, TYPE) \
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001292 result = PyString_FromFormat(FORMAT, (TYPE)1); \
Tim Peters8931ff12006-05-13 23:28:20 +00001293 if (result == NULL) \
1294 return NULL; \
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001295 if (strcmp(PyString_AsString(result), "1")) { \
Tim Peters8931ff12006-05-13 23:28:20 +00001296 msg = FORMAT " failed at 1"; \
1297 goto Fail; \
1298 } \
1299 Py_DECREF(result)
1300
1301 CHECK_1_FORMAT("%d", int);
1302 CHECK_1_FORMAT("%ld", long);
1303 /* The z width modifier was added in Python 2.5. */
1304 CHECK_1_FORMAT("%zd", Py_ssize_t);
1305
1306 /* The u type code was added in Python 2.5. */
1307 CHECK_1_FORMAT("%u", unsigned int);
1308 CHECK_1_FORMAT("%lu", unsigned long);
1309 CHECK_1_FORMAT("%zu", size_t);
1310
Mark Dickinson82864d12009-11-15 16:18:58 +00001311 /* "%lld" and "%llu" support added in Python 2.7. */
1312#ifdef HAVE_LONG_LONG
1313 CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
1314 CHECK_1_FORMAT("%lld", PY_LONG_LONG);
1315#endif
1316
Tim Peters8931ff12006-05-13 23:28:20 +00001317 Py_RETURN_NONE;
1318
1319 Fail:
1320 Py_XDECREF(result);
1321 return raiseTestError("test_string_from_format", msg);
1322
1323#undef CHECK_1_FORMAT
1324}
1325
Georg Brandle9462c72006-08-04 18:03:37 +00001326/* This is here to provide a docstring for test_descr. */
1327static PyObject *
1328test_with_docstring(PyObject *self)
1329{
1330 Py_RETURN_NONE;
1331}
1332
Brett Cannon141534e2008-04-28 03:23:50 +00001333/* To test the format of tracebacks as printed out. */
1334static PyObject *
Brett Cannon8dc43032008-04-28 04:50:06 +00001335traceback_print(PyObject *self, PyObject *args)
Brett Cannon141534e2008-04-28 03:23:50 +00001336{
1337 PyObject *file;
1338 PyObject *traceback;
1339 int result;
1340
Brett Cannon8dc43032008-04-28 04:50:06 +00001341 if (!PyArg_ParseTuple(args, "OO:traceback_print",
Brett Cannon141534e2008-04-28 03:23:50 +00001342 &traceback, &file))
1343 return NULL;
1344
1345 result = PyTraceBack_Print(traceback, file);
1346 if (result < 0)
1347 return NULL;
1348 Py_RETURN_NONE;
1349}
1350
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00001351/* To test that the result of PyCode_NewEmpty has the right members. */
1352static PyObject *
1353code_newempty(PyObject *self, PyObject *args)
1354{
1355 const char *filename;
1356 const char *funcname;
1357 int firstlineno;
1358
1359 if (!PyArg_ParseTuple(args, "ssi:code_newempty",
1360 &filename, &funcname, &firstlineno))
1361 return NULL;
1362
1363 return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
1364}
1365
Georg Brandl740cdc32009-12-28 08:34:58 +00001366/* Test PyErr_NewExceptionWithDoc (also exercise PyErr_NewException).
1367 Run via Lib/test/test_exceptions.py */
1368static PyObject *
1369make_exception_with_doc(PyObject *self, PyObject *args, PyObject *kwargs)
1370{
1371 char *name;
1372 char *doc = NULL;
1373 PyObject *base = NULL;
1374 PyObject *dict = NULL;
1375
1376 static char *kwlist[] = {"name", "doc", "base", "dict", NULL};
1377
1378 if (!PyArg_ParseTupleAndKeywords(args, kwargs,
1379 "s|sOO:make_exception_with_doc", kwlist,
1380 &name, &doc, &base, &dict))
1381 return NULL;
1382
1383 return PyErr_NewExceptionWithDoc(name, doc, base, dict);
1384}
1385
Tim Peters9ea17ac2001-02-02 05:57:15 +00001386static PyMethodDef TestMethods[] = {
Fred Drakeacee69f2002-04-01 14:28:58 +00001387 {"raise_exception", raise_exception, METH_VARARGS},
1388 {"test_config", (PyCFunction)test_config, METH_NOARGS},
Benjamin Peterson001e4a62009-12-13 21:27:53 +00001389 {"test_datetime_capi", test_datetime_capi, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001390 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
1391 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
Nick Coghlan180e4002008-12-30 01:18:48 +00001392 {"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001393 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
Mark Dickinsonbd335bf2009-12-21 12:15:48 +00001394 {"test_long_and_overflow", (PyCFunction)test_long_and_overflow,
1395 METH_NOARGS},
Tim Petersbaefd9e2003-01-28 20:37:45 +00001396 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
Thomas Hellera4ea6032003-04-17 18:55:45 +00001397 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Benjamin Peterson4caef5c2008-12-22 20:51:15 +00001398 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Neal Norwitz4a53dad2006-03-15 05:43:10 +00001399 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
Tim Peters8931ff12006-05-13 23:28:20 +00001400 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
Georg Brandle9462c72006-08-04 18:03:37 +00001401 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
1402 PyDoc_STR("This is a pretty normal docstring.")},
Thomas Heller3457e4b2003-04-24 16:14:27 +00001403
Georg Brandl5f135782006-07-26 08:03:10 +00001404 {"getargs_tuple", getargs_tuple, METH_VARARGS},
Christian Heimesea837932008-02-26 17:23:51 +00001405 {"getargs_keywords", (PyCFunction)getargs_keywords,
1406 METH_VARARGS|METH_KEYWORDS},
Georg Brandl7f573f72006-04-13 07:59:30 +00001407 {"getargs_b", getargs_b, METH_VARARGS},
1408 {"getargs_B", getargs_B, METH_VARARGS},
Mark Dickinsonb5e348b2009-12-20 15:57:56 +00001409 {"getargs_h", getargs_h, METH_VARARGS},
Georg Brandl7f573f72006-04-13 07:59:30 +00001410 {"getargs_H", getargs_H, METH_VARARGS},
1411 {"getargs_I", getargs_I, METH_VARARGS},
1412 {"getargs_k", getargs_k, METH_VARARGS},
1413 {"getargs_i", getargs_i, METH_VARARGS},
1414 {"getargs_l", getargs_l, METH_VARARGS},
1415 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001416#ifdef HAVE_LONG_LONG
Georg Brandl7f573f72006-04-13 07:59:30 +00001417 {"getargs_L", getargs_L, METH_VARARGS},
1418 {"getargs_K", getargs_K, METH_VARARGS},
1419 {"test_longlong_api", test_longlong_api, METH_NOARGS},
Mark Dickinsona36507c2010-01-30 10:08:33 +00001420 {"test_long_long_and_overflow",
1421 (PyCFunction)test_long_long_and_overflow, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +00001422 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Walter Dörwald9ae019b2006-03-18 14:22:26 +00001423 {"codec_incrementalencoder",
1424 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
1425 {"codec_incrementaldecoder",
1426 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +00001427#endif
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001428#ifdef Py_USING_UNICODE
Fred Drakeacee69f2002-04-01 14:28:58 +00001429 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
Mark Dickinson6b265f12009-03-18 16:07:26 +00001430 {"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +00001431#endif
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001432#ifdef WITH_THREAD
Georg Brandl7f573f72006-04-13 07:59:30 +00001433 {"_test_thread_state", test_thread_state, METH_VARARGS},
Kristján Valur Jónsson0e2d8c32009-01-09 21:35:16 +00001434 {"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +00001435#endif
Brett Cannon8dc43032008-04-28 04:50:06 +00001436 {"traceback_print", traceback_print, METH_VARARGS},
Jeffrey Yasskin1aa47002009-05-08 21:51:06 +00001437 {"code_newempty", code_newempty, METH_VARARGS},
Georg Brandl740cdc32009-12-28 08:34:58 +00001438 {"make_exception_with_doc", (PyCFunction)make_exception_with_doc,
1439 METH_VARARGS | METH_KEYWORDS},
Tim Peters9ea17ac2001-02-02 05:57:15 +00001440 {NULL, NULL} /* sentinel */
1441};
1442
Thomas Hellera4ea6032003-04-17 18:55:45 +00001443#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
1444
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001445typedef struct {
Georg Brandl32a3fb52008-01-21 21:23:15 +00001446 char bool_member;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001447 char byte_member;
1448 unsigned char ubyte_member;
1449 short short_member;
1450 unsigned short ushort_member;
1451 int int_member;
1452 unsigned int uint_member;
1453 long long_member;
1454 unsigned long ulong_member;
1455 float float_member;
1456 double double_member;
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001457#ifdef HAVE_LONG_LONG
1458 PY_LONG_LONG longlong_member;
1459 unsigned PY_LONG_LONG ulonglong_member;
1460#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001461} all_structmembers;
1462
1463typedef struct {
1464 PyObject_HEAD
1465 all_structmembers structmembers;
1466} test_structmembers;
1467
1468static struct PyMemberDef test_members[] = {
Georg Brandl32a3fb52008-01-21 21:23:15 +00001469 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001470 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
1471 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
1472 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
1473 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
1474 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
1475 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
1476 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
1477 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
1478 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
1479 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001480#ifdef HAVE_LONG_LONG
1481 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
1482 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
1483#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001484 {NULL}
1485};
1486
1487
Georg Brandlc09b94e2008-01-21 21:28:32 +00001488static PyObject *
1489test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
1490{
1491 static char *keywords[] = {
1492 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
1493 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
1494 "T_FLOAT", "T_DOUBLE",
1495#ifdef HAVE_LONG_LONG
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001496 "T_LONGLONG", "T_ULONGLONG",
Georg Brandlc09b94e2008-01-21 21:28:32 +00001497#endif
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001498 NULL};
Georg Brandlc09b94e2008-01-21 21:28:32 +00001499 static char *fmt = "|bbBhHiIlkfd"
1500#ifdef HAVE_LONG_LONG
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001501 "LK"
Georg Brandlc09b94e2008-01-21 21:28:32 +00001502#endif
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001503 ;
Georg Brandlc09b94e2008-01-21 21:28:32 +00001504 test_structmembers *ob;
1505 ob = PyObject_New(test_structmembers, type);
1506 if (ob == NULL)
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001507 return NULL;
1508 memset(&ob->structmembers, 0, sizeof(all_structmembers));
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001509 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
Georg Brandlc09b94e2008-01-21 21:28:32 +00001510 &ob->structmembers.bool_member,
1511 &ob->structmembers.byte_member,
1512 &ob->structmembers.ubyte_member,
1513 &ob->structmembers.short_member,
1514 &ob->structmembers.ushort_member,
1515 &ob->structmembers.int_member,
1516 &ob->structmembers.uint_member,
1517 &ob->structmembers.long_member,
1518 &ob->structmembers.ulong_member,
1519 &ob->structmembers.float_member,
1520 &ob->structmembers.double_member
1521#ifdef HAVE_LONG_LONG
1522 , &ob->structmembers.longlong_member,
1523 &ob->structmembers.ulonglong_member
1524#endif
1525 )) {
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001526 Py_DECREF(ob);
1527 return NULL;
Georg Brandlc09b94e2008-01-21 21:28:32 +00001528 }
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001529 return (PyObject *)ob;
1530}
1531
Georg Brandlc09b94e2008-01-21 21:28:32 +00001532static void
1533test_structmembers_free(PyObject *ob)
1534{
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001535 PyObject_FREE(ob);
1536}
1537
1538static PyTypeObject test_structmembersType = {
Martin v. Löwis68192102007-07-21 06:55:02 +00001539 PyVarObject_HEAD_INIT(NULL, 0)
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001540 "test_structmembersType",
1541 sizeof(test_structmembers), /* tp_basicsize */
1542 0, /* tp_itemsize */
1543 test_structmembers_free, /* destructor tp_dealloc */
1544 0, /* tp_print */
1545 0, /* tp_getattr */
1546 0, /* tp_setattr */
1547 0, /* tp_compare */
1548 0, /* tp_repr */
1549 0, /* tp_as_number */
1550 0, /* tp_as_sequence */
1551 0, /* tp_as_mapping */
1552 0, /* tp_hash */
1553 0, /* tp_call */
1554 0, /* tp_str */
Georg Brandlc09b94e2008-01-21 21:28:32 +00001555 PyObject_GenericGetAttr, /* tp_getattro */
1556 PyObject_GenericSetAttr, /* tp_setattro */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001557 0, /* tp_as_buffer */
1558 0, /* tp_flags */
1559 "Type containing all structmember types",
1560 0, /* traverseproc tp_traverse */
1561 0, /* tp_clear */
1562 0, /* tp_richcompare */
1563 0, /* tp_weaklistoffset */
1564 0, /* tp_iter */
1565 0, /* tp_iternext */
1566 0, /* tp_methods */
Georg Brandlc09b94e2008-01-21 21:28:32 +00001567 test_members, /* tp_members */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001568 0,
1569 0,
1570 0,
1571 0,
1572 0,
1573 0,
1574 0,
1575 0,
Georg Brandlc09b94e2008-01-21 21:28:32 +00001576 test_structmembers_new, /* tp_new */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001577};
1578
1579
Mark Hammond62b1ab12002-07-23 06:31:15 +00001580PyMODINIT_FUNC
Tim Petersd66595f2001-02-04 03:09:53 +00001581init_testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00001582{
Fred Drakeacee69f2002-04-01 14:28:58 +00001583 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001584
Tim Petersd66595f2001-02-04 03:09:53 +00001585 m = Py_InitModule("_testcapi", TestMethods);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001586 if (m == NULL)
1587 return;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001588
Hirokazu Yamamoto52c1e3c2008-12-31 05:24:37 +00001589 Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
Hirokazu Yamamoto3cda1db2008-12-31 05:47:19 +00001590
Christian Heimese93237d2007-12-19 02:37:44 +00001591 Py_TYPE(&test_structmembersType)=&PyType_Type;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001592 Py_INCREF(&test_structmembersType);
1593 PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
1594
1595 PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
1596 PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001597 PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001598 PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
1599 PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001600 PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001601 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
1602 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001603 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001604 PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001605 PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
1606 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
1607 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
1608 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
1609 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
1610 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001611 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
1612 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
1613 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
Georg Brandl635af322006-04-13 07:29:18 +00001614 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001615 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
Robert Schuppenies47629022008-07-10 17:13:55 +00001616 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyInt_FromSsize_t(sizeof(PyGC_Head)));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001617
Tim Petersd66595f2001-02-04 03:09:53 +00001618 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
Fred Drakeacee69f2002-04-01 14:28:58 +00001619 Py_INCREF(TestError);
1620 PyModule_AddObject(m, "error", TestError);
Tim Peters9ea17ac2001-02-02 05:57:15 +00001621}