bpo-33967: Fix singledispatch raised IndexError when no args (GH-8184)

This commit is contained in:
Dong-hee Na 2018-07-10 16:26:36 +09:00 committed by INADA Naoki
parent 7762e4d387
commit 445f1b35ce
3 changed files with 14 additions and 0 deletions

View File

@ -817,8 +817,13 @@ def singledispatch(func):
return func return func
def wrapper(*args, **kw): def wrapper(*args, **kw):
if not args:
raise TypeError(f'{funcname} requires at least '
'1 positional argument')
return dispatch(args[0].__class__)(*args, **kw) return dispatch(args[0].__class__)(*args, **kw)
funcname = getattr(func, '__name__', 'singledispatch function')
registry[object] = func registry[object] = func
wrapper.register = register wrapper.register = register
wrapper.dispatch = dispatch wrapper.dispatch = dispatch

View File

@ -2305,6 +2305,13 @@ class TestSingleDispatch(unittest.TestCase):
)) ))
self.assertTrue(str(exc.exception).endswith(msg_suffix)) self.assertTrue(str(exc.exception).endswith(msg_suffix))
def test_invalid_positional_argument(self):
@functools.singledispatch
def f(*args):
pass
msg = 'f requires at least 1 positional argument'
with self.assertRaisesRegexp(TypeError, msg):
f()
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -0,0 +1,2 @@
functools.singledispatch now raises TypeError instead of IndexError when no
positional arguments are passed.