From fcd651d16fc5ac3d07dd3f57f1001a861a2e7d23 Mon Sep 17 00:00:00 2001 From: Hai Shi Date: Tue, 17 Aug 2021 22:50:33 +0800 Subject: bpo-42035: Enhance test_get_type_name() of _testcapi (GH-27649) --- Modules/_testcapimodule.c | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c index 0a3c6e0..e5f1ad3 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; } -- cgit v0.12