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
40
41
42
43
44
|
#include "parts.h"
#include "util.h"
// Test PyImport_ImportModuleAttr()
static PyObject *
pyimport_importmoduleattr(PyObject *self, PyObject *args)
{
PyObject *mod_name, *attr_name;
if (!PyArg_ParseTuple(args, "OO", &mod_name, &attr_name)) {
return NULL;
}
NULLABLE(mod_name);
NULLABLE(attr_name);
return PyImport_ImportModuleAttr(mod_name, attr_name);
}
// Test PyImport_ImportModuleAttrString()
static PyObject *
pyimport_importmoduleattrstring(PyObject *self, PyObject *args)
{
const char *mod_name, *attr_name;
Py_ssize_t len;
if (!PyArg_ParseTuple(args, "z#z#", &mod_name, &len, &attr_name, &len)) {
return NULL;
}
return PyImport_ImportModuleAttrString(mod_name, attr_name);
}
static PyMethodDef test_methods[] = {
{"PyImport_ImportModuleAttr", pyimport_importmoduleattr, METH_VARARGS},
{"PyImport_ImportModuleAttrString", pyimport_importmoduleattrstring, METH_VARARGS},
{NULL},
};
int
_PyTestCapi_Init_Import(PyObject *m)
{
return PyModule_AddFunctions(m, test_methods);
}
|
