Kaydet (Commit) c6a7c96a authored tarafından Serhiy Storchaka's avatar Serhiy Storchaka

Backported tests for issue #18531.

üst 6904ca3a
......@@ -64,6 +64,12 @@ class Int:
def __int__(self):
return 99
class TupleSubclass(tuple):
pass
class DictSubclass(dict):
pass
class Unsigned_TestCase(unittest.TestCase):
def test_b(self):
from _testcapi import getargs_b
......@@ -268,6 +274,33 @@ class LongLong_TestCase(unittest.TestCase):
class Tuple_TestCase(unittest.TestCase):
def test_args(self):
from _testcapi import get_args
ret = get_args(1, 2)
self.assertEqual(ret, (1, 2))
self.assertIs(type(ret), tuple)
ret = get_args(1, *(2, 3))
self.assertEqual(ret, (1, 2, 3))
self.assertIs(type(ret), tuple)
ret = get_args(*[1, 2])
self.assertEqual(ret, (1, 2))
self.assertIs(type(ret), tuple)
ret = get_args(*TupleSubclass([1, 2]))
self.assertEqual(ret, (1, 2))
self.assertIsInstance(ret, tuple)
ret = get_args()
self.assertIn(ret, ((), None))
self.assertIn(type(ret), (tuple, type(None)))
ret = get_args(*())
self.assertIn(ret, ((), None))
self.assertIn(type(ret), (tuple, type(None)))
def test_tuple(self):
from _testcapi import getargs_tuple
......@@ -283,6 +316,29 @@ class Tuple_TestCase(unittest.TestCase):
self.assertRaises(TypeError, getargs_tuple, 1, seq())
class Keywords_TestCase(unittest.TestCase):
def test_kwargs(self):
from _testcapi import get_kwargs
ret = get_kwargs(a=1, b=2)
self.assertEqual(ret, {'a': 1, 'b': 2})
self.assertIs(type(ret), dict)
ret = get_kwargs(a=1, **{'b': 2, 'c': 3})
self.assertEqual(ret, {'a': 1, 'b': 2, 'c': 3})
self.assertIs(type(ret), dict)
ret = get_kwargs(**DictSubclass({'a': 1, 'b': 2}))
self.assertEqual(ret, {'a': 1, 'b': 2})
self.assertIsInstance(ret, dict)
ret = get_kwargs()
self.assertIn(ret, ({}, None))
self.assertIn(type(ret), (dict, type(None)))
ret = get_kwargs(**{})
self.assertIn(ret, ({}, None))
self.assertIn(type(ret), (dict, type(None)))
def test_positional_args(self):
# using all positional args
self.assertEqual(
......
......@@ -930,6 +930,26 @@ test_L_code(PyObject *self)
#endif /* ifdef HAVE_LONG_LONG */
static PyObject *
get_args(PyObject *self, PyObject *args)
{
if (args == NULL) {
args = Py_None;
}
Py_INCREF(args);
return args;
}
static PyObject *
get_kwargs(PyObject *self, PyObject *args, PyObject *kwargs)
{
if (kwargs == NULL) {
kwargs = Py_None;
}
Py_INCREF(kwargs);
return kwargs;
}
/* Test tuple argument processing */
static PyObject *
getargs_tuple(PyObject *self, PyObject *args)
......@@ -2335,6 +2355,8 @@ static PyMethodDef TestMethods[] = {
{"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
PyDoc_STR("This is a pretty normal docstring.")},
{"get_args", get_args, METH_VARARGS},
{"get_kwargs", (PyCFunction)get_kwargs, METH_VARARGS|METH_KEYWORDS},
{"getargs_tuple", getargs_tuple, METH_VARARGS},
{"getargs_keywords", (PyCFunction)getargs_keywords,
METH_VARARGS|METH_KEYWORDS},
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment