mirror of https://github.com/python/cpython.git
45 lines
1.0 KiB
C
45 lines
1.0 KiB
C
#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);
|
|
}
|
|
|