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 35 36 37 38 39
|
# mode: run
# tag: c-api
from cpython cimport mem
from cpython.pystate cimport PyGILState_Ensure, PyGILState_Release, PyGILState_STATE
def test_pymalloc():
"""
>>> test_pymalloc()
3
"""
cdef char* m2
cdef char* m = <char*> mem.PyMem_Malloc(20)
assert m
try:
m[0] = 1
m[1] = 2
m[2] = 3
m2 = <char*> mem.PyMem_Realloc(m, 10)
assert m2
m = m2
return m[2]
finally:
mem.PyMem_Free(m)
def test_gilstate():
"""
>>> test_gilstate()
'ok'
"""
# cython used to have invalid definition for PyGILState_STATE, which was
# making the following code fail to compile
cdef PyGILState_STATE gstate = PyGILState_Ensure()
# TODO assert that GIL is taken
PyGILState_Release(gstate)
return 'ok'
|