diff --git a/Lib/test/test_capi.py b/Lib/test/test_capi.py
index ae434c05baf2f2..a2183cfb0fdf6a 100644
--- a/Lib/test/test_capi.py
+++ b/Lib/test/test_capi.py
@@ -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):
diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c
index 76e619d11f98e3..624e878b20d822 100644
--- a/Modules/_testcapimodule.c
+++ b/Modules/_testcapimodule.c
@@ -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 */
};
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy