8000 Unit tests for PEP 3102, by Jiwon Seo. Forgot to add these earlier. · python/cpython@dfc9d4f · GitHub
[go: up one dir, main page]
Skip to content

Commit dfc9d4f

Browse files
gvanrossumgvanrossum
committed
Unit tests for PEP 3102, by Jiwon Seo. Forgot to add these earlier.
1 parent f59677a commit dfc9d4f

File tree

1 file changed

+151
-0
lines changed

1 file changed

+151
-0
lines changed

‎Lib/test/test_keywordonlyarg.py

Lines changed: 151 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,151 @@
1+
#!/usr/bin/env python
2+
3+
"""Unit tests for the keyword only argument specified in PEP 3102."""
4+
5+
__author__ = "Jiwon Seo"
6+
__email__ = "seojiwon at gmail dot com"
7+
8+
import unittest
9+
from test.test_support import run_unittest
10+
11+
def posonly_sum(pos_arg1, *arg, **kwarg):
12+
return pos_arg1 + sum(arg) + sum(kwarg.values())
13+
def keywordonly_sum(*, k1=0, k2):
14+
return k1 + k2
15+
def keywordonly_nodefaults_sum(*, k1, k2):
16+
return k1 + k2
17+
def keywordonly_and_kwarg_sum(*, k1, k2, **kwarg):
18+
return k1 + k2 + sum(kwarg.values())
19+
def mixedargs_sum(a, b=0, *arg, k1, k2=0):
20+
return a + b + k1 + k2 + sum(arg)
21+
def mixedargs_sum2(a, b=0, *arg, k1, k2=0, **kwargs):
22+
return a + b + k1 + k2 + sum(arg) + sum(kwargs.values())
23+
24+
def sortnum(*nums, reverse=False):
25+
return sorted(list(nums), reverse=reverse)
26+
27+
def sortwords(*words, reverse=False, **kwargs):
28+
return sorted(list(words), reverse=reverse)
29+
30+
class Foo:
31+
def __init__(self, *, k1, k2=0):
32+
self.k1 = k1
33+
self.k2 = k2
34+
def set(self, p1, *, k1, k2):
35+
self.k1 = k1
36+
self.k2 = k2
37+
def sum(self):
38+
return self.k1 + self.k2
39+
40+
class KeywordOnlyArgTestCase(unittest.TestCase):
41+
def assertRaisesSyntaxError(self, codestr):
42+
def shouldRaiseSyntaxError(s):
43+
compile(s, "<test>", "single")
44+
self.assertRaises(SyntaxError, shouldRaiseSyntaxError, codestr)
45+
46+
def testSyntaxErrorForFunctionDefinition(self):
47+
self.assertRaisesSyntaxError("def f(p, *):\n pass\n")
48+
self.assertRaisesSyntaxError("def f(p1, *, p1=100):\n pass\n")
49+
self.assertRaisesSyntaxError("def f(p1, *k1, k1=100):\n pass\n")
50+
self.assertRaisesSyntaxError("def f(p1, *, k1, k1=100):\n pass\n")
51+
self.assertRaisesSyntaxError("def f(p1, *, k1, **k1):\n pass\n")
52+
self.assertRaisesSyntaxError("def f(p1, *, None, **k1):\n pass\n")
53+
self.assertRaisesSyntaxError("def f(p, *, (k1, k2), **kw):\n pass\n")
54+
55+
def testSyntaxForManyArguments(self):
56+
fundef = "def f("
57+
for i in range(255):
58+
fundef += "i%d, "%i
59+
fundef += "*, key=100):\n pass\n"
60+
self.assertRaisesSyntaxError(fundef)
61+
62+
fundef2 = "def foo(i,*,"
63+
for i in range(255):
64+
fundef2 += "i%d, "%i
65+
fundef2 += "lastarg):\n pass\n"
66+
self.assertRaisesSyntaxError(fundef2)
67+
68+
# exactly 255 arguments, should compile ok
69+
fundef3 = "def f(i,*,"
70+
for i in range(253):
71+
fundef3 += "i%d, "%i
72+
fundef3 += "lastarg):\n pass\n"
73+
compile(fundef3, "<test>", "single")
74+
75+
def testSyntaxErrorForFunctionCall(self):
76+
self.assertRaisesSyntaxError("f(p, k=1, p2)")
77+
self.assertRaisesSyntaxError("f(p, *(1,2), k1=100)")
78+
79+
def testRaiseErrorFuncallWithUnexpectedKeywordArgument(self):
80+
self.assertRaises(TypeError, keywordonly_sum, ())
81+
self.assertRaises(TypeError, keywordonly_nodefaults_sum, ())
82+
self.assertRaises(TypeError, Foo, ())
83+
try:
84+
keywordonly_sum(k2=100, non_existing_arg=200)
85+
self.fail("should raise TypeError")
86+
except TypeError:
87+
pass
88+
try:
89+
keywordonly_nodefaults_sum(k2=2)
90+
self.fail("should raise TypeError")
91+
except TypeError:
92+
pass
93+
94+
def testFunctionCall(self):
95+
self.assertEquals(1, posonly_sum(1))
96+
self.assertEquals(1+2, posonly_sum(1,**{"2":2}))
97+
self.assertEquals(1+2+3, posonly_sum(1,*(2,3)))
98+
self.assertEquals(1+2+3+4, posonly_sum(1,*(2,3),**{"4":4}))
99+
100+
self.assertEquals(1, keywordonly_sum(k2=1))
101+
self.assertEquals(1+2, keywordonly_sum(k1=1, k2=2))
102+
103+
self.assertEquals(1+2, keywordonly_and_kwarg_sum(k1=1, k2=2))
104+
self.assertEquals(1+2+3, keywordonly_and_kwarg_sum(k1=1, k2=2, k3=3))
105+
self.assertEquals(1+2+3+4,
106+
keywordonly_and_kwarg_sum(k1=1, k2=2,
107+
**{"a":3,"b":4}))
108+
109+
self.assertEquals(1+2, mixedargs_sum(1, k1=2))
110+
self.assertEquals(1+2+3, mixedargs_sum(1, 2, k1=3))
111+
self.assertEquals(1+2+3+4, mixedargs_sum(1, 2, k1=3, k2=4))
112+
self.assertEquals(1+2+3+4+5, mixedargs_sum(1, 2, 3, k1=4, k2=5))
113+
114+
self.assertEquals(1+2, mixedargs_sum2(1, k1=2))
115+
self.assertEquals(1+2+3, mixedargs_sum2(1, 2, k1=3))
116+
self.assertEquals(1+2+3+4, mixedargs_sum2(1, 2, k1=3, k2=4))
117+
self.assertEquals(1+2+3+4+5, mixedargs_sum2(1, 2, 3, k1=4, k2=5))
118+
self.assertEquals(1+2+3+4+5+6,
119+
mixedargs_sum2(1, 2, 3, k1=4, k2=5, k3=6))
120+
self.assertEquals(1+2+3+4+5+6,
121+
mixedargs_sum2(1, 2, 3, k1=4, **{'k2':5, 'k3':6}))
122+
123+
self.assertEquals(1, Foo(k1=1).sum())
124+
self.assertEquals(1+2, Foo(k1=1,k2=2).sum())
125+
126+
self.assertEquals([1,2,3], sortnum(3,2,1))
127+
self.assertEquals([3,2,1], sortnum(1,2,3, reverse=True))
128+
129+
self.assertEquals(['a','b','c'], sortwords('a','c','b'))
130+
self.assertEquals(['c','b','a'], sortwords('a','c','b', reverse=True))
131+
self.assertEquals(['c','b','a'],
132+
sortwords('a','c','b', reverse=True, ignore='ignore'))
133+
134+
def testKwDefaults(self):
135+
def foo(p1,p2=0, *, k1, k2=0):
136+
return p1 + p2 + k1 + k2
137+
138+
self.assertEquals(2, foo.func_code.co_kwonlyargcount)
139+
self.assertEquals({"k2":0}, foo.func_kwdefaults)
140+
foo.func_kwdefaults = {"k1":0}
141+
try:
142+
foo(1,k1=10)
143+
self.fail("func_kwdefaults is not properly changed")
144+
except TypeError:
145+
pass
146+
147+
def test_main():
148+
run_unittest(KeywordOnlyArgTestCase)
149+
150+
if __name__ == "__main__":
151+
test_main()

0 commit comments

Comments
 (0)
0