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
85
86
87
88
89
90
|
from pluggy.hooks import varnames
from pluggy.manager import _formatdef
import sys
import pytest
def test_varnames():
def f(x):
i = 3 # noqa
class A(object):
def f(self, y):
pass
class B(object):
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(object):
def __init__(self, x):
pass
class D(object):
pass
class E(object):
def __init__(self, x):
pass
class F(object):
pass
assert varnames(C) == (("x",), ())
assert varnames(D) == ((), ())
assert varnames(E) == (("x",), ())
assert varnames(F) == ((), ())
@pytest.mark.skipif(
sys.version_info < (3,), reason="Keyword only arguments are Python 3 only"
)
def test_varnames_keyword_only():
# SyntaxError on Python 2, so we exec
ns = {}
exec(
"def f1(x, *, y): pass\n"
"def f2(x, *, y=3): pass\n"
"def f3(x=1, *, y=3): pass\n",
ns,
)
assert varnames(ns["f1"]) == (("x",), ())
assert varnames(ns["f2"]) == (("x",), ())
assert varnames(ns["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)"
|
