From 5087898b3461ba3bdc3b234ce0b5bb2886b3033e Mon Sep 17 00:00:00 2001 From: Hai Shi Date: Sat, 7 Aug 2021 09:55:08 +0800 Subject: [PATCH] Enhance test_get_type_name of _testcapi --- Modules/_testcapimodule.c | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c index f338e89f426da0..8e0993872c72df 100644 --- a/Modules/_testcapimodule.c +++ b/Modules/_testcapimodule.c @@ -1154,6 +1154,20 @@ test_get_type_name(PyObject *self, PyObject *Py_UNUSED(ignored)) assert(strcmp(PyUnicode_AsUTF8(tp_name), "HeapTypeNameType") == 0); Py_DECREF(tp_name); + PyObject *name = PyUnicode_FromString("test_name"); + if (name == NULL) { + goto done; + } + if (PyObject_SetAttrString(HeapTypeNameType, "__name__", name) < 0) { + Py_DECREF(name); + goto done; + } + tp_name = PyType_GetName((PyTypeObject *)HeapTypeNameType); + assert(strcmp(PyUnicode_AsUTF8(tp_name), "test_name") == 0); + Py_DECREF(name); + Py_DECREF(tp_name); + + done: Py_DECREF(HeapTypeNameType); Py_RETURN_NONE; }