mirror of
https://github.com/python/cpython.git
synced 2024-12-06 00:05:32 +08:00
208 lines
6.8 KiB
C
208 lines
6.8 KiB
C
/* Poor-man's template. Macros used:
|
|
TESTNAME name of the test (like test_long_api_inner)
|
|
TYPENAME the signed type (like long)
|
|
F_S_TO_PY convert signed to pylong; TYPENAME -> PyObject*
|
|
F_PY_TO_S convert pylong to signed; PyObject* -> TYPENAME
|
|
F_U_TO_PY convert unsigned to pylong; unsigned TYPENAME -> PyObject*
|
|
F_PY_TO_U convert pylong to unsigned; PyObject* -> unsigned TYPENAME
|
|
*/
|
|
|
|
static PyObject *
|
|
TESTNAME(PyObject *error(const char*))
|
|
{
|
|
const int NBITS = sizeof(TYPENAME) * 8;
|
|
unsigned TYPENAME base;
|
|
PyObject *pyresult;
|
|
int i;
|
|
|
|
/* Note: This test lets PyObjects leak if an error is raised. Since
|
|
an error should never be raised, leaks are impossible <wink>. */
|
|
|
|
/* Test native -> PyLong -> native roundtrip identity.
|
|
* Generate all powers of 2, and test them and their negations,
|
|
* plus the numbers +-1 off from them.
|
|
*/
|
|
base = 1;
|
|
for (i = 0;
|
|
i < NBITS + 1; /* on last, base overflows to 0 */
|
|
++i, base <<= 1)
|
|
{
|
|
int j;
|
|
for (j = 0; j < 6; ++j) {
|
|
TYPENAME in, out;
|
|
unsigned TYPENAME uin, uout;
|
|
|
|
/* For 0, 1, 2 use base; for 3, 4, 5 use -base */
|
|
uin = j < 3 ? base : 0U - base;
|
|
|
|
/* For 0 & 3, subtract 1.
|
|
* For 1 & 4, leave alone.
|
|
* For 2 & 5, add 1.
|
|
*/
|
|
uin += (unsigned TYPENAME)(TYPENAME)(j % 3 - 1);
|
|
|
|
pyresult = F_U_TO_PY(uin);
|
|
if (pyresult == NULL)
|
|
return error(
|
|
"unsigned unexpected null result");
|
|
|
|
uout = F_PY_TO_U(pyresult);
|
|
if (uout == (unsigned TYPENAME)-1 && PyErr_Occurred())
|
|
return error(
|
|
"unsigned unexpected -1 result");
|
|
if (uout != uin)
|
|
return error(
|
|
"unsigned output != input");
|
|
UNBIND(pyresult);
|
|
|
|
in = (TYPENAME)uin;
|
|
pyresult = F_S_TO_PY(in);
|
|
if (pyresult == NULL)
|
|
return error(
|
|
"signed unexpected null result");
|
|
|
|
out = F_PY_TO_S(pyresult);
|
|
if (out == (TYPENAME)-1 && PyErr_Occurred())
|
|
return error(
|
|
"signed unexpected -1 result");
|
|
if (out != in)
|
|
return error(
|
|
"signed output != input");
|
|
UNBIND(pyresult);
|
|
}
|
|
}
|
|
|
|
/* Overflow tests. The loop above ensured that all limit cases that
|
|
* should not overflow don't overflow, so all we need to do here is
|
|
* provoke one-over-the-limit cases (not exhaustive, but sharp).
|
|
*/
|
|
{
|
|
PyObject *one, *x, *y;
|
|
TYPENAME out;
|
|
unsigned TYPENAME uout;
|
|
|
|
one = PyLong_FromLong(1);
|
|
if (one == NULL)
|
|
return error(
|
|
"unexpected NULL from PyLong_FromLong");
|
|
|
|
/* Unsigned complains about -1? */
|
|
x = PyNumber_Negative(one);
|
|
if (x == NULL)
|
|
return error(
|
|
"unexpected NULL from PyNumber_Negative");
|
|
|
|
uout = F_PY_TO_U(x);
|
|
if (uout != (unsigned TYPENAME)-1 || !PyErr_Occurred())
|
|
return error(
|
|
"PyLong_AsUnsignedXXX(-1) didn't complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_OverflowError))
|
|
return error(
|
|
"PyLong_AsUnsignedXXX(-1) raised "
|
|
"something other than OverflowError");
|
|
PyErr_Clear();
|
|
UNBIND(x);
|
|
|
|
/* Unsigned complains about 2**NBITS? */
|
|
y = PyLong_FromLong((long)NBITS);
|
|
if (y == NULL)
|
|
return error(
|
|
"unexpected NULL from PyLong_FromLong");
|
|
|
|
x = PyNumber_Lshift(one, y); /* 1L << NBITS, == 2**NBITS */
|
|
UNBIND(y);
|
|
if (x == NULL)
|
|
return error(
|
|
"unexpected NULL from PyNumber_Lshift");
|
|
|
|
uout = F_PY_TO_U(x);
|
|
if (uout != (unsigned TYPENAME)-1 || !PyErr_Occurred())
|
|
return error(
|
|
"PyLong_AsUnsignedXXX(2**NBITS) didn't "
|
|
"complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_OverflowError))
|
|
return error(
|
|
"PyLong_AsUnsignedXXX(2**NBITS) raised "
|
|
"something other than OverflowError");
|
|
PyErr_Clear();
|
|
|
|
/* Signed complains about 2**(NBITS-1)?
|
|
x still has 2**NBITS. */
|
|
y = PyNumber_Rshift(x, one); /* 2**(NBITS-1) */
|
|
UNBIND(x);
|
|
if (y == NULL)
|
|
return error(
|
|
"unexpected NULL from PyNumber_Rshift");
|
|
|
|
out = F_PY_TO_S(y);
|
|
if (out != (TYPENAME)-1 || !PyErr_Occurred())
|
|
return error(
|
|
"PyLong_AsXXX(2**(NBITS-1)) didn't "
|
|
"complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_OverflowError))
|
|
return error(
|
|
"PyLong_AsXXX(2**(NBITS-1)) raised "
|
|
"something other than OverflowError");
|
|
PyErr_Clear();
|
|
|
|
/* Signed complains about -2**(NBITS-1)-1?;
|
|
y still has 2**(NBITS-1). */
|
|
x = PyNumber_Negative(y); /* -(2**(NBITS-1)) */
|
|
UNBIND(y);
|
|
if (x == NULL)
|
|
return error(
|
|
"unexpected NULL from PyNumber_Negative");
|
|
|
|
y = PyNumber_Subtract(x, one); /* -(2**(NBITS-1))-1 */
|
|
UNBIND(x);
|
|
if (y == NULL)
|
|
return error(
|
|
"unexpected NULL from PyNumber_Subtract");
|
|
|
|
out = F_PY_TO_S(y);
|
|
if (out != (TYPENAME)-1 || !PyErr_Occurred())
|
|
return error(
|
|
"PyLong_AsXXX(-2**(NBITS-1)-1) didn't "
|
|
"complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_OverflowError))
|
|
return error(
|
|
"PyLong_AsXXX(-2**(NBITS-1)-1) raised "
|
|
"something other than OverflowError");
|
|
PyErr_Clear();
|
|
UNBIND(y);
|
|
|
|
Py_XDECREF(x);
|
|
Py_XDECREF(y);
|
|
Py_DECREF(one);
|
|
}
|
|
|
|
/* Test F_PY_TO_{S,U} on non-pylong input. This should raise a TypeError. */
|
|
{
|
|
TYPENAME out;
|
|
unsigned TYPENAME uout;
|
|
|
|
Py_INCREF(Py_None);
|
|
|
|
out = F_PY_TO_S(Py_None);
|
|
if (out != (TYPENAME)-1 || !PyErr_Occurred())
|
|
return error("PyLong_AsXXX(None) didn't complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_TypeError))
|
|
return error("PyLong_AsXXX(None) raised "
|
|
"something other than TypeError");
|
|
PyErr_Clear();
|
|
|
|
uout = F_PY_TO_U(Py_None);
|
|
if (uout != (unsigned TYPENAME)-1 || !PyErr_Occurred())
|
|
return error("PyLong_AsXXX(None) didn't complain");
|
|
if (!PyErr_ExceptionMatches(PyExc_TypeError))
|
|
return error("PyLong_AsXXX(None) raised "
|
|
"something other than TypeError");
|
|
PyErr_Clear();
|
|
|
|
Py_DECREF(Py_None);
|
|
}
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|