1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
'''
To make the life of any analysis easier, we are generating Param objects
instead of simple parser objects.
'''
from textwrap import dedent
from parso import parse
def assert_params(param_string, **wanted_dct):
source = dedent('''
def x(%s):
pass
''') % param_string
module = parse(source)
funcdef = next(module.iter_funcdefs())
dct = dict((p.name.value, p.default and p.default.get_code())
for p in funcdef.get_params())
assert dct == wanted_dct
assert module.get_code() == source
def test_split_params_with_separation_star():
assert_params('x, y=1, *, z=3', x=None, y='1', z='3')
assert_params('*, x', x=None)
assert_params('*')
def test_split_params_with_stars():
assert_params('x, *args', x=None, args=None)
assert_params('**kwargs', kwargs=None)
assert_params('*args, **kwargs', args=None, kwargs=None)
def test_kw_only_no_kw(works_in_py):
"""
Parsing this should be working. In CPython the parser also parses this and
in a later step the AST complains.
"""
module = works_in_py.parse('def test(arg, *):\n pass')
if module is not None:
func = module.children[0]
open_, p1, asterisk, close = func._get_param_nodes()
assert p1.get_code('arg,')
assert asterisk.value == '*'
|
