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
|
# 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`
"""
from __future__ import division, print_function, absolute_import
import unittest
import sys
import astroid
from astroid import test_utils
from pylint.testutils import CheckerTestCase, Message, set_config
import pylint.extensions._check_docs_utils as utils
class SpaceIndentationTest(unittest.TestCase):
"""Tests for pylint_plugin.ParamDocChecker"""
def test_space_indentation(self):
self.assertEqual(utils.space_indentation('abc'), 0)
self.assertEqual(utils.space_indentation(''), 0)
self.assertEqual(utils.space_indentation(' abc'), 2)
self.assertEqual(utils.space_indentation('\n abc'), 0)
self.assertEqual(utils.space_indentation(' \n abc'), 3)
class PossibleExcTypesText(unittest.TestCase):
def test_exception_class(self):
raise_node = test_utils.extract_node('''
def my_func():
raise NotImplementedError #@
''')
found = utils.possible_exc_types(raise_node)
expected = set(["NotImplementedError"])
self.assertEqual(found, expected)
def test_exception_instance(self):
raise_node = test_utils.extract_node('''
def my_func():
raise NotImplementedError("Not implemented!") #@
''')
found = utils.possible_exc_types(raise_node)
expected = set(["NotImplementedError"])
self.assertEqual(found, expected)
def test_rethrow(self):
raise_node = test_utils.extract_node('''
def my_func():
try:
fake_func()
except RuntimeError:
raise #@
''')
found = utils.possible_exc_types(raise_node)
expected = set(["RuntimeError"])
self.assertEqual(found, expected)
def test_nested_in_if_rethrow(self):
raise_node = test_utils.extract_node('''
def my_func():
try:
fake_func()
except RuntimeError:
if another_func():
raise #@
''')
found = utils.possible_exc_types(raise_node)
expected = set(["RuntimeError"])
self.assertEqual(found, expected)
def test_nested_in_try(self):
raise_node = test_utils.extract_node('''
def my_func():
try:
fake_func()
except RuntimeError:
try:
another_func()
raise #@
except NameError:
pass
''')
found = utils.possible_exc_types(raise_node)
expected = set(["RuntimeError"])
self.assertEqual(found, expected)
def test_nested_in_try_except(self):
raise_node = test_utils.extract_node('''
def my_func():
try:
fake_func()
except RuntimeError:
try:
another_func()
except NameError:
raise #@
''')
found = utils.possible_exc_types(raise_node)
expected = set(["NameError"])
self.assertEqual(found, expected)
def test_no_rethrow_types(self):
raise_node = test_utils.extract_node('''
def my_func():
try:
fake_func()
except:
raise #@
''')
found = utils.possible_exc_types(raise_node)
expected = set()
self.assertEqual(found, expected)
def test_multiple_rethrow_types(self):
raise_node = test_utils.extract_node('''
def my_func():
try:
fake_func()
except (RuntimeError, ValueError):
raise #@
''')
found = utils.possible_exc_types(raise_node)
expected = set(["RuntimeError", "ValueError"])
self.assertEqual(found, expected)
def test_ignores_uninferable_type(self):
raise_node = astroid.extract_node('''
import not_a_module
def my_func():
try:
fake_func()
except not_a_module.Error:
raise #@
''')
found = utils.possible_exc_types(raise_node)
expected = set()
self.assertEqual(found, expected)
if __name__ == '__main__':
unittest.main()
|
