Update tests for Python 3.12 (#60)

vstinner · web-flow · commit 513acb3b0602 · 2023-06-01T12:57:27.000+02:00
* Install explicitly setuptools for Python 3.12
* Fix tests for Python 3.12 beta1 immortal objects PEP 683
* Fix refleak check for Python 3.12
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -58,6 +58,8 @@ jobs:
         uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.python }}
+      - name: Install setuptools
+        run: python -m pip install setuptools
       - name: Display the Python version
         run: python -VV
       - name: Run tests
diff --git a/tests/test_pythoncapi_compat.py b/tests/test_pythoncapi_compat.py
@@ -9,6 +9,7 @@
 """
 from __future__ import absolute_import
 from __future__ import print_function
+import gc
 import os.path
 import shutil
 import subprocess
@@ -82,6 +83,14 @@ def _run_tests(tests, verbose):
         test_func()
 
 
+_HAS_CLEAR_TYPE_CACHE = hasattr(sys, '_clear_type_cache')
+
+def _refleak_cleanup():
+    if _HAS_CLEAR_TYPE_CACHE:
+        sys._clear_type_cache()
+    gc.collect()
+
+
 def _check_refleak(test_func, verbose):
     nrun = 6
     for i in range(1, nrun + 1):
@@ -93,6 +102,7 @@ def _check_refleak(test_func, verbose):
 
         init_refcnt = sys.gettotalrefcount()
         test_func()
+        _refleak_cleanup()
         diff = sys.gettotalrefcount() - init_refcnt
 
         if i > 3 and diff:
diff --git a/tests/test_pythoncapi_compat_cext.c b/tests/test_pythoncapi_compat_cext.c
@@ -37,6 +37,13 @@
 #  define CHECK_REFCNT
 #endif
 
+// CPython 3.12 beta 1 implements immortal objects (PEP 683)
+#if 0x030C00B1 <= PY_VERSION_HEX && !defined(PYPY_VERSION)
+   // Don't check reference count of Python 3.12 immortal objects (ex: bool
+   // and str types)
+#  define IMMORTAL_OBJS
+#endif
+
 #ifdef CHECK_REFCNT
 #  define ASSERT_REFCNT(expr) assert(expr)
 #else
@@ -381,7 +388,9 @@ test_module_add_type(PyObject *module)
     if (PyModule_AddType(module, type) < 0) {
         return -1;
     }
+#ifndef IMMORTAL_OBJS
     ASSERT_REFCNT(Py_REFCNT(type) == refcnt + 1);
+#endif
 
     PyObject *attr = PyObject_GetAttrString(module, type_name);
     if (attr == _Py_NULL) {
@@ -412,7 +421,9 @@ test_module_addobjectref(PyObject *module)
         ASSERT_REFCNT(Py_REFCNT(obj) == refcnt);
         return -1;
     }
+#ifndef IMMORTAL_OBJS
     ASSERT_REFCNT(Py_REFCNT(obj) == refcnt + 1);
+#endif
 
     if (PyObject_DelAttrString(module, name) < 0) {
         return -1;
