2001-02-02 01:57:15 -04:00
|
|
|
/*
|
|
|
|
* C Extension module to test Python interpreter C APIs.
|
|
|
|
*
|
|
|
|
* The 'test_*' functions exported by this module are run as part of the
|
|
|
|
* standard Python regression test, via Lib/test/test_capi.py.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "Python.h"
|
2006-10-27 04:06:52 -03:00
|
|
|
#include <float.h>
|
2006-10-27 03:16:31 -03:00
|
|
|
#include "structmember.h"
|
2009-12-13 17:27:53 -04:00
|
|
|
#include "datetime.h"
|
2001-02-02 01:57:15 -04:00
|
|
|
|
2003-04-19 12:41:53 -03:00
|
|
|
#ifdef WITH_THREAD
|
|
|
|
#include "pythread.h"
|
|
|
|
#endif /* WITH_THREAD */
|
2001-02-02 01:57:15 -04:00
|
|
|
static PyObject *TestError; /* set to exception object in init */
|
|
|
|
|
2001-06-12 17:10:01 -03:00
|
|
|
/* Raise TestError with test_name + ": " + msg, and return NULL. */
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
raiseTestError(const char* test_name, const char* msg)
|
|
|
|
{
|
|
|
|
char buf[2048];
|
|
|
|
|
|
|
|
if (strlen(test_name) + strlen(msg) > sizeof(buf) - 50)
|
|
|
|
PyErr_SetString(TestError, "internal error msg too large");
|
|
|
|
else {
|
2001-11-28 16:27:42 -04:00
|
|
|
PyOS_snprintf(buf, sizeof(buf), "%s: %s", test_name, msg);
|
2001-06-12 17:10:01 -03:00
|
|
|
PyErr_SetString(TestError, buf);
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2001-07-26 10:41:06 -03:00
|
|
|
/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
|
2001-02-02 01:57:15 -04:00
|
|
|
|
|
|
|
The ones derived from autoconf on the UNIX-like OSes can be relied
|
|
|
|
upon (in the absence of sloppy cross-compiling), but the Windows
|
|
|
|
platforms have these hardcoded. Better safe than sorry.
|
|
|
|
*/
|
|
|
|
static PyObject*
|
2006-10-27 03:16:31 -03:00
|
|
|
sizeof_error(const char* fatname, const char* typname,
|
2001-02-02 01:57:15 -04:00
|
|
|
int expected, int got)
|
|
|
|
{
|
|
|
|
char buf[1024];
|
2003-01-28 16:37:45 -04:00
|
|
|
PyOS_snprintf(buf, sizeof(buf),
|
2001-11-28 16:27:42 -04:00
|
|
|
"%.200s #define == %d but sizeof(%.200s) == %d",
|
2006-10-27 03:16:31 -03:00
|
|
|
fatname, expected, typname, got);
|
2001-02-02 01:57:15 -04:00
|
|
|
PyErr_SetString(TestError, buf);
|
|
|
|
return (PyObject*)NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject*
|
2002-04-01 10:28:58 -04:00
|
|
|
test_config(PyObject *self)
|
2001-02-02 01:57:15 -04:00
|
|
|
{
|
|
|
|
#define CHECK_SIZEOF(FATNAME, TYPE) \
|
|
|
|
if (FATNAME != sizeof(TYPE)) \
|
|
|
|
return sizeof_error(#FATNAME, #TYPE, FATNAME, sizeof(TYPE))
|
|
|
|
|
2001-06-26 19:40:47 -03:00
|
|
|
CHECK_SIZEOF(SIZEOF_SHORT, short);
|
2001-02-02 01:57:15 -04:00
|
|
|
CHECK_SIZEOF(SIZEOF_INT, int);
|
|
|
|
CHECK_SIZEOF(SIZEOF_LONG, long);
|
|
|
|
CHECK_SIZEOF(SIZEOF_VOID_P, void*);
|
|
|
|
CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
|
|
|
|
#ifdef HAVE_LONG_LONG
|
2003-03-29 06:06:18 -04:00
|
|
|
CHECK_SIZEOF(SIZEOF_LONG_LONG, PY_LONG_LONG);
|
2001-02-02 01:57:15 -04:00
|
|
|
#endif
|
|
|
|
|
|
|
|
#undef CHECK_SIZEOF
|
|
|
|
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2001-02-12 18:13:26 -04:00
|
|
|
static PyObject*
|
2002-04-01 10:28:58 -04:00
|
|
|
test_list_api(PyObject *self)
|
2001-02-12 18:13:26 -04:00
|
|
|
{
|
|
|
|
PyObject* list;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
/* SF bug 132008: PyList_Reverse segfaults */
|
|
|
|
#define NLIST 30
|
|
|
|
list = PyList_New(NLIST);
|
|
|
|
if (list == (PyObject*)NULL)
|
|
|
|
return (PyObject*)NULL;
|
|
|
|
/* list = range(NLIST) */
|
|
|
|
for (i = 0; i < NLIST; ++i) {
|
|
|
|
PyObject* anint = PyInt_FromLong(i);
|
|
|
|
if (anint == (PyObject*)NULL) {
|
|
|
|
Py_DECREF(list);
|
|
|
|
return (PyObject*)NULL;
|
|
|
|
}
|
|
|
|
PyList_SET_ITEM(list, i, anint);
|
|
|
|
}
|
|
|
|
/* list.reverse(), via PyList_Reverse() */
|
|
|
|
i = PyList_Reverse(list); /* should not blow up! */
|
|
|
|
if (i != 0) {
|
|
|
|
Py_DECREF(list);
|
|
|
|
return (PyObject*)NULL;
|
|
|
|
}
|
|
|
|
/* Check that list == range(29, -1, -1) now */
|
|
|
|
for (i = 0; i < NLIST; ++i) {
|
|
|
|
PyObject* anint = PyList_GET_ITEM(list, i);
|
|
|
|
if (PyInt_AS_LONG(anint) != NLIST-1-i) {
|
|
|
|
PyErr_SetString(TestError,
|
|
|
|
"test_list_api: reverse screwed up");
|
|
|
|
Py_DECREF(list);
|
|
|
|
return (PyObject*)NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Py_DECREF(list);
|
|
|
|
#undef NLIST
|
|
|
|
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2001-04-13 14:08:15 -03:00
|
|
|
static int
|
|
|
|
test_dict_inner(int count)
|
|
|
|
{
|
2006-02-15 13:27:45 -04:00
|
|
|
Py_ssize_t pos = 0, iterations = 0;
|
|
|
|
int i;
|
2001-04-13 14:08:15 -03:00
|
|
|
PyObject *dict = PyDict_New();
|
|
|
|
PyObject *v, *k;
|
|
|
|
|
|
|
|
if (dict == NULL)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
for (i = 0; i < count; i++) {
|
|
|
|
v = PyInt_FromLong(i);
|
|
|
|
PyDict_SetItem(dict, v, v);
|
|
|
|
Py_DECREF(v);
|
|
|
|
}
|
|
|
|
|
|
|
|
while (PyDict_Next(dict, &pos, &k, &v)) {
|
|
|
|
PyObject *o;
|
|
|
|
iterations++;
|
|
|
|
|
|
|
|
i = PyInt_AS_LONG(v) + 1;
|
|
|
|
o = PyInt_FromLong(i);
|
|
|
|
if (o == NULL)
|
|
|
|
return -1;
|
|
|
|
if (PyDict_SetItem(dict, k, o) < 0) {
|
|
|
|
Py_DECREF(o);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
Py_DECREF(o);
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_DECREF(dict);
|
|
|
|
|
|
|
|
if (iterations != count) {
|
|
|
|
PyErr_SetString(
|
|
|
|
TestError,
|
|
|
|
"test_dict_iteration: dict iteration went wrong ");
|
|
|
|
return -1;
|
|
|
|
} else {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject*
|
2002-04-01 10:28:58 -04:00
|
|
|
test_dict_iteration(PyObject* self)
|
2001-04-13 14:08:15 -03:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
|
|
if (test_dict_inner(i) < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2001-06-12 17:10:01 -03:00
|
|
|
|
2008-12-29 21:18:48 -04:00
|
|
|
/* Issue #4701: Check that PyObject_Hash implicitly calls
|
|
|
|
* PyType_Ready if it hasn't already been called
|
|
|
|
*/
|
|
|
|
static PyTypeObject _HashInheritanceTester_Type = {
|
2008-12-31 01:24:37 -04:00
|
|
|
PyObject_HEAD_INIT(NULL)
|
2008-12-29 21:18:48 -04:00
|
|
|
0, /* Number of items for varobject */
|
|
|
|
"hashinheritancetester", /* Name of this type */
|
|
|
|
sizeof(PyObject), /* Basic object size */
|
|
|
|
0, /* Item size for varobject */
|
|
|
|
(destructor)PyObject_Del, /* tp_dealloc */
|
|
|
|
0, /* tp_print */
|
|
|
|
0, /* tp_getattr */
|
|
|
|
0, /* tp_setattr */
|
|
|
|
0, /* tp_compare */
|
|
|
|
0, /* tp_repr */
|
|
|
|
0, /* tp_as_number */
|
|
|
|
0, /* tp_as_sequence */
|
|
|
|
0, /* tp_as_mapping */
|
|
|
|
0, /* tp_hash */
|
|
|
|
0, /* tp_call */
|
|
|
|
0, /* tp_str */
|
|
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
|
|
0, /* tp_setattro */
|
|
|
|
0, /* tp_as_buffer */
|
|
|
|
Py_TPFLAGS_DEFAULT, /* tp_flags */
|
|
|
|
0, /* tp_doc */
|
|
|
|
0, /* tp_traverse */
|
|
|
|
0, /* tp_clear */
|
|
|
|
0, /* tp_richcompare */
|
|
|
|
0, /* tp_weaklistoffset */
|
|
|
|
0, /* tp_iter */
|
|
|
|
0, /* tp_iternext */
|
|
|
|
0, /* tp_methods */
|
|
|
|
0, /* tp_members */
|
|
|
|
0, /* tp_getset */
|
|
|
|
0, /* tp_base */
|
|
|
|
0, /* tp_dict */
|
|
|
|
0, /* tp_descr_get */
|
|
|
|
0, /* tp_descr_set */
|
|
|
|
0, /* tp_dictoffset */
|
|
|
|
0, /* tp_init */
|
|
|
|
0, /* tp_alloc */
|
|
|
|
PyType_GenericNew, /* tp_new */
|
|
|
|
};
|
|
|
|
|
|
|
|
static PyObject*
|
|
|
|
test_lazy_hash_inheritance(PyObject* self)
|
|
|
|
{
|
|
|
|
PyTypeObject *type;
|
|
|
|
PyObject *obj;
|
|
|
|
long hash;
|
|
|
|
|
|
|
|
type = &_HashInheritanceTester_Type;
|
2009-05-05 18:09:21 -03:00
|
|
|
|
2009-05-05 20:00:48 -03:00
|
|
|
if (type->tp_dict != NULL)
|
|
|
|
/* The type has already been initialized. This probably means
|
|
|
|
-R is being used. */
|
|
|
|
Py_RETURN_NONE;
|
2009-05-05 18:09:21 -03:00
|
|
|
|
|
|
|
|
2008-12-29 21:18:48 -04:00
|
|
|
obj = PyObject_New(PyObject, type);
|
|
|
|
if (obj == NULL) {
|
|
|
|
PyErr_Clear();
|
|
|
|
PyErr_SetString(
|
|
|
|
TestError,
|
|
|
|
"test_lazy_hash_inheritance: failed to create object");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (type->tp_dict != NULL) {
|
|
|
|
PyErr_SetString(
|
|
|
|
TestError,
|
|
|
|
"test_lazy_hash_inheritance: type initialised too soon");
|
|
|
|
Py_DECREF(obj);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
hash = PyObject_Hash(obj);
|
|
|
|
if ((hash == -1) && PyErr_Occurred()) {
|
|
|
|
PyErr_Clear();
|
|
|
|
PyErr_SetString(
|
|
|
|
TestError,
|
|
|
|
"test_lazy_hash_inheritance: could not hash object");
|
|
|
|
Py_DECREF(obj);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (type->tp_dict == NULL) {
|
|
|
|
PyErr_SetString(
|
|
|
|
TestError,
|
|
|
|
"test_lazy_hash_inheritance: type not initialised by hash()");
|
|
|
|
Py_DECREF(obj);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (type->tp_hash != PyType_Type.tp_hash) {
|
|
|
|
PyErr_SetString(
|
|
|
|
TestError,
|
|
|
|
"test_lazy_hash_inheritance: unexpected hash function");
|
|
|
|
Py_DECREF(obj);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2009-05-05 20:00:48 -03:00
|
|
|
Py_DECREF(obj);
|
2009-05-05 18:09:21 -03:00
|
|
|
|
2008-12-29 21:18:48 -04:00
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2001-06-13 21:55:41 -03:00
|
|
|
/* Tests of PyLong_{As, From}{Unsigned,}Long(), and (#ifdef HAVE_LONG_LONG)
|
2001-06-13 22:11:03 -03:00
|
|
|
PyLong_{As, From}{Unsigned,}LongLong().
|
2001-06-13 21:55:41 -03:00
|
|
|
|
|
|
|
Note that the meat of the test is contained in testcapi_long.h.
|
|
|
|
This is revolting, but delicate code duplication is worse: "almost
|
2003-03-29 06:06:18 -04:00
|
|
|
exactly the same" code is needed to test PY_LONG_LONG, but the ubiquitous
|
2001-06-13 21:55:41 -03:00
|
|
|
dependence on type names makes it impossible to use a parameterized
|
|
|
|
function. A giant macro would be even worse than this. A C++ template
|
|
|
|
would be perfect.
|
|
|
|
|
|
|
|
The "report an error" functions are deliberately not part of the #include
|
|
|
|
file: if the test fails, you can set a breakpoint in the appropriate
|
|
|
|
error function directly, and crawl back from there in the debugger.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#define UNBIND(X) Py_DECREF(X); (X) = NULL
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
raise_test_long_error(const char* msg)
|
|
|
|
{
|
|
|
|
return raiseTestError("test_long_api", msg);
|
|
|
|
}
|
|
|
|
|
|
|
|
#define TESTNAME test_long_api_inner
|
|
|
|
#define TYPENAME long
|
|
|
|
#define F_S_TO_PY PyLong_FromLong
|
|
|
|
#define F_PY_TO_S PyLong_AsLong
|
|
|
|
#define F_U_TO_PY PyLong_FromUnsignedLong
|
|
|
|
#define F_PY_TO_U PyLong_AsUnsignedLong
|
|
|
|
|
|
|
|
#include "testcapi_long.h"
|
|
|
|
|
|
|
|
static PyObject *
|
2002-04-01 10:28:58 -04:00
|
|
|
test_long_api(PyObject* self)
|
2001-06-13 21:55:41 -03:00
|
|
|
{
|
2001-06-16 05:10:13 -03:00
|
|
|
return TESTNAME(raise_test_long_error);
|
2001-06-13 21:55:41 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
#undef TESTNAME
|
|
|
|
#undef TYPENAME
|
|
|
|
#undef F_S_TO_PY
|
|
|
|
#undef F_PY_TO_S
|
|
|
|
#undef F_U_TO_PY
|
|
|
|
#undef F_PY_TO_U
|
|
|
|
|
|
|
|
#ifdef HAVE_LONG_LONG
|
2001-06-12 17:10:01 -03:00
|
|
|
|
2001-06-12 21:35:57 -03:00
|
|
|
static PyObject *
|
|
|
|
raise_test_longlong_error(const char* msg)
|
|
|
|
{
|
|
|
|
return raiseTestError("test_longlong_api", msg);
|
|
|
|
}
|
|
|
|
|
2001-06-13 21:55:41 -03:00
|
|
|
#define TESTNAME test_longlong_api_inner
|
2003-03-29 06:06:18 -04:00
|
|
|
#define TYPENAME PY_LONG_LONG
|
2001-06-13 21:55:41 -03:00
|
|
|
#define F_S_TO_PY PyLong_FromLongLong
|
|
|
|
#define F_PY_TO_S PyLong_AsLongLong
|
|
|
|
#define F_U_TO_PY PyLong_FromUnsignedLongLong
|
|
|
|
#define F_PY_TO_U PyLong_AsUnsignedLongLong
|
|
|
|
|
|
|
|
#include "testcapi_long.h"
|
2001-06-12 21:35:57 -03:00
|
|
|
|
2001-06-12 17:10:01 -03:00
|
|
|
static PyObject *
|
2006-04-17 22:01:41 -03:00
|
|
|
test_longlong_api(PyObject* self, PyObject *args)
|
2001-06-12 17:10:01 -03:00
|
|
|
{
|
2001-06-16 05:10:13 -03:00
|
|
|
return TESTNAME(raise_test_longlong_error);
|
2001-06-13 21:55:41 -03:00
|
|
|
}
|
2001-06-12 21:35:57 -03:00
|
|
|
|
2001-06-13 21:55:41 -03:00
|
|
|
#undef TESTNAME
|
|
|
|
#undef TYPENAME
|
|
|
|
#undef F_S_TO_PY
|
|
|
|
#undef F_PY_TO_S
|
|
|
|
#undef F_U_TO_PY
|
|
|
|
#undef F_PY_TO_U
|
2001-06-12 17:10:01 -03:00
|
|
|
|
2009-12-21 07:21:25 -04:00
|
|
|
/* Test the PyLong_AsLongAndOverflow API. General conversion to PY_LONG
|
|
|
|
is tested by test_long_api_inner. This test will concentrate on proper
|
|
|
|
handling of overflow.
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
test_long_and_overflow(PyObject *self)
|
|
|
|
{
|
2009-12-21 08:15:48 -04:00
|
|
|
PyObject *num, *one, *temp;
|
2009-12-21 07:21:25 -04:00
|
|
|
long value;
|
|
|
|
int overflow;
|
|
|
|
|
2009-12-21 08:15:48 -04:00
|
|
|
/* Test that overflow is set properly for a large value. */
|
|
|
|
/* num is a number larger than LONG_MAX even on 64-bit platforms */
|
2009-12-21 07:21:25 -04:00
|
|
|
num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
2009-12-21 08:15:48 -04:00
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
2009-12-21 07:21:25 -04:00
|
|
|
if (overflow != 1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"overflow was not set to 1");
|
|
|
|
|
2009-12-21 08:15:48 -04:00
|
|
|
/* Same again, with num = LONG_MAX + 1 */
|
|
|
|
num = PyLong_FromLong(LONG_MAX);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
one = PyLong_FromLong(1L);
|
|
|
|
if (one == NULL) {
|
|
|
|
Py_DECREF(num);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
temp = PyNumber_Add(num, one);
|
|
|
|
Py_DECREF(one);
|
|
|
|
Py_DECREF(num);
|
|
|
|
num = temp;
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
2009-12-21 07:21:25 -04:00
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
2009-12-21 08:15:48 -04:00
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
2009-12-21 07:21:25 -04:00
|
|
|
if (overflow != 1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
2009-12-21 08:15:48 -04:00
|
|
|
"overflow was not set to 1");
|
2009-12-21 07:31:54 -04:00
|
|
|
|
2009-12-21 08:15:48 -04:00
|
|
|
/* Test that overflow is set properly for a large negative value. */
|
|
|
|
/* num is a number smaller than LONG_MIN even on 64-bit platforms */
|
2009-12-21 07:21:25 -04:00
|
|
|
num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
2009-12-21 08:15:48 -04:00
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
2009-12-21 07:21:25 -04:00
|
|
|
if (overflow != -1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"overflow was not set to -1");
|
|
|
|
|
2009-12-21 08:15:48 -04:00
|
|
|
/* Same again, with num = LONG_MIN - 1 */
|
|
|
|
num = PyLong_FromLong(LONG_MIN);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
one = PyLong_FromLong(1L);
|
|
|
|
if (one == NULL) {
|
|
|
|
Py_DECREF(num);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
temp = PyNumber_Subtract(num, one);
|
|
|
|
Py_DECREF(one);
|
|
|
|
Py_DECREF(num);
|
|
|
|
num = temp;
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
2009-12-21 07:21:25 -04:00
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
2009-12-21 08:15:48 -04:00
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
2009-12-21 07:21:25 -04:00
|
|
|
if (overflow != -1)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
2009-12-21 08:15:48 -04:00
|
|
|
"overflow was not set to -1");
|
2009-12-21 07:31:54 -04:00
|
|
|
|
2009-12-21 08:15:48 -04:00
|
|
|
/* Test that overflow is cleared properly for small values. */
|
2009-12-21 07:21:25 -04:00
|
|
|
num = PyLong_FromString("FF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
2009-12-21 08:15:48 -04:00
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != 0xFF)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"expected return value 0xFF");
|
2009-12-21 07:21:25 -04:00
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"overflow was not cleared");
|
|
|
|
|
2009-12-21 08:15:48 -04:00
|
|
|
num = PyLong_FromString("-FF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
2009-12-21 07:21:25 -04:00
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
2009-12-21 08:15:48 -04:00
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -0xFF)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"expected return value 0xFF");
|
2009-12-21 07:21:25 -04:00
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"overflow was set incorrectly");
|
|
|
|
|
2009-12-21 08:15:48 -04:00
|
|
|
num = PyLong_FromLong(LONG_MAX);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
2009-12-21 07:31:54 -04:00
|
|
|
Py_DECREF(num);
|
2009-12-21 08:15:48 -04:00
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != LONG_MAX)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"expected return value LONG_MAX");
|
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"overflow was not cleared");
|
|
|
|
|
|
|
|
num = PyLong_FromLong(LONG_MIN);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != LONG_MIN)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"expected return value LONG_MIN");
|
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_and_overflow",
|
|
|
|
"overflow was not cleared");
|
2009-12-21 07:31:54 -04:00
|
|
|
|
2009-12-21 07:21:25 -04:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2010-01-30 06:08:33 -04:00
|
|
|
/* Test the PyLong_AsLongLongAndOverflow API. General conversion to
|
|
|
|
PY_LONG_LONG is tested by test_long_api_inner. This test will
|
|
|
|
concentrate on proper handling of overflow.
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
test_long_long_and_overflow(PyObject *self)
|
|
|
|
{
|
|
|
|
PyObject *num, *one, *temp;
|
|
|
|
PY_LONG_LONG value;
|
|
|
|
int overflow;
|
|
|
|
|
|
|
|
/* Test that overflow is set properly for a large value. */
|
|
|
|
/* num is a number larger than PY_LLONG_MAX on a typical machine. */
|
|
|
|
num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
|
|
|
if (overflow != 1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was not set to 1");
|
|
|
|
|
|
|
|
/* Same again, with num = PY_LLONG_MAX + 1 */
|
|
|
|
num = PyLong_FromLongLong(PY_LLONG_MAX);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
one = PyLong_FromLong(1L);
|
|
|
|
if (one == NULL) {
|
|
|
|
Py_DECREF(num);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
temp = PyNumber_Add(num, one);
|
|
|
|
Py_DECREF(one);
|
|
|
|
Py_DECREF(num);
|
|
|
|
num = temp;
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
|
|
|
if (overflow != 1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was not set to 1");
|
|
|
|
|
|
|
|
/* Test that overflow is set properly for a large negative value. */
|
|
|
|
/* num is a number smaller than PY_LLONG_MIN on a typical platform */
|
|
|
|
num = PyLong_FromString("-FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
|
|
|
if (overflow != -1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was not set to -1");
|
|
|
|
|
|
|
|
/* Same again, with num = PY_LLONG_MIN - 1 */
|
|
|
|
num = PyLong_FromLongLong(PY_LLONG_MIN);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
one = PyLong_FromLong(1L);
|
|
|
|
if (one == NULL) {
|
|
|
|
Py_DECREF(num);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
temp = PyNumber_Subtract(num, one);
|
|
|
|
Py_DECREF(one);
|
|
|
|
Py_DECREF(num);
|
|
|
|
num = temp;
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"return value was not set to -1");
|
|
|
|
if (overflow != -1)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was not set to -1");
|
|
|
|
|
|
|
|
/* Test that overflow is cleared properly for small values. */
|
|
|
|
num = PyLong_FromString("FF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != 0xFF)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"expected return value 0xFF");
|
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was not cleared");
|
|
|
|
|
|
|
|
num = PyLong_FromString("-FF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != -0xFF)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"expected return value 0xFF");
|
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was set incorrectly");
|
|
|
|
|
|
|
|
num = PyLong_FromLongLong(PY_LLONG_MAX);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 1234;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != PY_LLONG_MAX)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"expected return value PY_LLONG_MAX");
|
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was not cleared");
|
|
|
|
|
|
|
|
num = PyLong_FromLongLong(PY_LLONG_MIN);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
overflow = 0;
|
|
|
|
value = PyLong_AsLongLongAndOverflow(num, &overflow);
|
|
|
|
Py_DECREF(num);
|
|
|
|
if (value == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
if (value != PY_LLONG_MIN)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"expected return value PY_LLONG_MIN");
|
|
|
|
if (overflow != 0)
|
|
|
|
return raiseTestError("test_long_long_and_overflow",
|
|
|
|
"overflow was not cleared");
|
|
|
|
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2003-03-29 06:06:18 -04:00
|
|
|
/* Test the L code for PyArg_ParseTuple. This should deliver a PY_LONG_LONG
|
2001-09-30 02:09:37 -03:00
|
|
|
for both long and int arguments. The test may leak a little memory if
|
|
|
|
it fails.
|
|
|
|
*/
|
|
|
|
static PyObject *
|
2002-04-01 10:28:58 -04:00
|
|
|
test_L_code(PyObject *self)
|
2001-09-30 02:09:37 -03:00
|
|
|
{
|
|
|
|
PyObject *tuple, *num;
|
2003-03-29 06:06:18 -04:00
|
|
|
PY_LONG_LONG value;
|
2001-09-30 02:09:37 -03:00
|
|
|
|
|
|
|
tuple = PyTuple_New(1);
|
|
|
|
if (tuple == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
num = PyLong_FromLong(42);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
PyTuple_SET_ITEM(tuple, 0, num);
|
|
|
|
|
|
|
|
value = -1;
|
|
|
|
if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
|
|
|
|
return NULL;
|
|
|
|
if (value != 42)
|
|
|
|
return raiseTestError("test_L_code",
|
|
|
|
"L code returned wrong value for long 42");
|
|
|
|
|
|
|
|
Py_DECREF(num);
|
|
|
|
num = PyInt_FromLong(42);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
PyTuple_SET_ITEM(tuple, 0, num);
|
|
|
|
|
|
|
|
value = -1;
|
|
|
|
if (PyArg_ParseTuple(tuple, "L:test_L_code", &value) < 0)
|
|
|
|
return NULL;
|
|
|
|
if (value != 42)
|
|
|
|
return raiseTestError("test_L_code",
|
|
|
|
"L code returned wrong value for int 42");
|
|
|
|
|
|
|
|
Py_DECREF(tuple);
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2001-06-13 21:55:41 -03:00
|
|
|
#endif /* ifdef HAVE_LONG_LONG */
|
2001-06-12 17:10:01 -03:00
|
|
|
|
2006-07-26 05:03:10 -03:00
|
|
|
/* Test tuple argument processing */
|
|
|
|
static PyObject *
|
|
|
|
getargs_tuple(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
int a, b, c;
|
|
|
|
if (!PyArg_ParseTuple(args, "i(ii)", &a, &b, &c))
|
|
|
|
return NULL;
|
|
|
|
return Py_BuildValue("iii", a, b, c);
|
|
|
|
}
|
|
|
|
|
2008-02-26 13:23:51 -04:00
|
|
|
/* test PyArg_ParseTupleAndKeywords */
|
|
|
|
static PyObject *getargs_keywords(PyObject *self, PyObject *args, PyObject *kwargs)
|
|
|
|
{
|
|
|
|
static char *keywords[] = {"arg1","arg2","arg3","arg4","arg5", NULL};
|
|
|
|
static char *fmt="(ii)i|(i(ii))(iii)i";
|
|
|
|
int int_args[10]={-1, -1, -1, -1, -1, -1, -1, -1, -1, -1};
|
|
|
|
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
|
|
|
|
&int_args[0], &int_args[1], &int_args[2], &int_args[3], &int_args[4],
|
|
|
|
&int_args[5], &int_args[6], &int_args[7], &int_args[8], &int_args[9]))
|
|
|
|
return NULL;
|
|
|
|
return Py_BuildValue("iiiiiiiiii",
|
|
|
|
int_args[0], int_args[1], int_args[2], int_args[3], int_args[4],
|
|
|
|
int_args[5], int_args[6], int_args[7], int_args[8], int_args[9]);
|
|
|
|
}
|
|
|
|
|
2003-04-24 13:14:27 -03:00
|
|
|
/* Functions to call PyArg_ParseTuple with integer format codes,
|
|
|
|
and return the result.
|
|
|
|
*/
|
2003-04-17 15:55:45 -03:00
|
|
|
static PyObject *
|
2003-04-24 13:14:27 -03:00
|
|
|
getargs_b(PyObject *self, PyObject *args)
|
2003-04-17 15:55:45 -03:00
|
|
|
{
|
2003-04-24 13:14:27 -03:00
|
|
|
unsigned char value;
|
|
|
|
if (!PyArg_ParseTuple(args, "b", &value))
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromUnsignedLong((unsigned long)value);
|
|
|
|
}
|
2003-04-17 15:55:45 -03:00
|
|
|
|
2003-04-24 13:14:27 -03:00
|
|
|
static PyObject *
|
|
|
|
getargs_B(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
unsigned char value;
|
|
|
|
if (!PyArg_ParseTuple(args, "B", &value))
|
2003-04-17 15:55:45 -03:00
|
|
|
return NULL;
|
2003-04-24 13:14:27 -03:00
|
|
|
return PyLong_FromUnsignedLong((unsigned long)value);
|
2003-04-17 15:55:45 -03:00
|
|
|
}
|
|
|
|
|
2009-12-20 11:57:56 -04:00
|
|
|
static PyObject *
|
|
|
|
getargs_h(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
short value;
|
|
|
|
if (!PyArg_ParseTuple(args, "h", &value))
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromLong((long)value);
|
|
|
|
}
|
|
|
|
|
2003-04-17 15:55:45 -03:00
|
|
|
static PyObject *
|
2003-04-24 13:14:27 -03:00
|
|
|
getargs_H(PyObject *self, PyObject *args)
|
2003-04-17 15:55:45 -03:00
|
|
|
{
|
2003-04-24 13:14:27 -03:00
|
|
|
unsigned short value;
|
|
|
|
if (!PyArg_ParseTuple(args, "H", &value))
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromUnsignedLong((unsigned long)value);
|
|
|
|
}
|
2003-04-17 15:55:45 -03:00
|
|
|
|
2003-04-24 13:14:27 -03:00
|
|
|
static PyObject *
|
|
|
|
getargs_I(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
unsigned int value;
|
|
|
|
if (!PyArg_ParseTuple(args, "I", &value))
|
2003-04-17 15:55:45 -03:00
|
|
|
return NULL;
|
2003-04-24 13:14:27 -03:00
|
|
|
return PyLong_FromUnsignedLong((unsigned long)value);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
getargs_k(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
unsigned long value;
|
|
|
|
if (!PyArg_ParseTuple(args, "k", &value))
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromUnsignedLong(value);
|
2003-04-17 15:55:45 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
2003-04-24 13:14:27 -03:00
|
|
|
getargs_i(PyObject *self, PyObject *args)
|
2003-04-17 15:55:45 -03:00
|
|
|
{
|
2003-04-24 13:14:27 -03:00
|
|
|
int value;
|
|
|
|
if (!PyArg_ParseTuple(args, "i", &value))
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromLong((long)value);
|
|
|
|
}
|
2003-04-17 15:55:45 -03:00
|
|
|
|
2003-04-24 13:14:27 -03:00
|
|
|
static PyObject *
|
|
|
|
getargs_l(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
long value;
|
|
|
|
if (!PyArg_ParseTuple(args, "l", &value))
|
2003-04-17 15:55:45 -03:00
|
|
|
return NULL;
|
2003-04-24 13:14:27 -03:00
|
|
|
return PyLong_FromLong(value);
|
2003-04-17 15:55:45 -03:00
|
|
|
}
|
|
|
|
|
2006-04-13 04:59:30 -03:00
|
|
|
static PyObject *
|
|
|
|
getargs_n(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
Py_ssize_t value;
|
|
|
|
if (!PyArg_ParseTuple(args, "n", &value))
|
|
|
|
return NULL;
|
|
|
|
return PyInt_FromSsize_t(value);
|
|
|
|
}
|
|
|
|
|
2003-04-24 13:14:27 -03:00
|
|
|
#ifdef HAVE_LONG_LONG
|
2003-04-17 15:55:45 -03:00
|
|
|
static PyObject *
|
2003-04-24 13:14:27 -03:00
|
|
|
getargs_L(PyObject *self, PyObject *args)
|
2003-04-17 15:55:45 -03:00
|
|
|
{
|
2003-04-24 13:14:27 -03:00
|
|
|
PY_LONG_LONG value;
|
|
|
|
if (!PyArg_ParseTuple(args, "L", &value))
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromLongLong(value);
|
|
|
|
}
|
2003-04-17 15:55:45 -03:00
|
|
|
|
2003-04-24 13:14:27 -03:00
|
|
|
static PyObject *
|
|
|
|
getargs_K(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
unsigned PY_LONG_LONG value;
|
|
|
|
if (!PyArg_ParseTuple(args, "K", &value))
|
2003-04-17 15:55:45 -03:00
|
|
|
return NULL;
|
2003-04-24 13:14:27 -03:00
|
|
|
return PyLong_FromUnsignedLongLong(value);
|
2003-04-17 15:55:45 -03:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* This function not only tests the 'k' getargs code, but also the
|
|
|
|
PyInt_AsUnsignedLongMask() and PyInt_AsUnsignedLongMask() functions. */
|
|
|
|
static PyObject *
|
|
|
|
test_k_code(PyObject *self)
|
|
|
|
{
|
|
|
|
PyObject *tuple, *num;
|
|
|
|
unsigned long value;
|
|
|
|
|
|
|
|
tuple = PyTuple_New(1);
|
|
|
|
if (tuple == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
2003-04-21 22:28:57 -03:00
|
|
|
/* a number larger than ULONG_MAX even on 64-bit platforms */
|
2003-04-17 15:55:45 -03:00
|
|
|
num = PyLong_FromString("FFFFFFFFFFFFFFFFFFFFFFFF", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
value = PyInt_AsUnsignedLongMask(num);
|
2003-04-21 22:28:57 -03:00
|
|
|
if (value != ULONG_MAX)
|
2003-04-17 15:55:45 -03:00
|
|
|
return raiseTestError("test_k_code",
|
|
|
|
"PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
|
|
|
|
|
|
|
|
PyTuple_SET_ITEM(tuple, 0, num);
|
|
|
|
|
2006-04-09 22:03:32 -03:00
|
|
|
value = 0;
|
2003-04-17 15:55:45 -03:00
|
|
|
if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
|
|
|
|
return NULL;
|
2003-04-21 22:28:57 -03:00
|
|
|
if (value != ULONG_MAX)
|
2003-04-17 15:55:45 -03:00
|
|
|
return raiseTestError("test_k_code",
|
|
|
|
"k code returned wrong value for long 0xFFF...FFF");
|
|
|
|
|
|
|
|
Py_DECREF(num);
|
|
|
|
num = PyLong_FromString("-FFFFFFFF000000000000000042", NULL, 16);
|
|
|
|
if (num == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
value = PyInt_AsUnsignedLongMask(num);
|
|
|
|
if (value != (unsigned long)-0x42)
|
|
|
|
return raiseTestError("test_k_code",
|
|
|
|
"PyInt_AsUnsignedLongMask() returned wrong value for long 0xFFF...FFF");
|
|
|
|
|
|
|
|
PyTuple_SET_ITEM(tuple, 0, num);
|
|
|
|
|
2006-04-09 22:03:32 -03:00
|
|
|
value = 0;
|
2003-04-17 15:55:45 -03:00
|
|
|
if (PyArg_ParseTuple(tuple, "k:test_k_code", &value) < 0)
|
|
|
|
return NULL;
|
|
|
|
if (value != (unsigned long)-0x42)
|
|
|
|
return raiseTestError("test_k_code",
|
|
|
|
"k code returned wrong value for long -0xFFF..000042");
|
|
|
|
|
|
|
|
Py_DECREF(tuple);
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2002-01-09 12:21:27 -04:00
|
|
|
#ifdef Py_USING_UNICODE
|
|
|
|
|
2008-12-23 16:12:33 -04:00
|
|
|
static volatile int x;
|
2008-12-23 16:09:28 -04:00
|
|
|
|
2002-01-09 12:21:27 -04:00
|
|
|
/* Test the u and u# codes for PyArg_ParseTuple. May leak memory in case
|
|
|
|
of an error.
|
|
|
|
*/
|
|
|
|
static PyObject *
|
2002-04-01 10:28:58 -04:00
|
|
|
test_u_code(PyObject *self)
|
2002-01-09 12:21:27 -04:00
|
|
|
{
|
|
|
|
PyObject *tuple, *obj;
|
|
|
|
Py_UNICODE *value;
|
|
|
|
int len;
|
|
|
|
|
2008-10-14 18:47:22 -03:00
|
|
|
/* issue4122: Undefined reference to _Py_ascii_whitespace on Windows */
|
|
|
|
/* Just use the macro and check that it compiles */
|
2008-12-23 16:09:28 -04:00
|
|
|
x = Py_UNICODE_ISSPACE(25);
|
2008-10-14 18:47:22 -03:00
|
|
|
|
2002-01-09 12:21:27 -04:00
|
|
|
tuple = PyTuple_New(1);
|
|
|
|
if (tuple == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
obj = PyUnicode_Decode("test", strlen("test"),
|
|
|
|
"ascii", NULL);
|
|
|
|
if (obj == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
PyTuple_SET_ITEM(tuple, 0, obj);
|
|
|
|
|
|
|
|
value = 0;
|
|
|
|
if (PyArg_ParseTuple(tuple, "u:test_u_code", &value) < 0)
|
|
|
|
return NULL;
|
|
|
|
if (value != PyUnicode_AS_UNICODE(obj))
|
|
|
|
return raiseTestError("test_u_code",
|
|
|
|
"u code returned wrong value for u'test'");
|
|
|
|
value = 0;
|
|
|
|
if (PyArg_ParseTuple(tuple, "u#:test_u_code", &value, &len) < 0)
|
|
|
|
return NULL;
|
|
|
|
if (value != PyUnicode_AS_UNICODE(obj) ||
|
|
|
|
len != PyUnicode_GET_SIZE(obj))
|
|
|
|
return raiseTestError("test_u_code",
|
|
|
|
"u# code returned wrong values for u'test'");
|
2003-01-28 16:37:45 -04:00
|
|
|
|
2002-01-09 12:21:27 -04:00
|
|
|
Py_DECREF(tuple);
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2009-03-18 13:07:26 -03:00
|
|
|
static PyObject *
|
|
|
|
test_widechar(PyObject *self)
|
|
|
|
{
|
|
|
|
#if defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
|
|
|
|
const wchar_t wtext[2] = {(wchar_t)0x10ABCDu};
|
|
|
|
size_t wtextlen = 1;
|
|
|
|
#else
|
|
|
|
const wchar_t wtext[3] = {(wchar_t)0xDBEAu, (wchar_t)0xDFCDu};
|
|
|
|
size_t wtextlen = 2;
|
|
|
|
#endif
|
|
|
|
PyObject *wide, *utf8;
|
|
|
|
|
|
|
|
wide = PyUnicode_FromWideChar(wtext, wtextlen);
|
|
|
|
if (wide == NULL)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
utf8 = PyUnicode_FromString("\xf4\x8a\xaf\x8d");
|
|
|
|
if (utf8 == NULL) {
|
|
|
|
Py_DECREF(wide);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (PyUnicode_GET_SIZE(wide) != PyUnicode_GET_SIZE(utf8)) {
|
|
|
|
Py_DECREF(wide);
|
|
|
|
Py_DECREF(utf8);
|
|
|
|
return raiseTestError("test_widechar",
|
|
|
|
"wide string and utf8 string have different length");
|
|
|
|
}
|
|
|
|
if (PyUnicode_Compare(wide, utf8)) {
|
|
|
|
Py_DECREF(wide);
|
|
|
|
Py_DECREF(utf8);
|
|
|
|
if (PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
return raiseTestError("test_widechar",
|
|
|
|
"wide string and utf8 string are differents");
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_DECREF(wide);
|
|
|
|
Py_DECREF(utf8);
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
2008-12-22 16:51:15 -04:00
|
|
|
static PyObject *
|
|
|
|
test_empty_argparse(PyObject *self)
|
|
|
|
{
|
|
|
|
/* Test that formats can begin with '|'. See issue #4720. */
|
|
|
|
PyObject *tuple, *dict = NULL;
|
|
|
|
static char *kwlist[] = {NULL};
|
|
|
|
int result;
|
|
|
|
tuple = PyTuple_New(0);
|
|
|
|
if (!tuple)
|
|
|
|
return NULL;
|
|
|
|
if ((result = PyArg_ParseTuple(tuple, "|:test_empty_argparse")) < 0)
|
|
|
|
goto done;
|
|
|
|
dict = PyDict_New();
|
|
|
|
if (!dict)
|
|
|
|
goto done;
|
|
|
|
result = PyArg_ParseTupleAndKeywords(tuple, dict, "|:test_empty_argparse", kwlist);
|
|
|
|
done:
|
|
|
|
Py_DECREF(tuple);
|
|
|
|
Py_XDECREF(dict);
|
|
|
|
if (result < 0)
|
|
|
|
return NULL;
|
|
|
|
else {
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2006-05-13 20:28:20 -03:00
|
|
|
static PyObject *
|
|
|
|
codec_incrementalencoder(PyObject *self, PyObject *args)
|
2006-03-18 10:22:26 -04:00
|
|
|
{
|
|
|
|
const char *encoding, *errors = NULL;
|
2006-03-20 23:58:41 -04:00
|
|
|
if (!PyArg_ParseTuple(args, "s|s:test_incrementalencoder",
|
2006-03-18 10:22:26 -04:00
|
|
|
&encoding, &errors))
|
|
|
|
return NULL;
|
|
|
|
return PyCodec_IncrementalEncoder(encoding, errors);
|
|
|
|
}
|
|
|
|
|
2006-05-13 20:28:20 -03:00
|
|
|
static PyObject *
|
|
|
|
codec_incrementaldecoder(PyObject *self, PyObject *args)
|
2006-03-18 10:22:26 -04:00
|
|
|
{
|
|
|
|
const char *encoding, *errors = NULL;
|
2006-03-20 23:58:41 -04:00
|
|
|
if (!PyArg_ParseTuple(args, "s|s:test_incrementaldecoder",
|
2006-03-18 10:22:26 -04:00
|
|
|
&encoding, &errors))
|
|
|
|
return NULL;
|
|
|
|
return PyCodec_IncrementalDecoder(encoding, errors);
|
|
|
|
}
|
|
|
|
|
2002-01-09 12:21:27 -04:00
|
|
|
#endif
|
|
|
|
|
2003-01-31 11:52:05 -04:00
|
|
|
/* Simple test of _PyLong_NumBits and _PyLong_Sign. */
|
2003-01-28 16:37:45 -04:00
|
|
|
static PyObject *
|
|
|
|
test_long_numbits(PyObject *self)
|
|
|
|
{
|
2003-01-31 11:52:05 -04:00
|
|
|
struct triple {
|
2003-01-28 16:37:45 -04:00
|
|
|
long input;
|
2003-01-31 11:52:05 -04:00
|
|
|
size_t nbits;
|
|
|
|
int sign;
|
|
|
|
} testcases[] = {{0, 0, 0},
|
|
|
|
{1L, 1, 1},
|
|
|
|
{-1L, 1, -1},
|
|
|
|
{2L, 2, 1},
|
|
|
|
{-2L, 2, -1},
|
|
|
|
{3L, 2, 1},
|
|
|
|
{-3L, 2, -1},
|
|
|
|
{4L, 3, 1},
|
|
|
|
{-4L, 3, -1},
|
|
|
|
{0x7fffL, 15, 1}, /* one Python long digit */
|
|
|
|
{-0x7fffL, 15, -1},
|
|
|
|
{0xffffL, 16, 1},
|
|
|
|
{-0xffffL, 16, -1},
|
|
|
|
{0xfffffffL, 28, 1},
|
|
|
|
{-0xfffffffL, 28, -1}};
|
2003-01-28 16:37:45 -04:00
|
|
|
int i;
|
|
|
|
|
2003-01-31 11:52:05 -04:00
|
|
|
for (i = 0; i < sizeof(testcases) / sizeof(struct triple); ++i) {
|
|
|
|
PyObject *plong = PyLong_FromLong(testcases[i].input);
|
2003-01-28 16:37:45 -04:00
|
|
|
size_t nbits = _PyLong_NumBits(plong);
|
2003-01-31 11:52:05 -04:00
|
|
|
int sign = _PyLong_Sign(plong);
|
2003-01-28 16:37:45 -04:00
|
|
|
|
|
|
|
Py_DECREF(plong);
|
2003-01-31 11:52:05 -04:00
|
|
|
if (nbits != testcases[i].nbits)
|
2003-01-28 16:37:45 -04:00
|
|
|
return raiseTestError("test_long_numbits",
|
2003-01-31 11:52:05 -04:00
|
|
|
"wrong result for _PyLong_NumBits");
|
|
|
|
if (sign != testcases[i].sign)
|
|
|
|
return raiseTestError("test_long_numbits",
|
|
|
|
"wrong result for _PyLong_Sign");
|
2003-01-28 16:37:45 -04:00
|
|
|
}
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
2006-03-15 01:43:10 -04:00
|
|
|
/* Example passing NULLs to PyObject_Str(NULL) and PyObject_Unicode(NULL). */
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
test_null_strings(PyObject *self)
|
|
|
|
{
|
|
|
|
PyObject *o1 = PyObject_Str(NULL), *o2 = PyObject_Unicode(NULL);
|
|
|
|
PyObject *tuple = PyTuple_Pack(2, o1, o2);
|
|
|
|
Py_XDECREF(o1);
|
|
|
|
Py_XDECREF(o2);
|
|
|
|
return tuple;
|
|
|
|
}
|
|
|
|
|
2001-09-26 17:01:13 -03:00
|
|
|
static PyObject *
|
|
|
|
raise_exception(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
PyObject *exc;
|
|
|
|
PyObject *exc_args, *v;
|
|
|
|
int num_args, i;
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "Oi:raise_exception",
|
|
|
|
&exc, &num_args))
|
|
|
|
return NULL;
|
2008-08-23 17:27:43 -03:00
|
|
|
if (!PyExceptionClass_Check(exc)) {
|
|
|
|
PyErr_Format(PyExc_TypeError, "an exception class is required");
|
|
|
|
return NULL;
|
|
|
|
}
|
2001-09-26 17:01:13 -03:00
|
|
|
|
|
|
|
exc_args = PyTuple_New(num_args);
|
|
|
|
if (exc_args == NULL)
|
|
|
|
return NULL;
|
|
|
|
for (i = 0; i < num_args; ++i) {
|
|
|
|
v = PyInt_FromLong(i);
|
|
|
|
if (v == NULL) {
|
|
|
|
Py_DECREF(exc_args);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
PyTuple_SET_ITEM(exc_args, i, v);
|
|
|
|
}
|
|
|
|
PyErr_SetObject(exc, exc_args);
|
2003-08-15 10:03:30 -03:00
|
|
|
Py_DECREF(exc_args);
|
2001-09-26 17:01:13 -03:00
|
|
|
return NULL;
|
|
|
|
}
|
2001-06-12 17:10:01 -03:00
|
|
|
|
2003-04-19 12:41:53 -03:00
|
|
|
#ifdef WITH_THREAD
|
|
|
|
|
2009-12-13 17:27:53 -04:00
|
|
|
static PyObject *
|
|
|
|
test_datetime_capi(PyObject *self, PyObject *args) {
|
|
|
|
if (PyDateTimeAPI) {
|
|
|
|
PyErr_SetString(PyExc_AssertionError,
|
|
|
|
"PyDateTime_CAPI somehow initialized");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
PyDateTime_IMPORT;
|
|
|
|
if (PyDateTimeAPI)
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
else
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2006-03-20 23:58:41 -04:00
|
|
|
/* test_thread_state spawns a thread of its own, and that thread releases
|
|
|
|
* `thread_done` when it's finished. The driver code has to know when the
|
|
|
|
* thread finishes, because the thread uses a PyObject (the callable) that
|
|
|
|
* may go away when the driver finishes. The former lack of this explicit
|
|
|
|
* synchronization caused rare segfaults, so rare that they were seen only
|
|
|
|
* on a Mac buildbot (although they were possible on any box).
|
|
|
|
*/
|
|
|
|
static PyThread_type_lock thread_done = NULL;
|
|
|
|
|
2008-08-23 17:27:43 -03:00
|
|
|
static int
|
2006-03-20 23:58:41 -04:00
|
|
|
_make_call(void *callable)
|
2003-04-19 12:41:53 -03:00
|
|
|
{
|
|
|
|
PyObject *rc;
|
2008-08-23 17:27:43 -03:00
|
|
|
int success;
|
2003-04-19 12:41:53 -03:00
|
|
|
PyGILState_STATE s = PyGILState_Ensure();
|
2006-10-27 03:16:31 -03:00
|
|
|
rc = PyObject_CallFunction((PyObject *)callable, "");
|
2008-08-23 17:27:43 -03:00
|
|
|
success = (rc != NULL);
|
2003-04-19 12:41:53 -03:00
|
|
|
Py_XDECREF(rc);
|
|
|
|
PyGILState_Release(s);
|
2008-08-23 17:27:43 -03:00
|
|
|
return success;
|
2003-04-19 12:41:53 -03:00
|
|
|
}
|
|
|
|
|
2006-03-20 23:58:41 -04:00
|
|
|
/* Same thing, but releases `thread_done` when it returns. This variant
|
|
|
|
* should be called only from threads spawned by test_thread_state().
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
_make_call_from_thread(void *callable)
|
|
|
|
{
|
|
|
|
_make_call(callable);
|
|
|
|
PyThread_release_lock(thread_done);
|
|
|
|
}
|
|
|
|
|
2003-04-19 12:41:53 -03:00
|
|
|
static PyObject *
|
|
|
|
test_thread_state(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
PyObject *fn;
|
2008-08-23 17:27:43 -03:00
|
|
|
int success = 1;
|
2006-03-20 23:58:41 -04:00
|
|
|
|
2003-04-19 12:41:53 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
|
|
|
|
return NULL;
|
2006-03-20 23:58:41 -04:00
|
|
|
|
2008-08-23 17:27:43 -03:00
|
|
|
if (!PyCallable_Check(fn)) {
|
|
|
|
PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
|
|
|
|
fn->ob_type->tp_name);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2006-03-20 23:58:41 -04:00
|
|
|
/* Ensure Python is set up for threading */
|
2003-04-19 12:41:53 -03:00
|
|
|
PyEval_InitThreads();
|
2006-03-20 23:58:41 -04:00
|
|
|
thread_done = PyThread_allocate_lock();
|
|
|
|
if (thread_done == NULL)
|
|
|
|
return PyErr_NoMemory();
|
|
|
|
PyThread_acquire_lock(thread_done, 1);
|
|
|
|
|
|
|
|
/* Start a new thread with our callback. */
|
|
|
|
PyThread_start_new_thread(_make_call_from_thread, fn);
|
2003-04-19 12:41:53 -03:00
|
|
|
/* Make the callback with the thread lock held by this thread */
|
2008-08-23 17:27:43 -03:00
|
|
|
success &= _make_call(fn);
|
2003-04-19 12:41:53 -03:00
|
|
|
/* Do it all again, but this time with the thread-lock released */
|
|
|
|
Py_BEGIN_ALLOW_THREADS
|
2008-08-23 17:27:43 -03:00
|
|
|
success &= _make_call(fn);
|
2006-03-20 23:58:41 -04:00
|
|
|
PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
|
2003-04-19 12:41:53 -03:00
|
|
|
Py_END_ALLOW_THREADS
|
2006-03-20 23:58:41 -04:00
|
|
|
|
2003-04-19 12:41:53 -03:00
|
|
|
/* And once more with and without a thread
|
2006-03-20 23:58:41 -04:00
|
|
|
XXX - should use a lock and work out exactly what we are trying
|
|
|
|
to test <wink>
|
2003-04-19 12:41:53 -03:00
|
|
|
*/
|
|
|
|
Py_BEGIN_ALLOW_THREADS
|
2006-03-20 23:58:41 -04:00
|
|
|
PyThread_start_new_thread(_make_call_from_thread, fn);
|
2008-08-23 17:27:43 -03:00
|
|
|
success &= _make_call(fn);
|
2006-03-20 23:58:41 -04:00
|
|
|
PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
|
2003-04-19 12:41:53 -03:00
|
|
|
Py_END_ALLOW_THREADS
|
2006-03-20 23:58:41 -04:00
|
|
|
|
2006-10-28 18:17:51 -03:00
|
|
|
/* Release lock we acquired above. This is required on HP-UX. */
|
|
|
|
PyThread_release_lock(thread_done);
|
|
|
|
|
2006-03-20 23:58:41 -04:00
|
|
|
PyThread_free_lock(thread_done);
|
2008-08-23 17:27:43 -03:00
|
|
|
if (!success)
|
|
|
|
return NULL;
|
2006-03-20 23:58:41 -04:00
|
|
|
Py_RETURN_NONE;
|
2003-04-19 12:41:53 -03:00
|
|
|
}
|
2009-01-09 17:35:16 -04:00
|
|
|
|
|
|
|
/* test Py_AddPendingCalls using threads */
|
|
|
|
static int _pending_callback(void *arg)
|
|
|
|
{
|
|
|
|
/* we assume the argument is callable object to which we own a reference */
|
|
|
|
PyObject *callable = (PyObject *)arg;
|
|
|
|
PyObject *r = PyObject_CallObject(callable, NULL);
|
|
|
|
Py_DECREF(callable);
|
|
|
|
Py_XDECREF(r);
|
|
|
|
return r != NULL ? 0 : -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* The following requests n callbacks to _pending_callback. It can be
|
|
|
|
* run from any python thread.
|
|
|
|
*/
|
|
|
|
PyObject *pending_threadfunc(PyObject *self, PyObject *arg)
|
|
|
|
{
|
|
|
|
PyObject *callable;
|
|
|
|
int r;
|
|
|
|
if (PyArg_ParseTuple(arg, "O", &callable) == 0)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
/* create the reference for the callbackwhile we hold the lock */
|
|
|
|
Py_INCREF(callable);
|
|
|
|
|
|
|
|
Py_BEGIN_ALLOW_THREADS
|
|
|
|
r = Py_AddPendingCall(&_pending_callback, callable);
|
|
|
|
Py_END_ALLOW_THREADS
|
|
|
|
|
|
|
|
if (r<0) {
|
|
|
|
Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
|
|
|
|
Py_INCREF(Py_False);
|
|
|
|
return Py_False;
|
|
|
|
}
|
|
|
|
Py_INCREF(Py_True);
|
|
|
|
return Py_True;
|
|
|
|
}
|
2003-04-19 12:41:53 -03:00
|
|
|
#endif
|
|
|
|
|
2008-06-09 01:58:54 -03:00
|
|
|
/* Some tests of PyString_FromFormat(). This needs more tests. */
|
2006-05-13 20:28:20 -03:00
|
|
|
static PyObject *
|
|
|
|
test_string_from_format(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
PyObject *result;
|
|
|
|
char *msg;
|
|
|
|
|
|
|
|
#define CHECK_1_FORMAT(FORMAT, TYPE) \
|
2008-06-09 01:58:54 -03:00
|
|
|
result = PyString_FromFormat(FORMAT, (TYPE)1); \
|
2006-05-13 20:28:20 -03:00
|
|
|
if (result == NULL) \
|
|
|
|
return NULL; \
|
2008-06-09 01:58:54 -03:00
|
|
|
if (strcmp(PyString_AsString(result), "1")) { \
|
2006-05-13 20:28:20 -03:00
|
|
|
msg = FORMAT " failed at 1"; \
|
|
|
|
goto Fail; \
|
|
|
|
} \
|
|
|
|
Py_DECREF(result)
|
|
|
|
|
|
|
|
CHECK_1_FORMAT("%d", int);
|
|
|
|
CHECK_1_FORMAT("%ld", long);
|
|
|
|
/* The z width modifier was added in Python 2.5. */
|
|
|
|
CHECK_1_FORMAT("%zd", Py_ssize_t);
|
|
|
|
|
|
|
|
/* The u type code was added in Python 2.5. */
|
|
|
|
CHECK_1_FORMAT("%u", unsigned int);
|
|
|
|
CHECK_1_FORMAT("%lu", unsigned long);
|
|
|
|
CHECK_1_FORMAT("%zu", size_t);
|
|
|
|
|
2009-11-15 12:18:58 -04:00
|
|
|
/* "%lld" and "%llu" support added in Python 2.7. */
|
|
|
|
#ifdef HAVE_LONG_LONG
|
|
|
|
CHECK_1_FORMAT("%llu", unsigned PY_LONG_LONG);
|
|
|
|
CHECK_1_FORMAT("%lld", PY_LONG_LONG);
|
|
|
|
#endif
|
|
|
|
|
2006-05-13 20:28:20 -03:00
|
|
|
Py_RETURN_NONE;
|
|
|
|
|
|
|
|
Fail:
|
|
|
|
Py_XDECREF(result);
|
|
|
|
return raiseTestError("test_string_from_format", msg);
|
|
|
|
|
|
|
|
#undef CHECK_1_FORMAT
|
|
|
|
}
|
|
|
|
|
2006-08-04 15:03:37 -03:00
|
|
|
/* This is here to provide a docstring for test_descr. */
|
|
|
|
static PyObject *
|
|
|
|
test_with_docstring(PyObject *self)
|
|
|
|
{
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
2008-04-28 00:23:50 -03:00
|
|
|
/* To test the format of tracebacks as printed out. */
|
|
|
|
static PyObject *
|
2008-04-28 01:50:06 -03:00
|
|
|
traceback_print(PyObject *self, PyObject *args)
|
2008-04-28 00:23:50 -03:00
|
|
|
{
|
|
|
|
PyObject *file;
|
|
|
|
PyObject *traceback;
|
|
|
|
int result;
|
|
|
|
|
2008-04-28 01:50:06 -03:00
|
|
|
if (!PyArg_ParseTuple(args, "OO:traceback_print",
|
2008-04-28 00:23:50 -03:00
|
|
|
&traceback, &file))
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
result = PyTraceBack_Print(traceback, file);
|
|
|
|
if (result < 0)
|
|
|
|
return NULL;
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
2009-05-08 18:51:06 -03:00
|
|
|
/* To test that the result of PyCode_NewEmpty has the right members. */
|
|
|
|
static PyObject *
|
|
|
|
code_newempty(PyObject *self, PyObject *args)
|
|
|
|
{
|
|
|
|
const char *filename;
|
|
|
|
const char *funcname;
|
|
|
|
int firstlineno;
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "ssi:code_newempty",
|
|
|
|
&filename, &funcname, &firstlineno))
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
|
|
|
|
}
|
|
|
|
|
2009-12-28 04:34:58 -04:00
|
|
|
/* Test PyErr_NewExceptionWithDoc (also exercise PyErr_NewException).
|
|
|
|
Run via Lib/test/test_exceptions.py */
|
|
|
|
static PyObject *
|
|
|
|
make_exception_with_doc(PyObject *self, PyObject *args, PyObject *kwargs)
|
|
|
|
{
|
|
|
|
char *name;
|
|
|
|
char *doc = NULL;
|
|
|
|
PyObject *base = NULL;
|
|
|
|
PyObject *dict = NULL;
|
|
|
|
|
|
|
|
static char *kwlist[] = {"name", "doc", "base", "dict", NULL};
|
|
|
|
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwargs,
|
|
|
|
"s|sOO:make_exception_with_doc", kwlist,
|
|
|
|
&name, &doc, &base, &dict))
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
return PyErr_NewExceptionWithDoc(name, doc, base, dict);
|
|
|
|
}
|
|
|
|
|
2001-02-02 01:57:15 -04:00
|
|
|
static PyMethodDef TestMethods[] = {
|
2002-04-01 10:28:58 -04:00
|
|
|
{"raise_exception", raise_exception, METH_VARARGS},
|
|
|
|
{"test_config", (PyCFunction)test_config, METH_NOARGS},
|
2009-12-13 17:27:53 -04:00
|
|
|
{"test_datetime_capi", test_datetime_capi, METH_NOARGS},
|
2002-04-01 10:28:58 -04:00
|
|
|
{"test_list_api", (PyCFunction)test_list_api, METH_NOARGS},
|
|
|
|
{"test_dict_iteration", (PyCFunction)test_dict_iteration,METH_NOARGS},
|
2008-12-29 21:18:48 -04:00
|
|
|
{"test_lazy_hash_inheritance", (PyCFunction)test_lazy_hash_inheritance,METH_NOARGS},
|
2002-04-01 10:28:58 -04:00
|
|
|
{"test_long_api", (PyCFunction)test_long_api, METH_NOARGS},
|
2009-12-21 08:15:48 -04:00
|
|
|
{"test_long_and_overflow", (PyCFunction)test_long_and_overflow,
|
|
|
|
METH_NOARGS},
|
2003-01-28 16:37:45 -04:00
|
|
|
{"test_long_numbits", (PyCFunction)test_long_numbits, METH_NOARGS},
|
2003-04-17 15:55:45 -03:00
|
|
|
{"test_k_code", (PyCFunction)test_k_code, METH_NOARGS},
|
2008-12-22 16:51:15 -04:00
|
|
|
{"test_empty_argparse", (PyCFunction)test_empty_argparse,METH_NOARGS},
|
2006-03-15 01:43:10 -04:00
|
|
|
{"test_null_strings", (PyCFunction)test_null_strings, METH_NOARGS},
|
2006-05-13 20:28:20 -03:00
|
|
|
{"test_string_from_format", (PyCFunction)test_string_from_format, METH_NOARGS},
|
2006-08-04 15:03:37 -03:00
|
|
|
{"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
|
|
|
|
PyDoc_STR("This is a pretty normal docstring.")},
|
2003-04-24 13:14:27 -03:00
|
|
|
|
2006-07-26 05:03:10 -03:00
|
|
|
{"getargs_tuple", getargs_tuple, METH_VARARGS},
|
2008-02-26 13:23:51 -04:00
|
|
|
{"getargs_keywords", (PyCFunction)getargs_keywords,
|
|
|
|
METH_VARARGS|METH_KEYWORDS},
|
2006-04-13 04:59:30 -03:00
|
|
|
{"getargs_b", getargs_b, METH_VARARGS},
|
|
|
|
{"getargs_B", getargs_B, METH_VARARGS},
|
2009-12-20 11:57:56 -04:00
|
|
|
{"getargs_h", getargs_h, METH_VARARGS},
|
2006-04-13 04:59:30 -03:00
|
|
|
{"getargs_H", getargs_H, METH_VARARGS},
|
|
|
|
{"getargs_I", getargs_I, METH_VARARGS},
|
|
|
|
{"getargs_k", getargs_k, METH_VARARGS},
|
|
|
|
{"getargs_i", getargs_i, METH_VARARGS},
|
|
|
|
{"getargs_l", getargs_l, METH_VARARGS},
|
|
|
|
{"getargs_n", getargs_n, METH_VARARGS},
|
2001-06-12 17:10:01 -03:00
|
|
|
#ifdef HAVE_LONG_LONG
|
2006-04-13 04:59:30 -03:00
|
|
|
{"getargs_L", getargs_L, METH_VARARGS},
|
|
|
|
{"getargs_K", getargs_K, METH_VARARGS},
|
|
|
|
{"test_longlong_api", test_longlong_api, METH_NOARGS},
|
2010-01-30 06:08:33 -04:00
|
|
|
{"test_long_long_and_overflow",
|
|
|
|
(PyCFunction)test_long_long_and_overflow, METH_NOARGS},
|
2002-04-01 10:28:58 -04:00
|
|
|
{"test_L_code", (PyCFunction)test_L_code, METH_NOARGS},
|
2006-03-18 10:22:26 -04:00
|
|
|
{"codec_incrementalencoder",
|
|
|
|
(PyCFunction)codec_incrementalencoder, METH_VARARGS},
|
|
|
|
{"codec_incrementaldecoder",
|
|
|
|
(PyCFunction)codec_incrementaldecoder, METH_VARARGS},
|
2002-01-09 12:21:27 -04:00
|
|
|
#endif
|
|
|
|
#ifdef Py_USING_UNICODE
|
2002-04-01 10:28:58 -04:00
|
|
|
{"test_u_code", (PyCFunction)test_u_code, METH_NOARGS},
|
2009-03-18 13:07:26 -03:00
|
|
|
{"test_widechar", (PyCFunction)test_widechar, METH_NOARGS},
|
2003-04-19 12:41:53 -03:00
|
|
|
#endif
|
|
|
|
#ifdef WITH_THREAD
|
2006-04-13 04:59:30 -03:00
|
|
|
{"_test_thread_state", test_thread_state, METH_VARARGS},
|
2009-01-09 17:35:16 -04:00
|
|
|
{"_pending_threadfunc", pending_threadfunc, METH_VARARGS},
|
2001-06-12 17:10:01 -03:00
|
|
|
#endif
|
2008-04-28 01:50:06 -03:00
|
|
|
{"traceback_print", traceback_print, METH_VARARGS},
|
2009-05-08 18:51:06 -03:00
|
|
|
{"code_newempty", code_newempty, METH_VARARGS},
|
2009-12-28 04:34:58 -04:00
|
|
|
{"make_exception_with_doc", (PyCFunction)make_exception_with_doc,
|
|
|
|
METH_VARARGS | METH_KEYWORDS},
|
2001-02-02 01:57:15 -04:00
|
|
|
{NULL, NULL} /* sentinel */
|
|
|
|
};
|
|
|
|
|
2003-04-17 15:55:45 -03:00
|
|
|
#define AddSym(d, n, f, v) {PyObject *o = f(v); PyDict_SetItemString(d, n, o); Py_DECREF(o);}
|
|
|
|
|
2006-10-27 03:16:31 -03:00
|
|
|
typedef struct {
|
2008-01-21 17:23:15 -04:00
|
|
|
char bool_member;
|
2006-10-27 03:16:31 -03:00
|
|
|
char byte_member;
|
|
|
|
unsigned char ubyte_member;
|
|
|
|
short short_member;
|
|
|
|
unsigned short ushort_member;
|
|
|
|
int int_member;
|
|
|
|
unsigned int uint_member;
|
|
|
|
long long_member;
|
|
|
|
unsigned long ulong_member;
|
|
|
|
float float_member;
|
|
|
|
double double_member;
|
2007-06-09 04:42:52 -03:00
|
|
|
#ifdef HAVE_LONG_LONG
|
|
|
|
PY_LONG_LONG longlong_member;
|
|
|
|
unsigned PY_LONG_LONG ulonglong_member;
|
|
|
|
#endif
|
2006-10-27 03:16:31 -03:00
|
|
|
} all_structmembers;
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
PyObject_HEAD
|
|
|
|
all_structmembers structmembers;
|
|
|
|
} test_structmembers;
|
|
|
|
|
|
|
|
static struct PyMemberDef test_members[] = {
|
2008-01-21 17:23:15 -04:00
|
|
|
{"T_BOOL", T_BOOL, offsetof(test_structmembers, structmembers.bool_member), 0, NULL},
|
2006-10-27 03:16:31 -03:00
|
|
|
{"T_BYTE", T_BYTE, offsetof(test_structmembers, structmembers.byte_member), 0, NULL},
|
|
|
|
{"T_UBYTE", T_UBYTE, offsetof(test_structmembers, structmembers.ubyte_member), 0, NULL},
|
|
|
|
{"T_SHORT", T_SHORT, offsetof(test_structmembers, structmembers.short_member), 0, NULL},
|
|
|
|
{"T_USHORT", T_USHORT, offsetof(test_structmembers, structmembers.ushort_member), 0, NULL},
|
|
|
|
{"T_INT", T_INT, offsetof(test_structmembers, structmembers.int_member), 0, NULL},
|
|
|
|
{"T_UINT", T_UINT, offsetof(test_structmembers, structmembers.uint_member), 0, NULL},
|
|
|
|
{"T_LONG", T_LONG, offsetof(test_structmembers, structmembers.long_member), 0, NULL},
|
|
|
|
{"T_ULONG", T_ULONG, offsetof(test_structmembers, structmembers.ulong_member), 0, NULL},
|
|
|
|
{"T_FLOAT", T_FLOAT, offsetof(test_structmembers, structmembers.float_member), 0, NULL},
|
|
|
|
{"T_DOUBLE", T_DOUBLE, offsetof(test_structmembers, structmembers.double_member), 0, NULL},
|
2007-06-09 04:42:52 -03:00
|
|
|
#ifdef HAVE_LONG_LONG
|
|
|
|
{"T_LONGLONG", T_LONGLONG, offsetof(test_structmembers, structmembers.longlong_member), 0, NULL},
|
|
|
|
{"T_ULONGLONG", T_ULONGLONG, offsetof(test_structmembers, structmembers.ulonglong_member), 0, NULL},
|
|
|
|
#endif
|
2006-10-27 03:16:31 -03:00
|
|
|
{NULL}
|
|
|
|
};
|
|
|
|
|
|
|
|
|
2008-01-21 17:28:32 -04:00
|
|
|
static PyObject *
|
|
|
|
test_structmembers_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
|
|
|
|
{
|
|
|
|
static char *keywords[] = {
|
|
|
|
"T_BOOL", "T_BYTE", "T_UBYTE", "T_SHORT", "T_USHORT",
|
|
|
|
"T_INT", "T_UINT", "T_LONG", "T_ULONG",
|
|
|
|
"T_FLOAT", "T_DOUBLE",
|
|
|
|
#ifdef HAVE_LONG_LONG
|
2007-06-09 04:42:52 -03:00
|
|
|
"T_LONGLONG", "T_ULONGLONG",
|
2008-01-21 17:28:32 -04:00
|
|
|
#endif
|
2007-06-09 04:42:52 -03:00
|
|
|
NULL};
|
2008-01-21 17:28:32 -04:00
|
|
|
static char *fmt = "|bbBhHiIlkfd"
|
|
|
|
#ifdef HAVE_LONG_LONG
|
2007-06-09 04:42:52 -03:00
|
|
|
"LK"
|
2008-01-21 17:28:32 -04:00
|
|
|
#endif
|
2007-06-09 04:42:52 -03:00
|
|
|
;
|
2008-01-21 17:28:32 -04:00
|
|
|
test_structmembers *ob;
|
|
|
|
ob = PyObject_New(test_structmembers, type);
|
|
|
|
if (ob == NULL)
|
2006-10-27 03:16:31 -03:00
|
|
|
return NULL;
|
|
|
|
memset(&ob->structmembers, 0, sizeof(all_structmembers));
|
2007-06-09 04:42:52 -03:00
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwargs, fmt, keywords,
|
2008-01-21 17:28:32 -04:00
|
|
|
&ob->structmembers.bool_member,
|
|
|
|
&ob->structmembers.byte_member,
|
|
|
|
&ob->structmembers.ubyte_member,
|
|
|
|
&ob->structmembers.short_member,
|
|
|
|
&ob->structmembers.ushort_member,
|
|
|
|
&ob->structmembers.int_member,
|
|
|
|
&ob->structmembers.uint_member,
|
|
|
|
&ob->structmembers.long_member,
|
|
|
|
&ob->structmembers.ulong_member,
|
|
|
|
&ob->structmembers.float_member,
|
|
|
|
&ob->structmembers.double_member
|
|
|
|
#ifdef HAVE_LONG_LONG
|
|
|
|
, &ob->structmembers.longlong_member,
|
|
|
|
&ob->structmembers.ulonglong_member
|
|
|
|
#endif
|
|
|
|
)) {
|
2006-10-27 03:16:31 -03:00
|
|
|
Py_DECREF(ob);
|
|
|
|
return NULL;
|
2008-01-21 17:28:32 -04:00
|
|
|
}
|
2006-10-27 03:16:31 -03:00
|
|
|
return (PyObject *)ob;
|
|
|
|
}
|
|
|
|
|
2008-01-21 17:28:32 -04:00
|
|
|
static void
|
|
|
|
test_structmembers_free(PyObject *ob)
|
|
|
|
{
|
2006-10-27 03:16:31 -03:00
|
|
|
PyObject_FREE(ob);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyTypeObject test_structmembersType = {
|
2007-07-21 03:55:02 -03:00
|
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
2006-10-27 03:16:31 -03:00
|
|
|
"test_structmembersType",
|
|
|
|
sizeof(test_structmembers), /* tp_basicsize */
|
|
|
|
0, /* tp_itemsize */
|
|
|
|
test_structmembers_free, /* destructor tp_dealloc */
|
|
|
|
0, /* tp_print */
|
|
|
|
0, /* tp_getattr */
|
|
|
|
0, /* tp_setattr */
|
|
|
|
0, /* tp_compare */
|
|
|
|
0, /* tp_repr */
|
|
|
|
0, /* tp_as_number */
|
|
|
|
0, /* tp_as_sequence */
|
|
|
|
0, /* tp_as_mapping */
|
|
|
|
0, /* tp_hash */
|
|
|
|
0, /* tp_call */
|
|
|
|
0, /* tp_str */
|
2008-01-21 17:28:32 -04:00
|
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
|
|
PyObject_GenericSetAttr, /* tp_setattro */
|
2006-10-27 03:16:31 -03:00
|
|
|
0, /* tp_as_buffer */
|
|
|
|
0, /* tp_flags */
|
|
|
|
"Type containing all structmember types",
|
|
|
|
0, /* traverseproc tp_traverse */
|
|
|
|
0, /* tp_clear */
|
|
|
|
0, /* tp_richcompare */
|
|
|
|
0, /* tp_weaklistoffset */
|
|
|
|
0, /* tp_iter */
|
|
|
|
0, /* tp_iternext */
|
|
|
|
0, /* tp_methods */
|
2008-01-21 17:28:32 -04:00
|
|
|
test_members, /* tp_members */
|
2006-10-27 03:16:31 -03:00
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0,
|
2008-01-21 17:28:32 -04:00
|
|
|
test_structmembers_new, /* tp_new */
|
2006-10-27 03:16:31 -03:00
|
|
|
};
|
|
|
|
|
|
|
|
|
2002-07-23 03:31:15 -03:00
|
|
|
PyMODINIT_FUNC
|
2001-02-03 23:09:53 -04:00
|
|
|
init_testcapi(void)
|
2001-02-02 01:57:15 -04:00
|
|
|
{
|
2002-04-01 10:28:58 -04:00
|
|
|
PyObject *m;
|
2001-02-02 01:57:15 -04:00
|
|
|
|
2001-02-03 23:09:53 -04:00
|
|
|
m = Py_InitModule("_testcapi", TestMethods);
|
2006-01-19 02:09:39 -04:00
|
|
|
if (m == NULL)
|
|
|
|
return;
|
2001-02-02 01:57:15 -04:00
|
|
|
|
2008-12-31 01:24:37 -04:00
|
|
|
Py_TYPE(&_HashInheritanceTester_Type)=&PyType_Type;
|
2008-12-31 01:47:19 -04:00
|
|
|
|
2007-12-18 22:37:44 -04:00
|
|
|
Py_TYPE(&test_structmembersType)=&PyType_Type;
|
2006-10-27 03:16:31 -03:00
|
|
|
Py_INCREF(&test_structmembersType);
|
|
|
|
PyModule_AddObject(m, "test_structmembersType", (PyObject *)&test_structmembersType);
|
|
|
|
|
|
|
|
PyModule_AddObject(m, "CHAR_MAX", PyInt_FromLong(CHAR_MAX));
|
|
|
|
PyModule_AddObject(m, "CHAR_MIN", PyInt_FromLong(CHAR_MIN));
|
2003-04-17 15:55:45 -03:00
|
|
|
PyModule_AddObject(m, "UCHAR_MAX", PyInt_FromLong(UCHAR_MAX));
|
2006-10-27 03:16:31 -03:00
|
|
|
PyModule_AddObject(m, "SHRT_MAX", PyInt_FromLong(SHRT_MAX));
|
|
|
|
PyModule_AddObject(m, "SHRT_MIN", PyInt_FromLong(SHRT_MIN));
|
2003-04-17 15:55:45 -03:00
|
|
|
PyModule_AddObject(m, "USHRT_MAX", PyInt_FromLong(USHRT_MAX));
|
2006-10-27 03:16:31 -03:00
|
|
|
PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
|
|
|
|
PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
|
2003-04-17 15:55:45 -03:00
|
|
|
PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
|
|
|
|
PyModule_AddObject(m, "LONG_MAX", PyInt_FromLong(LONG_MAX));
|
2006-10-27 03:16:31 -03:00
|
|
|
PyModule_AddObject(m, "LONG_MIN", PyInt_FromLong(LONG_MIN));
|
|
|
|
PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
|
|
|
|
PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
|
|
|
|
PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
|
|
|
|
PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
|
|
|
|
PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
|
2007-06-09 04:42:52 -03:00
|
|
|
PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(PY_LLONG_MAX));
|
|
|
|
PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(PY_LLONG_MIN));
|
|
|
|
PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(PY_ULLONG_MAX));
|
2006-04-13 04:29:18 -03:00
|
|
|
PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyInt_FromSsize_t(PY_SSIZE_T_MAX));
|
2006-10-27 03:16:31 -03:00
|
|
|
PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyInt_FromSsize_t(PY_SSIZE_T_MIN));
|
2008-07-10 14:13:55 -03:00
|
|
|
PyModule_AddObject(m, "SIZEOF_PYGC_HEAD", PyInt_FromSsize_t(sizeof(PyGC_Head)));
|
2003-04-17 15:55:45 -03:00
|
|
|
|
2001-02-03 23:09:53 -04:00
|
|
|
TestError = PyErr_NewException("_testcapi.error", NULL, NULL);
|
2002-04-01 10:28:58 -04:00
|
|
|
Py_INCREF(TestError);
|
|
|
|
PyModule_AddObject(m, "error", TestError);
|
2001-02-02 01:57:15 -04:00
|
|
|
}
|