add test

python · vstinner · Nov 8, 2023 · Oct 31, 2023 · Oct 31, 2023 · Oct 31, 2023
commit 322eab934fe5b00b077b76ae19c72e57515655c4
diff --git a/Lib/te 8000 st/test_capi/test_list.py b/Lib/te 8000 st/test_capi/test_list.py
@@ -33,8 +33,18 @@ def test_list_check_exact(self):
 
     def test_list_new(self):
         list_new = _testcapi.list_new
-        lst = list_new()
+        lst = list_new(0)
         self.assertEqual(lst, [])
         self.assertIs(type(lst), list)
-        lst2 = list_new()
+        lst2 = list_new(0)
         self.assertIsNot(lst2, lst)
+
+    def test_list_size(self):
+        size = _testcapi.list_size
+        self.assertEqual(size([1, 2]), 2)
+        self.assertEqual(size(ListSubclass([1, 2])), 2)
+        self.assertRaises(SystemError, size, UserList())
+        self.assertRaises(SystemError, size, {})
+        self.assertRaises(SystemError, size, 23)
+        self.assertRaises(SystemError, size, object())
+        # CRASHES size(NULL)
diff --git a/Modules/_testcapi/list.c b/Modules/_testcapi/list.c
@@ -18,14 +18,22 @@ list_check_exact(PyObject* Py_UNUSED(module), PyObject *obj)
 static PyObject *
 list_new(PyObject *self, PyObject *obj)
 {
-    return PyList_New(obj);
+    return PyList_New(PyLong_AsSsize_t(obj));
 }
 
+static PyObject *
+list_size(PyObject *Py_UNUSED(module), PyObject *obj)
+{
+    NULLABLE(obj);
+    RETURN_SIZE(PyList_Size(obj));
+}
 
 static PyMethodDef test_methods[] = {
     {"list_check", list_check, METH_O},
     {"list_check_exact", list_check_exact, METH_O},
-    {"list_new", list_new, METH_NOARGS},
+    {"list_new", list_new, METH_O},
+    {"list_size", list_size, METH_O},
+
 };
 
 int