From 40490c15bea5556f9ee59600dd22774ca257192b Mon Sep 17 00:00:00 2001 From: Nikita Sobolev Date: Sat, 5 Nov 2022 16:07:59 +0300 Subject: [PATCH] gh-94808: add tests covering `PyEval_GetFuncDesc` function (GH-98300) (cherry picked from commit b5f711185bd11819566068ddf2a74a1402340e2d) Co-authored-by: Nikita Sobolev --- Lib/test/test_capi.py | 15 +++++++++++++++ Modules/_testcapimodule.c | 7 +++++++ 2 files changed, 22 insertions(+) diff --git a/Lib/test/test_capi.py b/Lib/test/test_capi.py index a4d643fb056e4c..3e6549625870eb 100644 --- a/Lib/test/test_capi.py +++ b/Lib/test/test_capi.py @@ -751,6 +751,21 @@ 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_eval_get_func_desc(self): + def function_example(): ... + + class A: + def method_example(self): ... + + self.assertEqual(_testcapi.eval_get_func_desc(function_example), + "()") + self.assertEqual(_testcapi.eval_get_func_desc(A.method_example), + "()") + self.assertEqual(_testcapi.eval_get_func_desc(A().method_example), + "()") + self.assertEqual(_testcapi.eval_get_func_desc(sum), "()") # c function + self.assertEqual(_testcapi.eval_get_func_desc(A), " object") + def test_function_get_code(self): import types diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c index 1c085a45f8c4f4..3ec91bf14cfcea 100644 --- a/Modules/_testcapimodule.c +++ b/Modules/_testcapimodule.c @@ -5975,6 +5975,12 @@ eval_get_func_name(PyObject *self, PyObject *func) return PyUnicode_FromString(PyEval_GetFuncName(func)); } +static PyObject * +eval_get_func_desc(PyObject *self, PyObject *func) +{ + return PyUnicode_FromString(PyEval_GetFuncDesc(func)); +} + static PyObject * get_feature_macros(PyObject *self, PyObject *Py_UNUSED(args)) { @@ -6452,6 +6458,7 @@ static PyMethodDef TestMethods[] = { {"frame_getbuiltins", frame_getbuiltins, METH_O, NULL}, {"frame_getlasti", frame_getlasti, METH_O, NULL}, {"eval_get_func_name", eval_get_func_name, METH_O, NULL}, + {"eval_get_func_desc", eval_get_func_desc, METH_O, NULL}, {"get_feature_macros", get_feature_macros, METH_NOARGS, NULL}, {"test_code_api", test_code_api, METH_NOARGS, NULL}, {"settrace_to_record", settrace_to_record, METH_O, NULL},