summaryrefslogtreecommitdiffstats
path: root/Modules/_testcapi
diff options
context:
space:
mode:
authorDong-hee Na <donghee.na@python.org>2023-05-02 15:05:30 (GMT)
committerGitHub <noreply@github.com>2023-05-02 15:05:30 (GMT)
commitd81ca7ec029ba05084751c8df64292bb48f4f30f (patch)
treecc994cf44d437ac03d2b2c829377d6e40e16f319 /Modules/_testcapi
parent87223f32aba872cfebde6fbe38673799eb79f248 (diff)
downloadcpython-d81ca7ec029ba05084751c8df64292bb48f4f30f.zip
cpython-d81ca7ec029ba05084751c8df64292bb48f4f30f.tar.gz
cpython-d81ca7ec029ba05084751c8df64292bb48f4f30f.tar.bz2
gh-84436: Add integration C API tests for immortal objects (gh-103962)
Diffstat (limited to 'Modules/_testcapi')
-rw-r--r--Modules/_testcapi/immortal.c47
-rw-r--r--Modules/_testcapi/parts.h1
2 files changed, 48 insertions, 0 deletions
diff --git a/Modules/_testcapi/immortal.c b/Modules/_testcapi/immortal.c
new file mode 100644
index 0000000..10e1733
--- /dev/null
+++ b/Modules/_testcapi/immortal.c
@@ -0,0 +1,47 @@
+#include "parts.h"
+
+int verify_immortality(PyObject *object)
+{
+ assert(_Py_IsImmortal(object));
+ Py_ssize_t old_count = Py_REFCNT(object);
+ for (int j = 0; j < 10000; j++) {
+ Py_DECREF(object);
+ }
+ Py_ssize_t current_count = Py_REFCNT(object);
+ return old_count == current_count;
+}
+
+static PyObject *
+test_immortal_builtins(PyObject *self, PyObject *Py_UNUSED(ignored))
+{
+ PyObject *objects[] = {Py_True, Py_False, Py_None, Py_Ellipsis};
+ Py_ssize_t n = Py_ARRAY_LENGTH(objects);
+ for (Py_ssize_t i = 0; i < n; i++) {
+ assert(verify_immortality(objects[i]));
+ }
+ Py_RETURN_NONE;
+}
+
+static PyObject *
+test_immortal_small_ints(PyObject *self, PyObject *Py_UNUSED(ignored))
+{
+ for (int i = -5; i <= 256; i++) {
+ assert(verify_immortality(PyLong_FromLong(i)));
+ }
+ Py_RETURN_NONE;
+}
+
+static PyMethodDef test_methods[] = {
+ {"test_immortal_builtins", test_immortal_builtins, METH_NOARGS},
+ {"test_immortal_small_ints", test_immortal_small_ints, METH_NOARGS},
+ {NULL},
+};
+
+int
+_PyTestCapi_Init_Immortal(PyObject *mod)
+{
+ if (PyModule_AddFunctions(mod, test_methods) < 0) {
+ return -1;
+ }
+ return 0;
+}
diff --git a/Modules/_testcapi/parts.h b/Modules/_testcapi/parts.h
index 60ec81d..4d2d683 100644
--- a/Modules/_testcapi/parts.h
+++ b/Modules/_testcapi/parts.h
@@ -39,6 +39,7 @@ int _PyTestCapi_Init_Structmember(PyObject *module);
int _PyTestCapi_Init_Exceptions(PyObject *module);
int _PyTestCapi_Init_Code(PyObject *module);
int _PyTestCapi_Init_PyOS(PyObject *module);
+int _PyTestCapi_Init_Immortal(PyObject *module);
#ifdef LIMITED_API_AVAILABLE
int _PyTestCapi_Init_VectorcallLimited(PyObject *module);