Commit c6a7c96a authored by Serhiy Storchaka's avatar Serhiy Storchaka

Backported tests for issue #18531.

parent 6904ca3a
...@@ -64,6 +64,12 @@ class Int: ...@@ -64,6 +64,12 @@ class Int:
def __int__(self): def __int__(self):
return 99 return 99
class TupleSubclass(tuple):
pass
class DictSubclass(dict):
pass
class Unsigned_TestCase(unittest.TestCase): class Unsigned_TestCase(unittest.TestCase):
def test_b(self): def test_b(self):
from _testcapi import getargs_b from _testcapi import getargs_b
...@@ -268,6 +274,33 @@ class LongLong_TestCase(unittest.TestCase): ...@@ -268,6 +274,33 @@ class LongLong_TestCase(unittest.TestCase):
class Tuple_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): def test_tuple(self):
from _testcapi import getargs_tuple from _testcapi import getargs_tuple
...@@ -283,6 +316,29 @@ class Tuple_TestCase(unittest.TestCase): ...@@ -283,6 +316,29 @@ class Tuple_TestCase(unittest.TestCase):
self.assertRaises(TypeError, getargs_tuple, 1, seq()) self.assertRaises(TypeError, getargs_tuple, 1, seq())
class Keywords_TestCase(unittest.TestCase): 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): def test_positional_args(self):
# using all positional args # using all positional args
self.assertEqual( self.assertEqual(
......
...@@ -930,6 +930,26 @@ test_L_code(PyObject *self) ...@@ -930,6 +930,26 @@ test_L_code(PyObject *self)
#endif /* ifdef HAVE_LONG_LONG */ #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 */ /* Test tuple argument processing */
static PyObject * static PyObject *
getargs_tuple(PyObject *self, PyObject *args) getargs_tuple(PyObject *self, PyObject *args)
...@@ -2335,6 +2355,8 @@ static PyMethodDef TestMethods[] = { ...@@ -2335,6 +2355,8 @@ static PyMethodDef TestMethods[] = {
{"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS, {"test_with_docstring", (PyCFunction)test_with_docstring, METH_NOARGS,
PyDoc_STR("This is a pretty normal docstring.")}, 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_tuple", getargs_tuple, METH_VARARGS},
{"getargs_keywords", (PyCFunction)getargs_keywords, {"getargs_keywords", (PyCFunction)getargs_keywords,
METH_VARARGS|METH_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