Add CAPI tests

Author: tiran

Lib/test/test_capi.py

@@ -971,5 +971,19 @@ def test_state_access(self):
                     increment_count(1, 2, 3)
 
 
+class Test_PyModuleConst_Def(unittest.TestCase):
+    def test_constants(self):
+        self.assertIs(_testcapi.const_none, None)
+        self.assertEqual(_testcapi.const_int, 42)
+        self.assertEqual(_testcapi.const_uint, _testcapi.ULONG_MAX)
+        self.assertIs(_testcapi.const_true, True)
+        self.assertIs(_testcapi.const_false, False)
+        self.assertEqual(_testcapi.const_almost_tau, 6.2831)
+        self.assertEqual(_testcapi.const_str, "Hello")
+        self.assertEqual(_testcapi.const_call, b"23")
+        self.assertEqual(_testcapi.CONST_INT, 7)
+        self.assertEqual(_testcapi.CONST_STRING, "world")
+
+
 if __name__ == "__main__":
     unittest.main()

Modules/_testcapimodule.c

@@ -7010,6 +7010,29 @@ static PyTypeObject ContainerNoGC_type = {
     .tp_new = ContainerNoGC_new,
 };
 
+static PyObject *
+_testcapimodule_const_call(PyObject *module)
+{
+    return PyBytes_FromString("23");
+}
+
+#define CONST_INT 7
+#define CONST_STRING "world"
+
+static PyModuleConst_Def _testcapimodule_consts[] = {
+    PyModuleConst_None("const_none"),
+    PyModuleConst_Long("const_int", 42),
+    PyModuleConst_ULong("const_uint", ULONG_MAX),
+    PyModuleConst_Bool("const_false", 0),
+    PyModuleConst_Bool("const_true", 1),
+    PyModuleConst_Double("const_almost_tau", 6.2831),
+    PyModuleConst_String("const_str", "Hello"),
+    PyModuleConst_Call("const_call", _testcapimodule_const_call),
+    PyModuleConst_LongMacro(CONST_INT),
+    PyModuleConst_StringMacro(CONST_STRING),
+    {NULL},
+};
+
 
 static struct PyModuleDef _testcapimodule = {
     PyModuleDef_HEAD_INIT,
@@ -7035,6 +7058,11 @@ PyInit__testcapi(void)
     if (m == NULL)
         return NULL;
 
+    if (PyModule_AddConstants(m, _testcapimodule_consts) < 0) {
+        Py_DECREF(m);
+        return NULL;
+    }
+
     Py_SET_TYPE(&_HashInheritanceTester_Type, &PyType_Type);
 
     Py_SET_TYPE(&test_structmembersType, &PyType_Type);