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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
# Copyright (c) 2016-2018, 2020 Claudiu Popa <pcmanticore@gmail.com>
# Copyright (c) 2016, 2019 Ashley Whetter <ashley@awhetter.co.uk>
# Copyright (c) 2016 Derek Gustafson <degustaf@gmail.com>
# Copyright (c) 2018 Anthony Sottile <asottile@umich.edu>
# Copyright (c) 2019 Hugo van Kemenade <hugovk@users.noreply.github.com>
# Copyright (c) 2019 Pierre Sassoulas <pierre.sassoulas@gmail.com>
# Copyright (c) 2020 hippo91 <guillaume.peillex@gmail.com>
# Copyright (c) 2020 Damien Baty <damien.baty@polyconseil.fr>
# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
# For details: https://github.com/PyCQA/pylint/blob/master/COPYING
"""Unit tests for the pylint checkers in :mod:`pylint.extensions.check_docs`,
in particular the parameter documentation checker `DocstringChecker`
"""
import astroid
import pytest
from pylint.extensions import _check_docs_utils as utils
@pytest.mark.parametrize(
"string,count",
[("abc", 0), ("", 0), (" abc", 2), ("\n abc", 0), (" \n abc", 3)],
)
def test_space_indentation(string, count):
"""Test for pylint_plugin.ParamDocChecker"""
assert utils.space_indentation(string) == count
@pytest.mark.parametrize(
"raise_node,expected",
[
(
astroid.extract_node(
"""
def my_func():
raise NotImplementedError #@
"""
),
{"NotImplementedError"},
),
(
astroid.extract_node(
"""
def my_func():
raise NotImplementedError("Not implemented!") #@
"""
),
{"NotImplementedError"},
),
(
astroid.extract_node(
"""
def my_func():
try:
fake_func()
except RuntimeError:
raise #@
"""
),
{"RuntimeError"},
),
(
astroid.extract_node(
"""
def my_func():
try:
fake_func()
except RuntimeError:
if another_func():
raise #@
"""
),
{"RuntimeError"},
),
(
astroid.extract_node(
"""
def my_func():
try:
fake_func()
except RuntimeError:
try:
another_func()
raise #@
except NameError:
pass
"""
),
{"RuntimeError"},
),
(
astroid.extract_node(
"""
def my_func():
try:
fake_func()
except RuntimeError:
try:
another_func()
except NameError:
raise #@
"""
),
{"NameError"},
),
(
astroid.extract_node(
"""
def my_func():
try:
fake_func()
except:
raise #@
"""
),
set(),
),
(
astroid.extract_node(
"""
def my_func():
try:
fake_func()
except (RuntimeError, ValueError):
raise #@
"""
),
{"RuntimeError", "ValueError"},
),
(
astroid.extract_node(
"""
import not_a_module
def my_func():
try:
fake_func()
except not_a_module.Error:
raise #@
"""
),
set(),
),
],
)
def test_exception(raise_node, expected):
found = utils.possible_exc_types(raise_node)
assert found == expected
|
