1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
from pypy.module.cpyext.test.test_cpyext import AppTestCpythonExtensionBase
class AppTestPyMem(AppTestCpythonExtensionBase):
def test_pymem_alloc(self):
module = self.import_extension('foo', [
("test", "METH_NOARGS",
"""
int *a, *b;
a = PyMem_RawCalloc(4, 50);
if (a[49] != 0) {
PyErr_SetString(PyExc_ValueError, "1");
return NULL;
}
a[49] = 123456;
b = PyMem_RawRealloc(a, 2000);
b[499] = 789123;
PyMem_RawFree(b);
a = PyMem_Calloc(4, 50);
if (a[49] != 0) {
PyErr_SetString(PyExc_ValueError, "2");
return NULL;
}
a[49] = 123456;
b = PyMem_Realloc(a, 2000);
b[499] = 789123;
PyMem_Free(b);
Py_RETURN_NONE;
"""),
])
res = module.test()
assert res is None
|
