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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
from pluggy._hooks import varnames
from pluggy._manager import _formatdef
def test_varnames():
def f(x):
i = 3 # noqa
class A:
def f(self, y):
pass
class B:
def __call__(self, z):
pass
assert varnames(f) == (("x",), ())
assert varnames(A().f) == (("y",), ())
assert varnames(B()) == (("z",), ())
def test_varnames_default():
def f(x, y=3):
pass
assert varnames(f) == (("x",), ("y",))
def test_varnames_class():
class C:
def __init__(self, x):
pass
class D:
pass
class E:
def __init__(self, x):
pass
class F:
pass
assert varnames(C) == (("x",), ())
assert varnames(D) == ((), ())
assert varnames(E) == (("x",), ())
assert varnames(F) == ((), ())
def test_varnames_keyword_only():
def f1(x, *, y):
pass
def f2(x, *, y=3):
pass
def f3(x=1, *, y=3):
pass
assert varnames(f1) == (("x",), ())
assert varnames(f2) == (("x",), ())
assert varnames(f3) == ((), ("x",))
def test_formatdef():
def function1():
pass
assert _formatdef(function1) == "function1()"
def function2(arg1):
pass
assert _formatdef(function2) == "function2(arg1)"
def function3(arg1, arg2="qwe"):
pass
assert _formatdef(function3) == "function3(arg1, arg2='qwe')"
def function4(arg1, *args, **kwargs):
pass
assert _formatdef(function4) == "function4(arg1, *args, **kwargs)"
|
