blob: 9b67b5ec835add83692d78d783d4add7e52ff620 [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
Neal Norwitz8866e0a2007-10-27 04:01:17 +00008#define PY_SSIZE_T_CLEAN
9
Tim Peters9ea17ac2001-02-02 05:57:15 +000010#include "Python.h"
Thomas Wouters89f507f2006-12-13 04:49:30 +000011#include <float.h>
12#include "structmember.h"
Tim Peters9ea17ac2001-02-02 05:57:15 +000013
Mark Hammond8d98d2c2003-04-19 15:41:53 +000014#ifdef WITH_THREAD
15#include "pythread.h"
16#endif /* WITH_THREAD */
Tim Peters9ea17ac2001-02-02 05:57:15 +000017static PyObject *TestError; /* set to exception object in init */
18
Tim Peters91621db2001-06-12 20:10:01 +000019/* Raise TestError with test_name + ": " + msg, and return NULL. */
20
21static PyObject *
22raiseTestError(const char* test_name, const char* msg)
23{
24 char buf[2048];
25
26 if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
27 PyErr_SetString(TestError, "internal error msg too large");
28 else {
Tim Peters885d4572001-11-28 20:27:42 +000029 PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
Tim Peters91621db2001-06-12 20:10:01 +000030 PyErr_SetString(TestError, buf);
31 }
32 return NULL;
33}
34
Martin v. Löwis4f1cd8b2001-07-26 13:41:06 +000035/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
Tim Peters9ea17ac2001-02-02 05:57:15 +000036
37 The ones derived from autoconf on the UNIX-like OSes can be relied
38 upon (in the absence of sloppy cross-compiling), but the Windows
39 platforms have these hardcoded. Better safe than sorry.
40*/
41static PyObject*
Thomas Wouters89f507f2006-12-13 04:49:30 +000042sizeof_error(const char* fatname, const char* typname,
Tim Peters9ea17ac2001-02-02 05:57:15 +000043 int expected, int got)
44{
45 char buf[1024];
Tim Petersbaefd9e2003-01-28 20:37:45 +000046 PyOS_snprintf(buf, sizeof(buf),
Tim Peters885d4572001-11-28 20:27:42 +000047 "%.200s #define == %d but sizeof(%.200s) == %d",
Thomas Wouters89f507f2006-12-13 04:49:30 +000048 fatname, expected, typname, got);
Tim Peters9ea17ac2001-02-02 05:57:15 +000049 PyErr_SetString(TestError, buf);
50 return (PyObject*)NULL;
51}
52
53static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000054test_config(PyObject *self)
Tim Peters9ea17ac2001-02-02 05:57:15 +000055{
Tim Peters9ea17ac2001-02-02 05:57:15 +000056#define CHECK_SIZEOF(FATNAME, TYPE) \
57 if (FATNAME != sizeof(TYPE)) \
58 return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
59
Tim Peters208efe52001-06-26 22:40:47 +000060 CHECK_SIZEOF(SIZEOF_SHORT, short);
Tim Peters9ea17ac2001-02-02 05:57:15 +000061 CHECK_SIZEOF(SIZEOF_INT, int);
62 CHECK_SIZEOF(SIZEOF_LONG, long);
63 CHECK_SIZEOF(SIZEOF_VOID_P, void*);
64 CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
65#ifdef HAVE_LONG_LONG
Martin v. Löwisb9a0f912003-03-29 10:06:18 +000066 CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
Tim Peters9ea17ac2001-02-02 05:57:15 +000067#endif
68
69#undef CHECK_SIZEOF
70
71 Py_INCREF(Py_None);
72 return Py_None;
73}
74
Tim Peters5c4d5bf2001-02-12 22:13:26 +000075static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +000076test_list_api(PyObject *self)
Tim Peters5c4d5bf2001-02-12 22:13:26 +000077{
78 PyObject* list;
79 int i;
Tim Peters5c4d5bf2001-02-12 22:13:26 +000080
81 /* SF bug 132008: PyList_Reverse segfaults */
82#define NLIST 30
83 list = PyList_New(NLIST);
84 if (list == (PyObject*)NULL)
85 return (PyObject*)NULL;
86 /* list = range(NLIST) */
87 for (i = 0; i < NLIST; ++i) {
88 PyObject* anint = PyInt_FromLong(i);
89 if (anint == (PyObject*)NULL) {
90 Py_DECREF(list);
91 return (PyObject*)NULL;
92 }
93 PyList_SET_ITEM(list, i, anint);
94 }
95 /* list.reverse(), via PyList_Reverse() */
96 i = PyList_Reverse(list); /* should not blow up! */
97 if (i != 0) {
98 Py_DECREF(list);
99 return (PyObject*)NULL;
100 }
101 /* Check that list == range(29, -1, -1) now */
102 for (i = 0; i < NLIST; ++i) {
103 PyObject* anint = PyList_GET_ITEM(list, i);
104 if (PyInt_AS_LONG(anint) != NLIST-1-i) {
105 PyErr_SetString(TestError,
106 "test_list_api: reverse screwed up");
107 Py_DECREF(list);
108 return (PyObject*)NULL;
109 }
110 }
111 Py_DECREF(list);
112#undef NLIST
113
114 Py_INCREF(Py_None);
115 return Py_None;
116}
117
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000118static int
119test_dict_inner(int count)
120{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000121 Py_ssize_t pos = 0, iterations = 0;
122 int i;
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000123 PyObject *dict = PyDict_New();
124 PyObject *v, *k;
125
126 if (dict == NULL)
127 return -1;
128
129 for (i = 0; i < count; i++) {
130 v = PyInt_FromLong(i);
131 PyDict_SetItem(dict, v, v);
132 Py_DECREF(v);
133 }
134
135 while (PyDict_Next(dict, &pos, &k, &v)) {
136 PyObject *o;
137 iterations++;
138
139 i = PyInt_AS_LONG(v) + 1;
140 o = PyInt_FromLong(i);
141 if (o == NULL)
142 return -1;
143 if (PyDict_SetItem(dict, k, o) < 0) {
144 Py_DECREF(o);
145 return -1;
146 }
147 Py_DECREF(o);
148 }
149
150 Py_DECREF(dict);
151
152 if (iterations != count) {
153 PyErr_SetString(
154 TestError,
155 "test_dict_iteration: dict iteration went wrong ");
156 return -1;
157 } else {
158 return 0;
159 }
160}
161
162static PyObject*
Fred Drakeacee69f2002-04-01 14:28:58 +0000163test_dict_iteration(PyObject* self)
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000164{
165 int i;
166
Guido van Rossumeb0d9922001-04-13 17:08:15 +0000167 for (i = 0; i < 200; i++) {
168 if (test_dict_inner(i) < 0) {
169 return NULL;
170 }
171 }
172
173 Py_INCREF(Py_None);
174 return Py_None;
175}
176
Tim Peters91621db2001-06-12 20:10:01 +0000177
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000178/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
Tim Petersff70d3c2001-06-14 01:11:03 +0000179 PyLong_{As, From}{Unsigned,}LongLong().
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000180
181 Note that the meat of the test is contained in testcapi_long.h.
182 This is revolting, but delicate code duplication is worse: "almost
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000183 exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000184 dependence on type names makes it impossible to use a parameterized
185 function. A giant macro would be even worse than this. A C++ template
186 would be perfect.
187
188 The "report an error" functions are deliberately not part of the #include
189 file: if the test fails, you can set a breakpoint in the appropriate
190 error function directly, and crawl back from there in the debugger.
191*/
192
193#define UNBIND(X) Py_DECREF(X); (X) = NULL
194
195static PyObject *
196raise_test_long_error(const char* msg)
197{
198 return raiseTestError("test_long_api", msg);
199}
200
201#define TESTNAME test_long_api_inner
202#define TYPENAME long
203#define F_S_TO_PY PyLong_FromLong
204#define F_PY_TO_S PyLong_AsLong
205#define F_U_TO_PY PyLong_FromUnsignedLong
206#define F_PY_TO_U PyLong_AsUnsignedLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000207
208#include "testcapi_long.h"
209
210static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000211test_long_api(PyObject* self)
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000212{
Tim Peters83c9edc2001-06-16 08:10:13 +0000213 return TESTNAME(raise_test_long_error);
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000214}
215
216#undef TESTNAME
217#undef TYPENAME
218#undef F_S_TO_PY
219#undef F_PY_TO_S
220#undef F_U_TO_PY
221#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000222
223#ifdef HAVE_LONG_LONG
Tim Peters91621db2001-06-12 20:10:01 +0000224
225static PyObject *
Tim Petersd1a7da62001-06-13 00:35:57 +0000226raise_test_longlong_error(const char* msg)
227{
228 return raiseTestError("test_longlong_api", msg);
229}
230
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000231#define TESTNAME test_longlong_api_inner
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000232#define TYPENAME PY_LONG_LONG
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000233#define F_S_TO_PY PyLong_FromLongLong
234#define F_PY_TO_S PyLong_AsLongLong
235#define F_U_TO_PY PyLong_FromUnsignedLongLong
236#define F_PY_TO_U PyLong_AsUnsignedLongLong
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000237
238#include "testcapi_long.h"
Tim Petersd1a7da62001-06-13 00:35:57 +0000239
240static PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000241test_longlong_api(PyObject* self, PyObject *args)
Tim Peters91621db2001-06-12 20:10:01 +0000242{
Tim Peters83c9edc2001-06-16 08:10:13 +0000243 return TESTNAME(raise_test_longlong_error);
Tim Peters91621db2001-06-12 20:10:01 +0000244}
245
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000246#undef TESTNAME
247#undef TYPENAME
248#undef F_S_TO_PY
249#undef F_PY_TO_S
250#undef F_U_TO_PY
251#undef F_PY_TO_U
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000252
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000253/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
Tim Petersd38b1c72001-09-30 05:09:37 +0000254 for both long and int arguments. The test may leak a little memory if
255 it fails.
256*/
257static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000258test_L_code(PyObject *self)
Tim Petersd38b1c72001-09-30 05:09:37 +0000259{
260 PyObject *tuple, *num;
Martin v. Löwisb9a0f912003-03-29 10:06:18 +0000261 PY_LONG_LONG value;
Tim Petersd38b1c72001-09-30 05:09:37 +0000262
Tim Petersd38b1c72001-09-30 05:09:37 +0000263 tuple = PyTuple_New(1);
264 if (tuple == NULL)
265 return NULL;
266
267 num = PyLong_FromLong(42);
268 if (num == NULL)
269 return NULL;
270
271 PyTuple_SET_ITEM(tuple, 0, num);
272
273 value = -1;
274 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
275 return NULL;
276 if (value != 42)
277 return raiseTestError("test_L_code",
278 "L code returned wrong value for long 42");
279
280 Py_DECREF(num);
281 num = PyInt_FromLong(42);
282 if (num == NULL)
283 return NULL;
284
285 PyTuple_SET_ITEM(tuple, 0, num);
286
287 value = -1;
288 if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
289 return NULL;
290 if (value != 42)
291 return raiseTestError("test_L_code",
292 "L code returned wrong value for int 42");
293
294 Py_DECREF(tuple);
295 Py_INCREF(Py_None);
296 return Py_None;
297}
298
Tim Peterse7c1f9b2001-06-14 00:55:41 +0000299#endif /* ifdef HAVE_LONG_LONG */
300
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000301/* Test tuple argument processing */
302static PyObject *
303getargs_tuple(PyObject *self, PyObject *args)
304{
305 int a, b, c;
306 if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
307 return NULL;
308 return Py_BuildValue("iii", a, b, c);
309}
310
Thomas Heller3457e4b2003-04-24 16:14:27 +0000311/* Functions to call PyArg_ParseTuple with integer format codes,
312 and return the result.
313*/
Thomas Hellera4ea6032003-04-17 18:55:45 +0000314static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000315getargs_b(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000316{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000317 unsigned char value;
318 if (!PyArg_ParseTuple(args, "b", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000319 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000320 return PyLong_FromUnsignedLong((unsigned long)value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000321}
322
Thomas Heller3457e4b2003-04-24 16:14:27 +0000323static PyObject *
324getargs_B(PyObject *self, PyObject *args)
325{
326 unsigned char value;
327 if (!PyArg_ParseTuple(args, "B", &value))
328 return NULL;
329 return PyLong_FromUnsignedLong((unsigned long)value);
330}
331
332static PyObject *
333getargs_H(PyObject *self, PyObject *args)
334{
335 unsigned short value;
336 if (!PyArg_ParseTuple(args, "H", &value))
337 return NULL;
338 return PyLong_FromUnsignedLong((unsigned long)value);
339}
340
341static PyObject *
342getargs_I(PyObject *self, PyObject *args)
343{
344 unsigned int value;
345 if (!PyArg_ParseTuple(args, "I", &value))
346 return NULL;
347 return PyLong_FromUnsignedLong((unsigned long)value);
348}
349
350static PyObject *
351getargs_k(PyObject *self, PyObject *args)
352{
353 unsigned long value;
354 if (!PyArg_ParseTuple(args, "k", &value))
355 return NULL;
356 return PyLong_FromUnsignedLong(value);
357}
358
359static PyObject *
360getargs_i(PyObject *self, PyObject *args)
361{
362 int value;
363 if (!PyArg_ParseTuple(args, "i", &value))
364 return NULL;
365 return PyLong_FromLong((long)value);
366}
367
Thomas Hellera4ea6032003-04-17 18:55:45 +0000368static PyObject *
369getargs_l(PyObject *self, PyObject *args)
370{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000371 long value;
372 if (!PyArg_ParseTuple(args, "l", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000373 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000374 return PyLong_FromLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000375}
376
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000377static PyObject *
378getargs_n(PyObject *self, PyObject *args)
379{
380 Py_ssize_t value;
381 if (!PyArg_ParseTuple(args, "n", &value))
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000382 return NULL;
383 return PyLong_FromSsize_t(value);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000384}
385
Thomas Hellera4ea6032003-04-17 18:55:45 +0000386#ifdef HAVE_LONG_LONG
Thomas Hellera4ea6032003-04-17 18:55:45 +0000387static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000388getargs_L(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000389{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000390 PY_LONG_LONG value;
391 if (!PyArg_ParseTuple(args, "L", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000392 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000393 return PyLong_FromLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000394}
395
Thomas Hellera4ea6032003-04-17 18:55:45 +0000396static PyObject *
Thomas Heller3457e4b2003-04-24 16:14:27 +0000397getargs_K(PyObject *self, PyObject *args)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000398{
Thomas Heller3457e4b2003-04-24 16:14:27 +0000399 unsigned PY_LONG_LONG value;
400 if (!PyArg_ParseTuple(args, "K", &value))
Thomas Hellera4ea6032003-04-17 18:55:45 +0000401 return NULL;
Thomas Heller3457e4b2003-04-24 16:14:27 +0000402 return PyLong_FromUnsignedLongLong(value);
Thomas Hellera4ea6032003-04-17 18:55:45 +0000403}
404#endif
405
406/* This function not only tests the 'k' getargs code, but also the
407 PyInt_AsUnsignedLongMask() and PyInt_AsUnsignedLongMask() functions. */
408static PyObject *
409test_k_code(PyObject *self)
410{
411 PyObject *tuple, *num;
412 unsigned long value;
413
414 tuple = PyTuple_New(1);
415 if (tuple == NULL)
416 return NULL;
417
Neal Norwitz699cbb72003-04-22 01:28:57 +0000418 /* a number larger than ULONG_MAX even on 64-bit platforms */
Thomas Hellera4ea6032003-04-17 18:55:45 +0000419 num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
420 if (num == NULL)
421 return NULL;
422
423 value = PyInt_AsUnsignedLongMask(num);
Neal Norwitz699cbb72003-04-22 01:28:57 +0000424 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000425 return raiseTestError("test_k_code",
426 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
427
428 PyTuple_SET_ITEM(tuple, 0, num);
429
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000430 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000431 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
432 return NULL;
Neal Norwitz699cbb72003-04-22 01:28:57 +0000433 if (value != ULONG_MAX)
Thomas Hellera4ea6032003-04-17 18:55:45 +0000434 return raiseTestError("test_k_code",
435 "k code returned wrong value for long 0xFFF...FFF");
436
437 Py_DECREF(num);
438 num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
439 if (num == NULL)
440 return NULL;
441
442 value = PyInt_AsUnsignedLongMask(num);
443 if (value != (unsigned long)-0x42)
444 return raiseTestError("test_k_code",
445 "PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
446
447 PyTuple_SET_ITEM(tuple, 0, num);
448
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000449 value = 0;
Thomas Hellera4ea6032003-04-17 18:55:45 +0000450 if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
451 return NULL;
452 if (value != (unsigned long)-0x42)
453 return raiseTestError("test_k_code",
454 "k code returned wrong value for long -0xFFF..000042");
455
456 Py_DECREF(tuple);
457 Py_INCREF(Py_None);
458 return Py_None;
459}
460
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000461
462/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
463 of an error.
464*/
465static PyObject *
Fred Drakeacee69f2002-04-01 14:28:58 +0000466test_u_code(PyObject *self)
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000467{
468 PyObject *tuple, *obj;
469 Py_UNICODE *value;
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000470 Py_ssize_t len;
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000471
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000472 tuple = PyTuple_New(1);
473 if (tuple == NULL)
474 return NULL;
475
476 obj = PyUnicode_Decode("test", strlen("test"),
477 "ascii", NULL);
478 if (obj == NULL)
479 return NULL;
480
481 PyTuple_SET_ITEM(tuple, 0, obj);
482
483 value = 0;
484 if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
485 return NULL;
486 if (value != PyUnicode_AS_UNICODE(obj))
487 return raiseTestError("test_u_code",
488 "u code returned wrong value for u'test'");
489 value = 0;
490 if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
491 return NULL;
492 if (value != PyUnicode_AS_UNICODE(obj) ||
493 len != PyUnicode_GET_SIZE(obj))
494 return raiseTestError("test_u_code",
495 "u# code returned wrong values for u'test'");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000496
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000497 Py_DECREF(tuple);
498 Py_INCREF(Py_None);
499 return Py_None;
500}
501
Guido van Rossumfb67be22007-08-29 18:38:11 +0000502/* Test Z and Z# codes for PyArg_ParseTuple */
503static PyObject *
504test_Z_code(PyObject *self)
505{
506 PyObject *tuple, *obj;
507 Py_UNICODE *value1, *value2;
Neal Norwitz8866e0a2007-10-27 04:01:17 +0000508 Py_ssize_t len1, len2;
Guido van Rossumfb67be22007-08-29 18:38:11 +0000509
510 tuple = PyTuple_New(2);
511 if (tuple == NULL)
512 return NULL;
513
514 obj = PyUnicode_FromString("test");
515 PyTuple_SET_ITEM(tuple, 0, obj);
516 Py_INCREF(Py_None);
517 PyTuple_SET_ITEM(tuple, 1, Py_None);
518
519 /* swap values on purpose */
520 value1 = NULL;
521 value2 = PyUnicode_AS_UNICODE(obj);
522
523 /* Test Z for both values */
524 if (PyArg_ParseTuple(tuple, "ZZ:test_Z_code", &value1, &value2) < 0)
525 return NULL;
526 if (value1 != PyUnicode_AS_UNICODE(obj))
527 return raiseTestError("test_Z_code",
528 "Z code returned wrong value for 'test'");
529 if (value2 != NULL)
530 return raiseTestError("test_Z_code",
531 "Z code returned wrong value for None");
532
533 value1 = NULL;
534 value2 = PyUnicode_AS_UNICODE(obj);
535 len1 = -1;
536 len2 = -1;
537
538 /* Test Z# for both values */
539 if (PyArg_ParseTuple(tuple, "Z#Z#:test_Z_code", &value1, &len1,
540 &value2, &len2) < 0)
541 return NULL;
542 if (value1 != PyUnicode_AS_UNICODE(obj) ||
543 len1 != PyUnicode_GET_SIZE(obj))
544 return raiseTestError("test_Z_code",
545 "Z# code returned wrong values for 'test'");
546 if (value2 != NULL ||
547 len2 != 0)
548 return raiseTestError("test_Z_code",
549 "Z# code returned wrong values for None'");
550
551 Py_DECREF(tuple);
552 Py_RETURN_NONE;
553}
554
Thomas Wouters477c8d52006-05-27 19:21:47 +0000555static PyObject *
556codec_incrementalencoder(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000557{
558 const char *encoding, *errors = NULL;
559 if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
560 &encoding, &errors))
561 return NULL;
562 return PyCodec_IncrementalEncoder(encoding, errors);
563}
564
Thomas Wouters477c8d52006-05-27 19:21:47 +0000565static PyObject *
566codec_incrementaldecoder(PyObject *self, PyObject *args)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000567{
568 const char *encoding, *errors = NULL;
569 if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
570 &encoding, &errors))
571 return NULL;
572 return PyCodec_IncrementalDecoder(encoding, errors);
573}
574
Marc-André Lemburg3e3eacb2002-01-09 16:21:27 +0000575
Tim Peters5b8132f2003-01-31 15:52:05 +0000576/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
Tim Petersbaefd9e2003-01-28 20:37:45 +0000577static PyObject *
578test_long_numbits(PyObject *self)
579{
Tim Peters5b8132f2003-01-31 15:52:05 +0000580 struct triple {
Tim Petersbaefd9e2003-01-28 20:37:45 +0000581 long input;
Tim Peters5b8132f2003-01-31 15:52:05 +0000582 size_t nbits;
583 int sign;
584 } testcases[] = {{0, 0, 0},
585 {1L, 1, 1},
586 {-1L, 1, -1},
587 {2L, 2, 1},
588 {-2L, 2, -1},
589 {3L, 2, 1},
590 {-3L, 2, -1},
591 {4L, 3, 1},
592 {-4L, 3, -1},
593 {0x7fffL, 15, 1}, /* one Python long digit */
594 {-0x7fffL, 15, -1},
595 {0xffffL, 16, 1},
596 {-0xffffL, 16, -1},
597 {0xfffffffL, 28, 1},
598 {-0xfffffffL, 28, -1}};
Tim Petersbaefd9e2003-01-28 20:37:45 +0000599 int i;
600
Tim Peters5b8132f2003-01-31 15:52:05 +0000601 for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
602 PyObject *plong = PyLong_FromLong(testcases[i].input);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000603 size_t nbits = _PyLong_NumBits(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000604 int sign = _PyLong_Sign(plong);
Tim Petersbaefd9e2003-01-28 20:37:45 +0000605
606 Py_DECREF(plong);
Tim Peters5b8132f2003-01-31 15:52:05 +0000607 if (nbits != testcases[i].nbits)
Tim Petersbaefd9e2003-01-28 20:37:45 +0000608 return raiseTestError("test_long_numbits",
Tim Peters5b8132f2003-01-31 15:52:05 +0000609 "wrong result for _PyLong_NumBits");
610 if (sign != testcases[i].sign)
611 return raiseTestError("test_long_numbits",
612 "wrong result for _PyLong_Sign");
Tim Petersbaefd9e2003-01-28 20:37:45 +0000613 }
614 Py_INCREF(Py_None);
615 return Py_None;
616}
617
Thomas Heller519a0422007-11-15 20:48:54 +0000618/* Example passing NULLs to PyObject_Str(NULL). */
Thomas Woutersa9773292006-04-21 09:43:23 +0000619
620static PyObject *
621test_null_strings(PyObject *self)
622{
Thomas Heller519a0422007-11-15 20:48:54 +0000623 PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Str(NULL);
Thomas Woutersa9773292006-04-21 09:43:23 +0000624 PyObject *tuple = PyTuple_Pack(2, o1, o2);
625 Py_XDECREF(o1);
626 Py_XDECREF(o2);
627 return tuple;
628}
629
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000630static PyObject *
631raise_exception(PyObject *self, PyObject *args)
632{
633 PyObject *exc;
634 PyObject *exc_args, *v;
635 int num_args, i;
636
637 if (!PyArg_ParseTuple(args, "Oi:raise_exception",
638 &exc, &num_args))
639 return NULL;
640
641 exc_args = PyTuple_New(num_args);
642 if (exc_args == NULL)
643 return NULL;
644 for (i = 0; i < num_args; ++i) {
645 v = PyInt_FromLong(i);
646 if (v == NULL) {
647 Py_DECREF(exc_args);
648 return NULL;
649 }
650 PyTuple_SET_ITEM(exc_args, i, v);
651 }
652 PyErr_SetObject(exc, exc_args);
Michael W. Hudsonf02bcee2003-08-15 13:03:30 +0000653 Py_DECREF(exc_args);
Jeremy Hyltonede049b2001-09-26 20:01:13 +0000654 return NULL;
655}
Tim Peters91621db2001-06-12 20:10:01 +0000656
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000657#ifdef WITH_THREAD
658
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000659/* test_thread_state spawns a thread of its own, and that thread releases
660 * `thread_done` when it's finished. The driver code has to know when the
661 * thread finishes, because the thread uses a PyObject (the callable) that
662 * may go away when the driver finishes. The former lack of this explicit
663 * synchronization caused rare segfaults, so rare that they were seen only
664 * on a Mac buildbot (although they were possible on any box).
665 */
666static PyThread_type_lock thread_done = NULL;
667
668static void
669_make_call(void *callable)
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000670{
671 PyObject *rc;
672 PyGILState_STATE s = PyGILState_Ensure();
Thomas Wouters89f507f2006-12-13 04:49:30 +0000673 rc = PyObject_CallFunction((PyObject *)callable, "");
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000674 Py_XDECREF(rc);
675 PyGILState_Release(s);
676}
677
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000678/* Same thing, but releases `thread_done` when it returns. This variant
679 * should be called only from threads spawned by test_thread_state().
680 */
681static void
682_make_call_from_thread(void *callable)
683{
684 _make_call(callable);
685 PyThread_release_lock(thread_done);
686}
687
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000688static PyObject *
689test_thread_state(PyObject *self, PyObject *args)
690{
691 PyObject *fn;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000692
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000693 if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
694 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000695
696 /* Ensure Python is set up for threading */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000697 PyEval_InitThreads();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000698 thread_done = PyThread_allocate_lock();
699 if (thread_done == NULL)
700 return PyErr_NoMemory();
701 PyThread_acquire_lock(thread_done, 1);
702
703 /* Start a new thread with our callback. */
704 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000705 /* Make the callback with the thread lock held by this thread */
706 _make_call(fn);
707 /* Do it all again, but this time with the thread-lock released */
708 Py_BEGIN_ALLOW_THREADS
709 _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000710 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000711 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000712
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000713 /* And once more with and without a thread
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000714 XXX - should use a lock and work out exactly what we are trying
715 to test <wink>
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000716 */
717 Py_BEGIN_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000718 PyThread_start_new_thread(_make_call_from_thread, fn);
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000719 _make_call(fn);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000720 PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000721 Py_END_ALLOW_THREADS
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000722
Thomas Wouters89f507f2006-12-13 04:49:30 +0000723 /* Release lock we acquired above. This is required on HP-UX. */
724 PyThread_release_lock(thread_done);
725
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000726 PyThread_free_lock(thread_done);
727 Py_RETURN_NONE;
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000728}
729#endif
730
Neal Norwitzb0d26332007-08-25 00:49:05 +0000731/* Some tests of PyUnicode_FromFormat(). This needs more tests. */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000732static PyObject *
733test_string_from_format(PyObject *self, PyObject *args)
734{
735 PyObject *result;
736 char *msg;
737
738#define CHECK_1_FORMAT(FORMAT, TYPE) \
Neal Norwitzb0d26332007-08-25 00:49:05 +0000739 result = PyUnicode_FromFormat(FORMAT, (TYPE)1); \
Thomas Wouters477c8d52006-05-27 19:21:47 +0000740 if (result == NULL) \
741 return NULL; \
Neal Norwitzb0d26332007-08-25 00:49:05 +0000742 if (strcmp(PyUnicode_AsString(result), "1")) { \
Thomas Wouters477c8d52006-05-27 19:21:47 +0000743 msg = FORMAT " failed at 1"; \
744 goto Fail; \
745 } \
746 Py_DECREF(result)
747
748 CHECK_1_FORMAT("%d", int);
749 CHECK_1_FORMAT("%ld", long);
750 /* The z width modifier was added in Python 2.5. */
751 CHECK_1_FORMAT("%zd", Py_ssize_t);
752
753 /* The u type code was added in Python 2.5. */
754 CHECK_1_FORMAT("%u", unsigned int);
755 CHECK_1_FORMAT("%lu", unsigned long);
756 CHECK_1_FORMAT("%zu", size_t);
757
758 Py_RETURN_NONE;
759
760 Fail:
761 Py_XDECREF(result);
762 return raiseTestError("test_string_from_format", msg);
763
764#undef CHECK_1_FORMAT
765}
766
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000767/* This is here to provide a docstring for test_descr. */
768static PyObject *
769test_with_docstring(PyObject *self)
770{
771 Py_RETURN_NONE;
772}
773
Guido van Rossumddefaf32007-01-14 03:31:43 +0000774#ifdef HAVE_GETTIMEOFDAY
775/* Profiling of integer performance */
776void print_delta(int test, struct timeval *s, struct timeval *e)
777{
778 e->tv_sec -= s->tv_sec;
779 e->tv_usec -= s->tv_usec;
780 if (e->tv_usec < 0) {
781 e->tv_sec -=1;
782 e->tv_usec += 1000000;
783 }
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000784 printf("Test %d: %d.%06ds\n", test, (int)e->tv_sec, (int)e->tv_usec);
Guido van Rossumddefaf32007-01-14 03:31:43 +0000785}
786
787static PyObject *
788profile_int(PyObject *self, PyObject* args)
789{
790 int i, k;
791 struct timeval start, stop;
792 PyObject *single, **multiple, *op1, *result;
793
794 /* Test 1: Allocate and immediately deallocate
795 many small integers */
796 gettimeofday(&start, NULL);
797 for(k=0; k < 20000; k++)
798 for(i=0; i < 1000; i++) {
799 single = PyInt_FromLong(i);
800 Py_DECREF(single);
801 }
802 gettimeofday(&stop, NULL);
803 print_delta(1, &start, &stop);
804
805 /* Test 2: Allocate and immediately deallocate
806 many large integers */
807 gettimeofday(&start, NULL);
808 for(k=0; k < 20000; k++)
809 for(i=0; i < 1000; i++) {
810 single = PyInt_FromLong(i+1000000);
811 Py_DECREF(single);
812 }
813 gettimeofday(&stop, NULL);
814 print_delta(2, &start, &stop);
815
816 /* Test 3: Allocate a few integers, then release
817 them all simultaneously. */
818 multiple = malloc(sizeof(PyObject*) * 1000);
819 gettimeofday(&start, NULL);
820 for(k=0; k < 20000; k++) {
821 for(i=0; i < 1000; i++) {
822 multiple[i] = PyInt_FromLong(i+1000000);
823 }
824 for(i=0; i < 1000; i++) {
825 Py_DECREF(multiple[i]);
826 }
827 }
828 gettimeofday(&stop, NULL);
829 print_delta(3, &start, &stop);
830
831 /* Test 4: Allocate many integers, then release
832 them all simultaneously. */
833 multiple = malloc(sizeof(PyObject*) * 1000000);
834 gettimeofday(&start, NULL);
835 for(k=0; k < 20; k++) {
836 for(i=0; i < 1000000; i++) {
837 multiple[i] = PyInt_FromLong(i+1000000);
838 }
839 for(i=0; i < 1000000; i++) {
840 Py_DECREF(multiple[i]);
841 }
842 }
843 gettimeofday(&stop, NULL);
844 print_delta(4, &start, &stop);
845
846 /* Test 5: Allocate many integers < 32000 */
847 multiple = malloc(sizeof(PyObject*) * 1000000);
848 gettimeofday(&start, NULL);
849 for(k=0; k < 10; k++) {
850 for(i=0; i < 1000000; i++) {
851 multiple[i] = PyInt_FromLong(i+1000);
852 }
853 for(i=0; i < 1000000; i++) {
854 Py_DECREF(multiple[i]);
855 }
856 }
857 gettimeofday(&stop, NULL);
858 print_delta(5, &start, &stop);
859
860 /* Test 6: Perform small int addition */
861 op1 = PyInt_FromLong(1);
862 gettimeofday(&start, NULL);
863 for(i=0; i < 10000000; i++) {
864 result = PyNumber_Add(op1, op1);
865 Py_DECREF(result);
866 }
867 gettimeofday(&stop, NULL);
868 Py_DECREF(op1);
869 print_delta(6, &start, &stop);
870
871 /* Test 7: Perform medium int addition */
872 op1 = PyInt_FromLong(1000);
873 gettimeofday(&start, NULL);
874 for(i=0; i < 10000000; i++) {
875 result = PyNumber_Add(op1, op1);
876 Py_DECREF(result);
877 }
878 gettimeofday(&stop, NULL);
879 Py_DECREF(op1);
880 print_delta(7, &start, &stop);
881
882 Py_INCREF(Py_None);
883 return Py_None;
884}
885#endif
886
Tim Peters9ea17ac2001-02-02 05:57:15 +0000887static PyMethodDef TestMethods[] = {
Fred Drakeacee69f2002-04-01 14:28:58 +0000888 {"raise_exception", raise_exception, METH_VARARGS},
889 {"test_config", (PyCFunction)test_config, METH_NOARGS},
890 {"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
891 {"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
892 {"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
Tim Petersbaefd9e2003-01-28 20:37:45 +0000893 {"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
Thomas Hellera4ea6032003-04-17 18:55:45 +0000894 {"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
Thomas Woutersa9773292006-04-21 09:43:23 +0000895 {"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
Thomas Wouters477c8d52006-05-27 19:21:47 +0000896 {"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000897 {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
898 PyDoc_STR("This is a pretty normal docstring.")},
Thomas Heller3457e4b2003-04-24 16:14:27 +0000899
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000900 {"getargs_tuple", getargs_tuple, METH_VARARGS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000901 {"getargs_b", getargs_b, METH_VARARGS},
902 {"getargs_B", getargs_B, METH_VARARGS},
903 {"getargs_H", getargs_H, METH_VARARGS},
904 {"getargs_I", getargs_I, METH_VARARGS},
905 {"getargs_k", getargs_k, METH_VARARGS},
906 {"getargs_i", getargs_i, METH_VARARGS},
907 {"getargs_l", getargs_l, METH_VARARGS},
908 {"getargs_n", getargs_n, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +0000909#ifdef HAVE_LONG_LONG
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000910 {"getargs_L", getargs_L, METH_VARARGS},
911 {"getargs_K", getargs_K, METH_VARARGS},
912 {"test_longlong_api", test_longlong_api, METH_NOARGS},
Fred Drakeacee69f2002-04-01 14:28:58 +0000913 {"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000914 {"codec_incrementalencoder",
915 (PyCFunction)codec_incrementalencoder, METH_VARARGS},
916 {"codec_incrementaldecoder",
917 (PyCFunction)codec_incrementaldecoder, METH_VARARGS},
Tim Peters91621db2001-06-12 20:10:01 +0000918#endif
Fred Drakeacee69f2002-04-01 14:28:58 +0000919 {"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
Guido van Rossumfb67be22007-08-29 18:38:11 +0000920 {"test_Z_code", (PyCFunction)test_Z_code, METH_NOARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000921#ifdef WITH_THREAD
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000922 {"_test_thread_state", test_thread_state, METH_VARARGS},
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000923#endif
Guido van Rossumddefaf32007-01-14 03:31:43 +0000924#ifdef HAVE_GETTIMEOFDAY
925 {"profile_int", profile_int, METH_NOARGS},
926#endif
Tim Peters9ea17ac2001-02-02 05:57:15 +0000927 {NULL, NULL} /* sentinel */
928};
929
Thomas Hellera4ea6032003-04-17 18:55:45 +0000930#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
931
Thomas Wouters89f507f2006-12-13 04:49:30 +0000932typedef struct {
933 char byte_member;
934 unsigned char ubyte_member;
935 short short_member;
936 unsigned short ushort_member;
937 int int_member;
938 unsigned int uint_member;
939 long long_member;
940 unsigned long ulong_member;
941 float float_member;
942 double double_member;
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000943#ifdef HAVE_LONG_LONG
944 PY_LONG_LONG longlong_member;
945 unsigned PY_LONG_LONG ulonglong_member;
946#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +0000947} all_structmembers;
948
949typedef struct {
950 PyObject_HEAD
951 all_structmembers structmembers;
952} test_structmembers;
953
954static struct PyMemberDef test_members[] = {
955 {"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
956 {"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
957 {"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
958 {"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
959 {"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
960 {"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
961 {"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
962 {"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
963 {"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
964 {"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000965#ifdef HAVE_LONG_LONG
966 {"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
967 {"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
968#endif
Thomas Wouters89f507f2006-12-13 04:49:30 +0000969 {NULL}
970};
971
972
973static PyObject *test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs){
974 static char *keywords[]={"T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT", "T_INT", "T_UINT",
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000975 "T_LONG", "T_ULONG", "T_FLOAT", "T_DOUBLE",
976 #ifdef HAVE_LONG_LONG
977 "T_LONGLONG", "T_ULONGLONG",
978 #endif
979 NULL};
980 static char *fmt="|bBhHiIlkfd"
981 #ifdef HAVE_LONG_LONG
982 "LK"
983 #endif
984 ;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000985 test_structmembers *ob=PyObject_New(test_structmembers, type);
986 if (ob==NULL)
987 return NULL;
988 memset(&ob->structmembers, 0, sizeof(all_structmembers));
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000989 if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
Thomas Wouters89f507f2006-12-13 04:49:30 +0000990 &ob->structmembers.byte_member, &ob->structmembers.ubyte_member,
991 &ob->structmembers.short_member, &ob->structmembers.ushort_member,
992 &ob->structmembers.int_member, &ob->structmembers.uint_member,
993 &ob->structmembers.long_member, &ob->structmembers.ulong_member,
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000994 &ob->structmembers.float_member, &ob->structmembers.double_member
995 #ifdef HAVE_LONG_LONG
996 ,&ob->structmembers.longlong_member, &ob->structmembers.ulonglong_member
997 #endif
998 )){
Thomas Wouters89f507f2006-12-13 04:49:30 +0000999 Py_DECREF(ob);
1000 return NULL;
1001 }
1002 return (PyObject *)ob;
1003}
1004
1005static void test_structmembers_free(PyObject *ob){
1006 PyObject_FREE(ob);
1007}
1008
1009static PyTypeObject test_structmembersType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001010 PyVarObject_HEAD_INIT(NULL, 0)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001011 "test_structmembersType",
1012 sizeof(test_structmembers), /* tp_basicsize */
1013 0, /* tp_itemsize */
1014 test_structmembers_free, /* destructor tp_dealloc */
1015 0, /* tp_print */
1016 0, /* tp_getattr */
1017 0, /* tp_setattr */
1018 0, /* tp_compare */
1019 0, /* tp_repr */
1020 0, /* tp_as_number */
1021 0, /* tp_as_sequence */
1022 0, /* tp_as_mapping */
1023 0, /* tp_hash */
1024 0, /* tp_call */
1025 0, /* tp_str */
1026 PyObject_GenericGetAttr,
1027 PyObject_GenericSetAttr,
1028 0, /* tp_as_buffer */
1029 0, /* tp_flags */
1030 "Type containing all structmember types",
1031 0, /* traverseproc tp_traverse */
1032 0, /* tp_clear */
1033 0, /* tp_richcompare */
1034 0, /* tp_weaklistoffset */
1035 0, /* tp_iter */
1036 0, /* tp_iternext */
1037 0, /* tp_methods */
1038 test_members, /* tp_members */
1039 0,
1040 0,
1041 0,
1042 0,
1043 0,
1044 0,
1045 0,
1046 0,
1047 test_structmembers_new, /* tp_new */
1048};
1049
1050
Mark Hammond62b1ab12002-07-23 06:31:15 +00001051PyMODINIT_FUNC
Tim Petersd66595f2001-02-04 03:09:53 +00001052init_testcapi(void)
Tim Peters9ea17ac2001-02-02 05:57:15 +00001053{
Fred Drakeacee69f2002-04-01 14:28:58 +00001054 PyObject *m;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001055
Tim Petersd66595f2001-02-04 03:09:53 +00001056 m = Py_InitModule("_testcapi", TestMethods);
Neal Norwitz1ac754f2006-01-19 06:09:39 +00001057 if (m == NULL)
1058 return;
Tim Peters9ea17ac2001-02-02 05:57:15 +00001059
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001060 Py_Type(&test_structmembersType)=&PyType_Type;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001061 Py_INCREF(&test_structmembersType);
1062 PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
1063
1064 PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
1065 PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001066 PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001067 PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
1068 PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001069 PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001070 PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
1071 PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001072 PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001073 PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001074 PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
1075 PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
1076 PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
1077 PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
1078 PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
1079 PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001080 PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
1081 PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
1082 PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001083 PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
Thomas Wouters89f507f2006-12-13 04:49:30 +00001084 PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
Thomas Hellera4ea6032003-04-17 18:55:45 +00001085
Tim Petersd66595f2001-02-04 03:09:53 +00001086 TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
Fred Drakeacee69f2002-04-01 14:28:58 +00001087 Py_INCREF(TestError);
1088 PyModule_AddObject(m, "error", TestError);
Tim Peters9ea17ac2001-02-02 05:57:15 +00001089}