1 /* Poor-man's template. Macros used:
2 TESTNAME name of the test (like test_long_api_inner)
3 TYPENAME the signed type (like long)
4 F_S_TO_PY convert signed to pylong; TYPENAME -> PyObject*
5 F_PY_TO_S convert pylong to signed; PyObject* -> TYPENAME
6 F_U_TO_PY convert unsigned to pylong; unsigned TYPENAME -> PyObject*
7 F_PY_TO_U convert pylong to unsigned; PyObject* -> unsigned TYPENAME
8 */
9
10 static PyObject *
TESTNAME(PyObject * error (const char *))11 TESTNAME(PyObject *error(const char*))
12 {
13 const int NBITS = sizeof(TYPENAME) * 8;
14 unsigned TYPENAME base;
15 PyObject *pyresult;
16 int i;
17
18 /* Note: This test lets PyObjects leak if an error is raised. Since
19 an error should never be raised, leaks are impossible <wink>. */
20
21 /* Test native -> PyLong -> native roundtrip identity.
22 * Generate all powers of 2, and test them and their negations,
23 * plus the numbers +-1 off from them.
24 */
25 base = 1;
26 for (i = 0;
27 i < NBITS + 1; /* on last, base overflows to 0 */
28 ++i, base <<= 1)
29 {
30 int j;
31 for (j = 0; j < 6; ++j) {
32 TYPENAME in, out;
33 unsigned TYPENAME uin, uout;
34
35 /* For 0, 1, 2 use base; for 3, 4, 5 use -base */
36 uin = j < 3 ? base
37 : (unsigned TYPENAME)(-(TYPENAME)base);
38
39 /* For 0 & 3, subtract 1.
40 * For 1 & 4, leave alone.
41 * For 2 & 5, add 1.
42 */
43 uin += (unsigned TYPENAME)(TYPENAME)(j % 3 - 1);
44
45 pyresult = F_U_TO_PY(uin);
46 if (pyresult == NULL)
47 return error(
48 "unsigned unexpected null result");
49
50 uout = F_PY_TO_U(pyresult);
51 if (uout == (unsigned TYPENAME)-1 && PyErr_Occurred())
52 return error(
53 "unsigned unexpected -1 result");
54 if (uout != uin)
55 return error(
56 "unsigned output != input");
57 UNBIND(pyresult);
58
59 in = (TYPENAME)uin;
60 pyresult = F_S_TO_PY(in);
61 if (pyresult == NULL)
62 return error(
63 "signed unexpected null result");
64
65 out = F_PY_TO_S(pyresult);
66 if (out == (TYPENAME)-1 && PyErr_Occurred())
67 return error(
68 "signed unexpected -1 result");
69 if (out != in)
70 return error(
71 "signed output != input");
72 UNBIND(pyresult);
73 }
74 }
75
76 /* Overflow tests. The loop above ensured that all limit cases that
77 * should not overflow don't overflow, so all we need to do here is
78 * provoke one-over-the-limit cases (not exhaustive, but sharp).
79 */
80 {
81 PyObject *one, *x, *y;
82 TYPENAME out;
83 unsigned TYPENAME uout;
84
85 one = PyLong_FromLong(1);
86 if (one == NULL)
87 return error(
88 "unexpected NULL from PyLong_FromLong");
89
90 /* Unsigned complains about -1? */
91 x = PyNumber_Negative(one);
92 if (x == NULL)
93 return error(
94 "unexpected NULL from PyNumber_Negative");
95
96 uout = F_PY_TO_U(x);
97 if (uout != (unsigned TYPENAME)-1 || !PyErr_Occurred())
98 return error(
99 "PyLong_AsUnsignedXXX(-1) didn't complain");
100 if (!PyErr_ExceptionMatches(PyExc_OverflowError))
101 return error(
102 "PyLong_AsUnsignedXXX(-1) raised "
103 "something other than OverflowError");
104 PyErr_Clear();
105 UNBIND(x);
106
107 /* Unsigned complains about 2**NBITS? */
108 y = PyLong_FromLong((long)NBITS);
109 if (y == NULL)
110 return error(
111 "unexpected NULL from PyLong_FromLong");
112
113 x = PyNumber_Lshift(one, y); /* 1L << NBITS, == 2**NBITS */
114 UNBIND(y);
115 if (x == NULL)
116 return error(
117 "unexpected NULL from PyNumber_Lshift");
118
119 uout = F_PY_TO_U(x);
120 if (uout != (unsigned TYPENAME)-1 || !PyErr_Occurred())
121 return error(
122 "PyLong_AsUnsignedXXX(2**NBITS) didn't "
123 "complain");
124 if (!PyErr_ExceptionMatches(PyExc_OverflowError))
125 return error(
126 "PyLong_AsUnsignedXXX(2**NBITS) raised "
127 "something other than OverflowError");
128 PyErr_Clear();
129
130 /* Signed complains about 2**(NBITS-1)?
131 x still has 2**NBITS. */
132 y = PyNumber_Rshift(x, one); /* 2**(NBITS-1) */
133 UNBIND(x);
134 if (y == NULL)
135 return error(
136 "unexpected NULL from PyNumber_Rshift");
137
138 out = F_PY_TO_S(y);
139 if (out != (TYPENAME)-1 || !PyErr_Occurred())
140 return error(
141 "PyLong_AsXXX(2**(NBITS-1)) didn't "
142 "complain");
143 if (!PyErr_ExceptionMatches(PyExc_OverflowError))
144 return error(
145 "PyLong_AsXXX(2**(NBITS-1)) raised "
146 "something other than OverflowError");
147 PyErr_Clear();
148
149 /* Signed complains about -2**(NBITS-1)-1?;
150 y still has 2**(NBITS-1). */
151 x = PyNumber_Negative(y); /* -(2**(NBITS-1)) */
152 UNBIND(y);
153 if (x == NULL)
154 return error(
155 "unexpected NULL from PyNumber_Negative");
156
157 y = PyNumber_Subtract(x, one); /* -(2**(NBITS-1))-1 */
158 UNBIND(x);
159 if (y == NULL)
160 return error(
161 "unexpected NULL from PyNumber_Subtract");
162
163 out = F_PY_TO_S(y);
164 if (out != (TYPENAME)-1 || !PyErr_Occurred())
165 return error(
166 "PyLong_AsXXX(-2**(NBITS-1)-1) didn't "
167 "complain");
168 if (!PyErr_ExceptionMatches(PyExc_OverflowError))
169 return error(
170 "PyLong_AsXXX(-2**(NBITS-1)-1) raised "
171 "something other than OverflowError");
172 PyErr_Clear();
173 UNBIND(y);
174
175 Py_XDECREF(x);
176 Py_XDECREF(y);
177 Py_DECREF(one);
178 }
179
180 Py_INCREF(Py_None);
181 return Py_None;
182 }
183