blob: d640c39131a8d5be89c59fa0131007ceaecdff6d [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"
Tim Peters9ea17ac2001-02-02 05:57:15 +000011
Mark Hammond8d98d2c2003-04-19 15:41:53 +000012#ifdef WITH_THREAD
13#include "pythread.h"
14#endif /* WITH_THREAD */
Tim Peters9ea17ac2001-02-02 05:57:15 +000015static PyObject *TestError; /* set to exception object in init */
16
Tim Peters91621db2001-06-12 20:10:01 +000017/* Raise TestError with test_name + ": " + msg, and return NULL. */
18
19static PyObject *
20raiseTestError(const char* test_name, const char* msg)
21{
22 char buf[2048];
23
24 if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
25 PyErr_SetString(TestError, "internal error msg too large");
26 else {
Tim Peters885d4572001-11-28 20:27:42 +000027 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
Tim Peters91621db2001-06-12 20:10:01 +000028 PyErr_SetString(TestError, buf);
29 }
30 return NULL;
31}
32
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000033/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
Tim Peters9ea17ac2001-02-02 05:57:15 +000034
35 The ones derived from autoconf on the UNIX-like OSes can be relied
36 upon (in the absence of sloppy cross-compiling), but the Windows
37 platforms have these hardcoded. Better safe than sorry.
38*/
39static PyObject*
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000040sizeof_error(const char* fatname, const char* typname,
Tim Peters9ea17ac2001-02-02 05:57:15 +000041 int expected, int got)
42{
43 char buf[1024];
Tim Petersbaefd9e2003-01-28 20:37:45 +000044 PyOS_snprintf(buf, sizeof(buf),
Tim Peters885d4572001-11-28 20:27:42 +000045 "%.200s #define == %d but sizeof(%.200s) == %d",
Martin v. Löwisb5bc5372006-10-27 06:16:31 +000046 fatname, expected, typname, got);
Tim Peters9ea17ac2001-02-02 05:57:15 +000047 PyErr_SetString(TestError, buf);
48 return (PyObject*)NULL;
49}
50
51static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000052test_config(PyObject *self)
Tim Peters9ea17ac2001-02-02 05:57:15 +000053{
Tim Peters9ea17ac2001-02-02 05:57:15 +000054#define CHECK_SIZEOF(FATNAME, TYPE) \
55 if (FATNAME != sizeof(TYPE)) \
56 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
57
Tim Peters208efe52001-06-26 22:40:47 +000058 CHECK_SIZEOF(SIZEOF_SHORT, short);
Tim Peters9ea17ac2001-02-02 05:57:15 +000059 CHECK_SIZEOF(SIZEOF_INT, int);
60 CHECK_SIZEOF(SIZEOF_LONG, long);
61 CHECK_SIZEOF(SIZEOF_VOID_P, void*);
62 CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
63#ifdef HAVE_LONG_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +000064 CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
Tim Peters9ea17ac2001-02-02 05:57:15 +000065#endif
66
67#undef CHECK_SIZEOF
68
69 Py_INCREF(Py_None);
70 return Py_None;
71}
72
Tim Peters5c4d5bf2001-02-12 22:13:26 +000073static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000074test_list_api(PyObject *self)
Tim Peters5c4d5bf2001-02-12 22:13:26 +000075{
76 PyObject* list;
77 int i;
Tim Peters5c4d5bf2001-02-12 22:13:26 +000078
79 /* SF bug 132008: PyList_Reverse segfaults */
80#define NLIST 30
81 list = PyList_New(NLIST);
82 if (list == (PyObject*)NULL)
83 return (PyObject*)NULL;
84 /* list = range(NLIST) */
85 for (i = 0; i < NLIST; ++i) {
86 PyObject* anint = PyInt_FromLong(i);
87 if (anint == (PyObject*)NULL) {
88 Py_DECREF(list);
89 return (PyObject*)NULL;
90 }
91 PyList_SET_ITEM(list, i, anint);
92 }
93 /* list.reverse(), via PyList_Reverse() */
94 i = PyList_Reverse(list); /* should not blow up! */
95 if (i != 0) {
96 Py_DECREF(list);
97 return (PyObject*)NULL;
98 }
99 /* Check that list == range(29, -1, -1) now */
100 for (i = 0; i < NLIST; ++i) {
101 PyObject* anint = PyList_GET_ITEM(list, i);
102 if (PyInt_AS_LONG(anint) != NLIST-1-i) {
103 PyErr_SetString(TestError,
104 "test_list_api: reverse screwed up");
105 Py_DECREF(list);
106 return (PyObject*)NULL;
107 }
108 }
109 Py_DECREF(list);
110#undef NLIST
111
112 Py_INCREF(Py_None);
113 return Py_None;
114}
115
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000116static int
117test_dict_inner(int count)
118{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000119 Py_ssize_t pos = 0, iterations = 0;
120 int i;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000121 PyObject *dict = PyDict_New();
122 PyObject *v, *k;
123
124 if (dict == NULL)
125 return -1;
126
127 for (i = 0; i < count; i++) {
128 v = PyInt_FromLong(i);
129 PyDict_SetItem(dict, v, v);
130 Py_DECREF(v);
131 }
132
133 while (PyDict_Next(dict, &pos, &k, &v)) {
134 PyObject *o;
135 iterations++;
136
137 i = PyInt_AS_LONG(v) + 1;
138 o = PyInt_FromLong(i);
139 if (o == NULL)
140 return -1;
141 if (PyDict_SetItem(dict, k, o) < 0) {
142 Py_DECREF(o);
143 return -1;
144 }
145 Py_DECREF(o);
146 }
147
148 Py_DECREF(dict);
149
150 if (iterations != count) {
151 PyErr_SetString(
152 TestError,
153 "test_dict_iteration: dict iteration went wrong ");
154 return -1;
155 } else {
156 return 0;
157 }
158}
159
160static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +0000161test_dict_iteration(PyObject* self)
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000162{
163 int i;
164
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000165 for (i = 0; i < 200; i++) {
166 if (test_dict_inner(i) < 0) {
167 return NULL;
168 }
169 }
170
171 Py_INCREF(Py_None);
172 return Py_None;
173}
174
Tim Peters91621db2001-06-12 20:10:01 +0000175
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000176/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
Tim Petersff70d3c2001-06-14 01:11:03 +0000177 PyLong_{As, From}{Unsigned,}LongLong().
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000178
179 Note that the meat of the test is contained in testcapi_long.h.
180 This is revolting, but delicate code duplication is worse: "almost
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000181 exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000182 dependence on type names makes it impossible to use a parameterized
183 function. A giant macro would be even worse than this. A C++ template
184 would be perfect.
185
186 The "report an error" functions are deliberately not part of the #include
187 file: if the test fails, you can set a breakpoint in the appropriate
188 error function directly, and crawl back from there in the debugger.
189*/
190
191#define UNBIND(X) Py_DECREF(X); (X) = NULL
192
193static PyObject *
194raise_test_long_error(const char* msg)
195{
196 return raiseTestError("test_long_api", msg);
197}
198
199#define TESTNAME test_long_api_inner
200#define TYPENAME long
201#define F_S_TO_PY PyLong_FromLong
202#define F_PY_TO_S PyLong_AsLong
203#define F_U_TO_PY PyLong_FromUnsignedLong
204#define F_PY_TO_U PyLong_AsUnsignedLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000205
206#include "testcapi_long.h"
207
208static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000209test_long_api(PyObject* self)
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000210{
Tim Peters83c9edc2001-06-16 08:10:13 +0000211 return TESTNAME(raise_test_long_error);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000212}
213
214#undef TESTNAME
215#undef TYPENAME
216#undef F_S_TO_PY
217#undef F_PY_TO_S
218#undef F_U_TO_PY
219#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000220
221#ifdef HAVE_LONG_LONG
Tim Peters91621db2001-06-12 20:10:01 +0000222
223static PyObject *
Tim Petersd1a7da62001-06-13 00:35:57 +0000224raise_test_longlong_error(const char* msg)
225{
226 return raiseTestError("test_longlong_api", msg);
227}
228
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000229#define TESTNAME test_longlong_api_inner
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000230#define TYPENAME PY_LONG_LONG
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000231#define F_S_TO_PY PyLong_FromLongLong
232#define F_PY_TO_S PyLong_AsLongLong
233#define F_U_TO_PY PyLong_FromUnsignedLongLong
234#define F_PY_TO_U PyLong_AsUnsignedLongLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000235
236#include "testcapi_long.h"
Tim Petersd1a7da62001-06-13 00:35:57 +0000237
238static PyObject *
Skip Montanaro9582c142006-04-18 01:01:41 +0000239test_longlong_api(PyObject* self, PyObject *args)
Tim Peters91621db2001-06-12 20:10:01 +0000240{
Tim Peters83c9edc2001-06-16 08:10:13 +0000241 return TESTNAME(raise_test_longlong_error);
Tim Peters91621db2001-06-12 20:10:01 +0000242}
243
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000244#undef TESTNAME
245#undef TYPENAME
246#undef F_S_TO_PY
247#undef F_PY_TO_S
248#undef F_U_TO_PY
249#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000250
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000251/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000252 for both long and int arguments. The test may leak a little memory if
253 it fails.
254*/
255static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000256test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000257{
258 PyObject *tuple, *num;
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000259 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000260
Tim Petersd38b1c72001-09-30 05:09:37 +0000261 tuple = PyTuple_New(1);
262 if (tuple == NULL)
263 return NULL;
264
265 num = PyLong_FromLong(42);
266 if (num == NULL)
267 return NULL;
268
269 PyTuple_SET_ITEM(tuple, 0, num);
270
271 value = -1;
272 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
273 return NULL;
274 if (value != 42)
275 return raiseTestError("test_L_code",
276 "L code returned wrong value for long 42");
277
278 Py_DECREF(num);
279 num = PyInt_FromLong(42);
280 if (num == NULL)
281 return NULL;
282
283 PyTuple_SET_ITEM(tuple, 0, num);
284
285 value = -1;
286 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
287 return NULL;
288 if (value != 42)
289 return raiseTestError("test_L_code",
290 "L code returned wrong value for int 42");
291
292 Py_DECREF(tuple);
293 Py_INCREF(Py_None);
294 return Py_None;
295}
296
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000297#endif /* ifdef HAVE_LONG_LONG */
298
Georg Brandl5f135782006-07-26 08:03:10 +0000299/* Test tuple argument processing */
300static PyObject *
301getargs_tuple(PyObject *self, PyObject *args)
302{
303 int a, b, c;
304 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
305 return NULL;
306 return Py_BuildValue("iii", a, b, c);
307}
308
Christian Heimesea837932008-02-26 17:23:51 +0000309/* test PyArg_ParseTupleAndKeywords */
310static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
311{
312 static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
313 static char *fmt="(ii)i|(i(ii))(iii)i";
314 int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
315
316 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
317 &int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
318 &int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
319 return NULL;
320 return Py_BuildValue("iiiiiiiiii",
321 int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
322 int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
323}
324
Thomas Heller3457e4b2003-04-24 16:14:27 +0000325/* Functions to call PyArg_ParseTuple with integer format codes,
326 and return the result.
327*/
Thomas Hellera4ea6032003-04-17 18:55:45 +0000328static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000329getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000330{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000331 unsigned char value;
332 if (!PyArg_ParseTuple(args, "b", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000333 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000334 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000335}
336
Thomas Heller3457e4b2003-04-24 16:14:27 +0000337static PyObject *
338getargs_B(PyObject *self, PyObject *args)
339{
340 unsigned char value;
341 if (!PyArg_ParseTuple(args, "B", &value))
342 return NULL;
343 return PyLong_FromUnsignedLong((unsigned long)value);
344}
345
346static PyObject *
347getargs_H(PyObject *self, PyObject *args)
348{
349 unsigned short value;
350 if (!PyArg_ParseTuple(args, "H", &value))
351 return NULL;
352 return PyLong_FromUnsignedLong((unsigned long)value);
353}
354
355static PyObject *
356getargs_I(PyObject *self, PyObject *args)
357{
358 unsigned int value;
359 if (!PyArg_ParseTuple(args, "I", &value))
360 return NULL;
361 return PyLong_FromUnsignedLong((unsigned long)value);
362}
363
364static PyObject *
365getargs_k(PyObject *self, PyObject *args)
366{
367 unsigned long value;
368 if (!PyArg_ParseTuple(args, "k", &value))
369 return NULL;
370 return PyLong_FromUnsignedLong(value);
371}
372
373static PyObject *
374getargs_i(PyObject *self, PyObject *args)
375{
376 int value;
377 if (!PyArg_ParseTuple(args, "i", &value))
378 return NULL;
379 return PyLong_FromLong((long)value);
380}
381
Thomas Hellera4ea6032003-04-17 18:55:45 +0000382static PyObject *
383getargs_l(PyObject *self, PyObject *args)
384{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000385 long value;
386 if (!PyArg_ParseTuple(args, "l", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000387 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000388 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000389}
390
Georg Brandl7f573f72006-04-13 07:59:30 +0000391static PyObject *
392getargs_n(PyObject *self, PyObject *args)
393{
394 Py_ssize_t value;
395 if (!PyArg_ParseTuple(args, "n", &value))
396 return NULL;
397 return PyInt_FromSsize_t(value);
398}
399
Thomas Hellera4ea6032003-04-17 18:55:45 +0000400#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +0000401static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000402getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000403{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000404 PY_LONG_LONG value;
405 if (!PyArg_ParseTuple(args, "L", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000406 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000407 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000408}
409
Thomas Hellera4ea6032003-04-17 18:55:45 +0000410static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000411getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000412{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000413 unsigned PY_LONG_LONG value;
414 if (!PyArg_ParseTuple(args, "K", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000415 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000416 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000417}
418#endif
419
420/* This function not only tests the 'k' getargs code, but also the
421 PyInt_AsUnsignedLongMask() and PyInt_AsUnsignedLongMask() functions. */
422static PyObject *
423test_k_code(PyObject *self)
424{
425 PyObject *tuple, *num;
426 unsigned long value;
427
428 tuple = PyTuple_New(1);
429 if (tuple == NULL)
430 return NULL;
431
Neal Norwitz699cbb72003-04-22 01:28:57 +0000432 /* a number larger than ULONG_MAX even on 64-bit platforms */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000433 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
434 if (num == NULL)
435 return NULL;
436
437 value = PyInt_AsUnsignedLongMask(num);
Neal Norwitz699cbb72003-04-22 01:28:57 +0000438 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000439 return raiseTestError("test_k_code",
440 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
441
442 PyTuple_SET_ITEM(tuple, 0, num);
443
Neal Norwitzb183a252006-04-10 01:03:32 +0000444 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000445 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
446 return NULL;
Neal Norwitz699cbb72003-04-22 01:28:57 +0000447 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000448 return raiseTestError("test_k_code",
449 "k code returned wrong value for long 0xFFF...FFF");
450
451 Py_DECREF(num);
452 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
453 if (num == NULL)
454 return NULL;
455
456 value = PyInt_AsUnsignedLongMask(num);
457 if (value != (unsigned long)-0x42)
458 return raiseTestError("test_k_code",
459 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
460
461 PyTuple_SET_ITEM(tuple, 0, num);
462
Neal Norwitzb183a252006-04-10 01:03:32 +0000463 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000464 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
465 return NULL;
466 if (value != (unsigned long)-0x42)
467 return raiseTestError("test_k_code",
468 "k code returned wrong value for long -0xFFF..000042");
469
470 Py_DECREF(tuple);
471 Py_INCREF(Py_None);
472 return Py_None;
473}
474
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000475#ifdef Py_USING_UNICODE
476
Benjamin Peterson828a7062008-12-27 17:05:29 +0000477static volatile int x;
478
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000479/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
480 of an error.
481*/
482static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000483test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000484{
485 PyObject *tuple, *obj;
486 Py_UNICODE *value;
487 int len;
488
Amaury Forgeot d'Arc38e46fc2008-10-14 22:00:06 +0000489 /* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
490 /* Just use the macro and check that it compiles */
Benjamin Peterson828a7062008-12-27 17:05:29 +0000491 x = Py_UNICODE_ISSPACE(25);
Amaury Forgeot d'Arc38e46fc2008-10-14 22:00:06 +0000492
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000493 tuple = PyTuple_New(1);
494 if (tuple == NULL)
495 return NULL;
496
497 obj = PyUnicode_Decode("test", strlen("test"),
498 "ascii", NULL);
499 if (obj == NULL)
500 return NULL;
501
502 PyTuple_SET_ITEM(tuple, 0, obj);
503
504 value = 0;
505 if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
506 return NULL;
507 if (value != PyUnicode_AS_UNICODE(obj))
508 return raiseTestError("test_u_code",
509 "u code returned wrong value for u'test'");
510 value = 0;
511 if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
512 return NULL;
513 if (value != PyUnicode_AS_UNICODE(obj) ||
514 len != PyUnicode_GET_SIZE(obj))
515 return raiseTestError("test_u_code",
516 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000517
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000518 Py_DECREF(tuple);
519 Py_INCREF(Py_None);
520 return Py_None;
521}
522
Tim Peters8931ff12006-05-13 23:28:20 +0000523static PyObject *
Benjamin Peterson828a7062008-12-27 17:05:29 +0000524test_empty_argparse(PyObject *self)
525{
526 /* Test that formats can begin with '|'. See issue #4720. */
527 PyObject *tuple, *dict = NULL;
528 static char *kwlist[] = {NULL};
529 int result;
530 tuple = PyTuple_New(0);
531 if (!tuple)
532 return NULL;
533 if ((result = PyArg_ParseTuple(tuple, "|:test_empty_argparse")) < 0)
534 goto done;
535 dict = PyDict_New();
536 if (!dict)
537 goto done;
538 result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
539 done:
540 Py_DECREF(tuple);
541 Py_XDECREF(dict);
542 if (result < 0)
543 return NULL;
544 else {
545 Py_RETURN_NONE;
546 }
547}
548
549static PyObject *
Tim Peters8931ff12006-05-13 23:28:20 +0000550codec_incrementalencoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +0000551{
552 const char *encoding, *errors = NULL;
Tim Peters59b96c12006-03-21 03:58:41 +0000553 if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
Walter Dörwald9ae019b2006-03-18 14:22:26 +0000554 &encoding, &errors))
555 return NULL;
556 return PyCodec_IncrementalEncoder(encoding, errors);
557}
558
Tim Peters8931ff12006-05-13 23:28:20 +0000559static PyObject *
560codec_incrementaldecoder(PyObject *self, PyObject *args)
Walter Dörwald9ae019b2006-03-18 14:22:26 +0000561{
562 const char *encoding, *errors = NULL;
Tim Peters59b96c12006-03-21 03:58:41 +0000563 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
Walter Dörwald9ae019b2006-03-18 14:22:26 +0000564 &encoding, &errors))
565 return NULL;
566 return PyCodec_IncrementalDecoder(encoding, errors);
567}
568
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000569#endif
570
Tim Peters5b8132f2003-01-31 15:52:05 +0000571/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +0000572static PyObject *
573test_long_numbits(PyObject *self)
574{
Tim Peters5b8132f2003-01-31 15:52:05 +0000575 struct triple {
Tim Petersbaefd9e2003-01-28 20:37:45 +0000576 long input;
Tim Peters5b8132f2003-01-31 15:52:05 +0000577 size_t nbits;
578 int sign;
579 } testcases[] = {{0, 0, 0},
580 {1L, 1, 1},
581 {-1L, 1, -1},
582 {2L, 2, 1},
583 {-2L, 2, -1},
584 {3L, 2, 1},
585 {-3L, 2, -1},
586 {4L, 3, 1},
587 {-4L, 3, -1},
588 {0x7fffL, 15, 1}, /* one Python long digit */
589 {-0x7fffL, 15, -1},
590 {0xffffL, 16, 1},
591 {-0xffffL, 16, -1},
592 {0xfffffffL, 28, 1},
593 {-0xfffffffL, 28, -1}};
Tim Petersbaefd9e2003-01-28 20:37:45 +0000594 int i;
595
Tim Peters5b8132f2003-01-31 15:52:05 +0000596 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
597 PyObject *plong = PyLong_FromLong(testcases[i].input);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000598 size_t nbits = _PyLong_NumBits(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000599 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000600
601 Py_DECREF(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000602 if (nbits != testcases[i].nbits)
Tim Petersbaefd9e2003-01-28 20:37:45 +0000603 return raiseTestError("test_long_numbits",
Tim Peters5b8132f2003-01-31 15:52:05 +0000604 "wrong result for _PyLong_NumBits");
605 if (sign != testcases[i].sign)
606 return raiseTestError("test_long_numbits",
607 "wrong result for _PyLong_Sign");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000608 }
609 Py_INCREF(Py_None);
610 return Py_None;
611}
612
Neal Norwitz4a53dad2006-03-15 05:43:10 +0000613/* Example passing NULLs to PyObject_Str(NULL) and PyObject_Unicode(NULL). */
614
615static PyObject *
616test_null_strings(PyObject *self)
617{
618 PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Unicode(NULL);
619 PyObject *tuple = PyTuple_Pack(2, o1, o2);
620 Py_XDECREF(o1);
621 Py_XDECREF(o2);
622 return tuple;
623}
624
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000625static PyObject *
626raise_exception(PyObject *self, PyObject *args)
627{
628 PyObject *exc;
629 PyObject *exc_args, *v;
630 int num_args, i;
631
632 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
633 &exc, &num_args))
634 return NULL;
Benjamin Peterson37346b22008-08-23 20:27:43 +0000635 if (!PyExceptionClass_Check(exc)) {
636 PyErr_Format(PyExc_TypeError, "an exception class is required");
637 return NULL;
638 }
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000639
640 exc_args = PyTuple_New(num_args);
641 if (exc_args == NULL)
642 return NULL;
643 for (i = 0; i < num_args; ++i) {
644 v = PyInt_FromLong(i);
645 if (v == NULL) {
646 Py_DECREF(exc_args);
647 return NULL;
648 }
649 PyTuple_SET_ITEM(exc_args, i, v);
650 }
651 PyErr_SetObject(exc, exc_args);
Michael W. Hudsonf02bcee2003-08-15 13:03:30 +0000652 Py_DECREF(exc_args);
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000653 return NULL;
654}
Tim Peters91621db2001-06-12 20:10:01 +0000655
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000656#ifdef WITH_THREAD
657
Tim Peters59b96c12006-03-21 03:58:41 +0000658/* test_thread_state spawns a thread of its own, and that thread releases
659 * `thread_done` when it's finished. The driver code has to know when the
660 * thread finishes, because the thread uses a PyObject (the callable) that
661 * may go away when the driver finishes. The former lack of this explicit
662 * synchronization caused rare segfaults, so rare that they were seen only
663 * on a Mac buildbot (although they were possible on any box).
664 */
665static PyThread_type_lock thread_done = NULL;
666
Benjamin Peterson37346b22008-08-23 20:27:43 +0000667static int
Tim Peters59b96c12006-03-21 03:58:41 +0000668_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000669{
670 PyObject *rc;
Benjamin Peterson37346b22008-08-23 20:27:43 +0000671 int success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000672 PyGILState_STATE s = PyGILState_Ensure();
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000673 rc = PyObject_CallFunction((PyObject *)callable, "");
Benjamin Peterson37346b22008-08-23 20:27:43 +0000674 success = (rc != NULL);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000675 Py_XDECREF(rc);
676 PyGILState_Release(s);
Benjamin Peterson37346b22008-08-23 20:27:43 +0000677 return success;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000678}
679
Tim Peters59b96c12006-03-21 03:58:41 +0000680/* Same thing, but releases `thread_done` when it returns. This variant
681 * should be called only from threads spawned by test_thread_state().
682 */
683static void
684_make_call_from_thread(void *callable)
685{
686 _make_call(callable);
687 PyThread_release_lock(thread_done);
688}
689
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000690static PyObject *
691test_thread_state(PyObject *self, PyObject *args)
692{
693 PyObject *fn;
Benjamin Peterson37346b22008-08-23 20:27:43 +0000694 int success = 1;
Tim Peters59b96c12006-03-21 03:58:41 +0000695
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000696 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
697 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +0000698
Benjamin Peterson37346b22008-08-23 20:27:43 +0000699 if (!PyCallable_Check(fn)) {
700 PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
701 fn->ob_type->tp_name);
702 return NULL;
703 }
704
Tim Peters59b96c12006-03-21 03:58:41 +0000705 /* Ensure Python is set up for threading */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000706 PyEval_InitThreads();
Tim Peters59b96c12006-03-21 03:58:41 +0000707 thread_done = PyThread_allocate_lock();
708 if (thread_done == NULL)
709 return PyErr_NoMemory();
710 PyThread_acquire_lock(thread_done, 1);
711
712 /* Start a new thread with our callback. */
713 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000714 /* Make the callback with the thread lock held by this thread */
Benjamin Peterson37346b22008-08-23 20:27:43 +0000715 success &= _make_call(fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000716 /* Do it all again, but this time with the thread-lock released */
717 Py_BEGIN_ALLOW_THREADS
Benjamin Peterson37346b22008-08-23 20:27:43 +0000718 success &= _make_call(fn);
Tim Peters59b96c12006-03-21 03:58:41 +0000719 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000720 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +0000721
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000722 /* And once more with and without a thread
Tim Peters59b96c12006-03-21 03:58:41 +0000723 XXX - should use a lock and work out exactly what we are trying
724 to test <wink>
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000725 */
726 Py_BEGIN_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +0000727 PyThread_start_new_thread(_make_call_from_thread, fn);
Benjamin Peterson37346b22008-08-23 20:27:43 +0000728 success &= _make_call(fn);
Tim Peters59b96c12006-03-21 03:58:41 +0000729 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000730 Py_END_ALLOW_THREADS
Tim Peters59b96c12006-03-21 03:58:41 +0000731
Neal Norwitz97a57222006-10-28 21:17:51 +0000732 /* Release lock we acquired above. This is required on HP-UX. */
733 PyThread_release_lock(thread_done);
734
Tim Peters59b96c12006-03-21 03:58:41 +0000735 PyThread_free_lock(thread_done);
Benjamin Peterson37346b22008-08-23 20:27:43 +0000736 if (!success)
737 return NULL;
Tim Peters59b96c12006-03-21 03:58:41 +0000738 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000739}
740#endif
741
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000742/* Some tests of PyString_FromFormat(). This needs more tests. */
Tim Peters8931ff12006-05-13 23:28:20 +0000743static PyObject *
744test_string_from_format(PyObject *self, PyObject *args)
745{
746 PyObject *result;
747 char *msg;
748
749#define CHECK_1_FORMAT(FORMAT, TYPE) \
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000750 result = PyString_FromFormat(FORMAT, (TYPE)1); \
Tim Peters8931ff12006-05-13 23:28:20 +0000751 if (result == NULL) \
752 return NULL; \
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000753 if (strcmp(PyString_AsString(result), "1")) { \
Tim Peters8931ff12006-05-13 23:28:20 +0000754 msg = FORMAT " failed at 1"; \
755 goto Fail; \
756 } \
757 Py_DECREF(result)
758
759 CHECK_1_FORMAT("%d", int);
760 CHECK_1_FORMAT("%ld", long);
761 /* The z width modifier was added in Python 2.5. */
762 CHECK_1_FORMAT("%zd", Py_ssize_t);
763
764 /* The u type code was added in Python 2.5. */
765 CHECK_1_FORMAT("%u", unsigned int);
766 CHECK_1_FORMAT("%lu", unsigned long);
767 CHECK_1_FORMAT("%zu", size_t);
768
769 Py_RETURN_NONE;
770
771 Fail:
772 Py_XDECREF(result);
773 return raiseTestError("test_string_from_format", msg);
774
775#undef CHECK_1_FORMAT
776}
777
Georg Brandle9462c72006-08-04 18:03:37 +0000778/* This is here to provide a docstring for test_descr. */
779static PyObject *
780test_with_docstring(PyObject *self)
781{
782 Py_RETURN_NONE;
783}
784
Brett Cannon141534e2008-04-28 03:23:50 +0000785/* To test the format of tracebacks as printed out. */
786static PyObject *
Brett Cannon8dc43032008-04-28 04:50:06 +0000787traceback_print(PyObject *self, PyObject *args)
Brett Cannon141534e2008-04-28 03:23:50 +0000788{
789 PyObject *file;
790 PyObject *traceback;
791 int result;
792
Brett Cannon8dc43032008-04-28 04:50:06 +0000793 if (!PyArg_ParseTuple(args, "OO:traceback_print",
Brett Cannon141534e2008-04-28 03:23:50 +0000794 &traceback, &file))
795 return NULL;
796
797 result = PyTraceBack_Print(traceback, file);
798 if (result < 0)
799 return NULL;
800 Py_RETURN_NONE;
801}
802
Tim Peters9ea17ac2001-02-02 05:57:15 +0000803static PyMethodDef TestMethods[] = {
Fred Drakeacee69f2002-04-01 14:28:58 +0000804 {"raise_exception", raise_exception, METH_VARARGS},
805 {"test_config", (PyCFunction)test_config, METH_NOARGS},
806 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
807 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
808 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
Tim Petersbaefd9e2003-01-28 20:37:45 +0000809 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
Thomas Hellera4ea6032003-04-17 18:55:45 +0000810 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Benjamin Peterson828a7062008-12-27 17:05:29 +0000811 {"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
Neal Norwitz4a53dad2006-03-15 05:43:10 +0000812 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
Tim Peters8931ff12006-05-13 23:28:20 +0000813 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
Georg Brandle9462c72006-08-04 18:03:37 +0000814 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
815 PyDoc_STR("This is a pretty normal docstring.")},
Thomas Heller3457e4b2003-04-24 16:14:27 +0000816
Georg Brandl5f135782006-07-26 08:03:10 +0000817 {"getargs_tuple", getargs_tuple, METH_VARARGS},
Christian Heimesea837932008-02-26 17:23:51 +0000818 {"getargs_keywords", (PyCFunction)getargs_keywords,
819 METH_VARARGS|METH_KEYWORDS},
Georg Brandl7f573f72006-04-13 07:59:30 +0000820 {"getargs_b", getargs_b, METH_VARARGS},
821 {"getargs_B", getargs_B, METH_VARARGS},
822 {"getargs_H", getargs_H, METH_VARARGS},
823 {"getargs_I", getargs_I, METH_VARARGS},
824 {"getargs_k", getargs_k, METH_VARARGS},
825 {"getargs_i", getargs_i, METH_VARARGS},
826 {"getargs_l", getargs_l, METH_VARARGS},
827 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +0000828#ifdef HAVE_LONG_LONG
Georg Brandl7f573f72006-04-13 07:59:30 +0000829 {"getargs_L", getargs_L, METH_VARARGS},
830 {"getargs_K", getargs_K, METH_VARARGS},
831 {"test_longlong_api", test_longlong_api, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +0000832 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Walter Dörwald9ae019b2006-03-18 14:22:26 +0000833 {"codec_incrementalencoder",
834 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
835 {"codec_incrementaldecoder",
836 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +0000837#endif
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000838#ifdef Py_USING_UNICODE
Fred Drakeacee69f2002-04-01 14:28:58 +0000839 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000840#endif
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000841#ifdef WITH_THREAD
Georg Brandl7f573f72006-04-13 07:59:30 +0000842 {"_test_thread_state", test_thread_state, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000843#endif
Brett Cannon8dc43032008-04-28 04:50:06 +0000844 {"traceback_print", traceback_print, METH_VARARGS},
Tim Peters9ea17ac2001-02-02 05:57:15 +0000845 {NULL, NULL} /* sentinel */
846};
847
Thomas Hellera4ea6032003-04-17 18:55:45 +0000848#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
849
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000850typedef struct {
Georg Brandl32a3fb52008-01-21 21:23:15 +0000851 char bool_member;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000852 char byte_member;
853 unsigned char ubyte_member;
854 short short_member;
855 unsigned short ushort_member;
856 int int_member;
857 unsigned int uint_member;
858 long long_member;
859 unsigned long ulong_member;
860 float float_member;
861 double double_member;
Martin v. Löwis6371cd82007-06-09 07:42:52 +0000862#ifdef HAVE_LONG_LONG
863 PY_LONG_LONG longlong_member;
864 unsigned PY_LONG_LONG ulonglong_member;
865#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000866} all_structmembers;
867
868typedef struct {
869 PyObject_HEAD
870 all_structmembers structmembers;
871} test_structmembers;
872
873static struct PyMemberDef test_members[] = {
Georg Brandl32a3fb52008-01-21 21:23:15 +0000874 {"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000875 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
876 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
877 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
878 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
879 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
880 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
881 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
882 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
883 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
884 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
Martin v. Löwis6371cd82007-06-09 07:42:52 +0000885#ifdef HAVE_LONG_LONG
886 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
887 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
888#endif
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000889 {NULL}
890};
891
892
Georg Brandlc09b94e2008-01-21 21:28:32 +0000893static PyObject *
894test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
895{
896 static char *keywords[] = {
897 "T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
898 "T_INT", "T_UINT", "T_LONG", "T_ULONG",
899 "T_FLOAT", "T_DOUBLE",
900#ifdef HAVE_LONG_LONG
Martin v. Löwis6371cd82007-06-09 07:42:52 +0000901 "T_LONGLONG", "T_ULONGLONG",
Georg Brandlc09b94e2008-01-21 21:28:32 +0000902#endif
Martin v. Löwis6371cd82007-06-09 07:42:52 +0000903 NULL};
Georg Brandlc09b94e2008-01-21 21:28:32 +0000904 static char *fmt = "|bbBhHiIlkfd"
905#ifdef HAVE_LONG_LONG
Martin v. Löwis6371cd82007-06-09 07:42:52 +0000906 "LK"
Georg Brandlc09b94e2008-01-21 21:28:32 +0000907#endif
Martin v. Löwis6371cd82007-06-09 07:42:52 +0000908 ;
Georg Brandlc09b94e2008-01-21 21:28:32 +0000909 test_structmembers *ob;
910 ob = PyObject_New(test_structmembers, type);
911 if (ob == NULL)
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000912 return NULL;
913 memset(&ob->structmembers, 0, sizeof(all_structmembers));
Martin v. Löwis6371cd82007-06-09 07:42:52 +0000914 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
Georg Brandlc09b94e2008-01-21 21:28:32 +0000915 &ob->structmembers.bool_member,
916 &ob->structmembers.byte_member,
917 &ob->structmembers.ubyte_member,
918 &ob->structmembers.short_member,
919 &ob->structmembers.ushort_member,
920 &ob->structmembers.int_member,
921 &ob->structmembers.uint_member,
922 &ob->structmembers.long_member,
923 &ob->structmembers.ulong_member,
924 &ob->structmembers.float_member,
925 &ob->structmembers.double_member
926#ifdef HAVE_LONG_LONG
927 , &ob->structmembers.longlong_member,
928 &ob->structmembers.ulonglong_member
929#endif
930 )) {
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000931 Py_DECREF(ob);
932 return NULL;
Georg Brandlc09b94e2008-01-21 21:28:32 +0000933 }
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000934 return (PyObject *)ob;
935}
936
Georg Brandlc09b94e2008-01-21 21:28:32 +0000937static void
938test_structmembers_free(PyObject *ob)
939{
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000940 PyObject_FREE(ob);
941}
942
943static PyTypeObject test_structmembersType = {
Martin v. Löwis68192102007-07-21 06:55:02 +0000944 PyVarObject_HEAD_INIT(NULL, 0)
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000945 "test_structmembersType",
946 sizeof(test_structmembers), /* tp_basicsize */
947 0, /* tp_itemsize */
948 test_structmembers_free, /* destructor tp_dealloc */
949 0, /* tp_print */
950 0, /* tp_getattr */
951 0, /* tp_setattr */
952 0, /* tp_compare */
953 0, /* tp_repr */
954 0, /* tp_as_number */
955 0, /* tp_as_sequence */
956 0, /* tp_as_mapping */
957 0, /* tp_hash */
958 0, /* tp_call */
959 0, /* tp_str */
Georg Brandlc09b94e2008-01-21 21:28:32 +0000960 PyObject_GenericGetAttr, /* tp_getattro */
961 PyObject_GenericSetAttr, /* tp_setattro */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000962 0, /* tp_as_buffer */
963 0, /* tp_flags */
964 "Type containing all structmember types",
965 0, /* traverseproc tp_traverse */
966 0, /* tp_clear */
967 0, /* tp_richcompare */
968 0, /* tp_weaklistoffset */
969 0, /* tp_iter */
970 0, /* tp_iternext */
971 0, /* tp_methods */
Georg Brandlc09b94e2008-01-21 21:28:32 +0000972 test_members, /* tp_members */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000973 0,
974 0,
975 0,
976 0,
977 0,
978 0,
979 0,
980 0,
Georg Brandlc09b94e2008-01-21 21:28:32 +0000981 test_structmembers_new, /* tp_new */
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000982};
983
984
Mark Hammond62b1ab12002-07-23 06:31:15 +0000985PyMODINIT_FUNC
Tim Petersd66595f2001-02-04 03:09:53 +0000986init_testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +0000987{
Fred Drakeacee69f2002-04-01 14:28:58 +0000988 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +0000989
Tim Petersd66595f2001-02-04 03:09:53 +0000990 m = Py_InitModule("_testcapi", TestMethods);
Neal Norwitz1ac754f2006-01-19 06:09:39 +0000991 if (m == NULL)
992 return;
Tim Peters9ea17ac2001-02-02 05:57:15 +0000993
Christian Heimese93237d2007-12-19 02:37:44 +0000994 Py_TYPE(&test_structmembersType)=&PyType_Type;
Martin v. Löwisb5bc5372006-10-27 06:16:31 +0000995 Py_INCREF(&test_structmembersType);
996 PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
997
998 PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
999 PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001000 PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001001 PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
1002 PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001003 PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001004 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
1005 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001006 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001007 PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001008 PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
1009 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
1010 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
1011 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
1012 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
1013 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
Martin v. Löwis6371cd82007-06-09 07:42:52 +00001014 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
1015 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
1016 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
Georg Brandl635af322006-04-13 07:29:18 +00001017 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
Martin v. Löwisb5bc5372006-10-27 06:16:31 +00001018 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
Robert Schuppenies47629022008-07-10 17:13:55 +00001019 PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyInt_FromSsize_t(sizeof(PyGC_Head)));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001020
Tim Petersd66595f2001-02-04 03:09:53 +00001021 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
Fred Drakeacee69f2002-04-01 14:28:58 +00001022 Py_INCREF(TestError);
1023 PyModule_AddObject(m, "error", TestError);
Tim Peters9ea17ac2001-02-02 05:57:15 +00001024}