mirror of
https://github.com/python/cpython
synced 2024-10-14 17:35:44 +00:00
gh-94808: Cover PyFunction_GetCode
, PyFunction_GetGlobals
, PyFunction_GetModule
(#98158)
This commit is contained in:
parent
660f10248b
commit
7b48d02933
|
@ -895,6 +895,41 @@ def method_example(self): ...
|
|||
self.assertEqual(_testcapi.eval_get_func_name(sum), "sum") # c function
|
||||
self.assertEqual(_testcapi.eval_get_func_name(A), "type")
|
||||
|
||||
def test_function_get_code(self):
|
||||
import types
|
||||
|
||||
def some():
|
||||
pass
|
||||
|
||||
code = _testcapi.function_get_code(some)
|
||||
self.assertIsInstance(code, types.CodeType)
|
||||
self.assertEqual(code, some.__code__)
|
||||
|
||||
with self.assertRaises(SystemError):
|
||||
_testcapi.function_get_code(None) # not a function
|
||||
|
||||
def test_function_get_globals(self):
|
||||
def some():
|
||||
pass
|
||||
|
||||
globals_ = _testcapi.function_get_globals(some)
|
||||
self.assertIsInstance(globals_, dict)
|
||||
self.assertEqual(globals_, some.__globals__)
|
||||
|
||||
with self.assertRaises(SystemError):
|
||||
_testcapi.function_get_globals(None) # not a function
|
||||
|
||||
def test_function_get_module(self):
|
||||
def some():
|
||||
pass
|
||||
|
||||
module = _testcapi.function_get_module(some)
|
||||
self.assertIsInstance(module, str)
|
||||
self.assertEqual(module, some.__module__)
|
||||
|
||||
with self.assertRaises(SystemError):
|
||||
_testcapi.function_get_module(None) # not a function
|
||||
|
||||
|
||||
class TestPendingCalls(unittest.TestCase):
|
||||
|
||||
|
|
|
@ -5658,6 +5658,42 @@ test_macros(PyObject *self, PyObject *Py_UNUSED(args))
|
|||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
function_get_code(PyObject *self, PyObject *func)
|
||||
{
|
||||
PyObject *code = PyFunction_GetCode(func);
|
||||
if (code != NULL) {
|
||||
Py_INCREF(code);
|
||||
return code;
|
||||
} else {
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
function_get_globals(PyObject *self, PyObject *func)
|
||||
{
|
||||
PyObject *globals = PyFunction_GetGlobals(func);
|
||||
if (globals != NULL) {
|
||||
Py_INCREF(globals);
|
||||
return globals;
|
||||
} else {
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static PyObject *
|
||||
function_get_module(PyObject *self, PyObject *func)
|
||||
{
|
||||
PyObject *module = PyFunction_GetModule(func);
|
||||
if (module != NULL) {
|
||||
Py_INCREF(module);
|
||||
return module;
|
||||
} else {
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static PyObject *test_buildvalue_issue38913(PyObject *, PyObject *);
|
||||
static PyObject *getargs_s_hash_int(PyObject *, PyObject *, PyObject*);
|
||||
|
@ -5942,6 +5978,9 @@ static PyMethodDef TestMethods[] = {
|
|||
{"watch_dict", watch_dict, METH_VARARGS, NULL},
|
||||
{"unwatch_dict", unwatch_dict, METH_VARARGS, NULL},
|
||||
{"get_dict_watcher_events", get_dict_watcher_events, METH_NOARGS, NULL},
|
||||
{"function_get_code", function_get_code, METH_O, NULL},
|
||||
{"function_get_globals", function_get_globals, METH_O, NULL},
|
||||
{"function_get_module", function_get_module, METH_O, NULL},
|
||||
{NULL, NULL} /* sentinel */
|
||||
};
|
||||
|
||||
|
|
Loading…
Reference in a new issue