pythongh-101101: Fix test_code_extra to reset value for refleak test

corona10 · Mar 1, 2023 · dcb4d94 · dcb4d94
1 parent 7d1d663
commit dcb4d94
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Modules/_testcapi/code.c b/Modules/_testcapi/code.c
@@ -92,7 +92,11 @@ test_code_extra(PyObject* self, PyObject *Py_UNUSED(callable))
         goto finally;
     }
     assert ((uintptr_t)extra == 77);
-
+    // Need to reset code extra value.
+    res = PyUnstable_Code_SetExtra(test_func_code, code_extra_index, NULL);
+    if (res < 0) {
+        goto finally;
+    }
     result = Py_NewRef(Py_None);
 finally:
     Py_XDECREF(test_module);