Add test

Author: rruuaanng

Lib/test/test_capi/test_abstract.py

@@ -339,6 +339,13 @@ def test_object_delattrstring(self):
         # CRASHES delattrstring(obj, NULL)
         # CRASHES delattrstring(NULL, b'a')
 
+    def test_copy_to_object(self):
+        s1 = _testcapi.object_copy_to_object(bytes(5), 'abcde', 'C')
+        s2 = _testcapi.object_copy_to_object(bytes(5), 'abcde', 'F')
+        s3 = _testcapi.object_copy_to_object(bytes(5), 'abcde', 'A')
+        self.assertEqual(s1, s2)
+        self.assertEqual(s2, s3)
+        self.assertEqual(s1, s3)
 
     def test_mapping_check(self):
         check = _testlimitedcapi.mapping_check

Modules/_testcapi/abstract.c

@@ -78,6 +78,25 @@ object_hasattrstringwitherror(PyObject *self, PyObject *args)
     RETURN_INT(PyObject_HasAttrStringWithError(obj, attr_name));
 }
 
+static PyObject *
+object_copy_to_object(PyObject *self, PyObject *args)
+{
+    PyObject *obj;
+    Py_ssize_t len;
+    int result;
+    char *buf, fort;
+
+    if (!PyArg_ParseTuple(args, "Os#C", &obj, &buf, &len, &fort)) {
+        return NULL;
+    }
+    result = PyObject_CopyToObject(obj, buf, len, fort);
+    if (result < 0) {
+        return NULL;
+    }
+    Py_INCREF(obj);
+    return obj;
+}
+
 static PyObject *
 mapping_getoptionalitemstring(PyObject *self, PyObject *args)
 {
@@ -162,6 +181,8 @@ static PyMethodDef test_methods[] = {
     {"object_getoptionalattrstring", object_getoptionalattrstring, METH_VARARGS},
     {"object_hasattrwitherror", object_hasattrwitherror, METH_VARARGS},
     {"object_hasattrstringwitherror", object_hasattrstringwitherror, METH_VARARGS},
+    {"object_copy_to_object", object_copy_to_object, METH_VARARGS},
+
     {"mapping_getoptionalitem", mapping_getoptionalitem, METH_VARARGS},
     {"mapping_getoptionalitemstring", mapping_getoptionalitemstring, METH_VARARGS},