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
|
import ast
import glob
import os
import re
import sys
import pytest
from flake8_import_order.checker import ImportOrderChecker
from flake8_import_order.styles import lookup_entry_point
ERROR_RX = re.compile("# ((I[0-9]{3} ?)+) ?.*$")
def _extract_expected_errors(data):
lines = data.splitlines()
expected_codes = []
for line in lines:
match = ERROR_RX.search(line)
if match is not None:
codes = match.group(1).split()
expected_codes.extend(codes)
return expected_codes
def _load_test_cases():
base_path = os.path.dirname(__file__)
test_cases = []
test_case_path = os.path.join(base_path, "test_cases")
wildcard_path = os.path.join(test_case_path, "*.py")
for filename in glob.glob(wildcard_path):
# The namespace.py test only works with Python3
if filename.endswith("namespace.py") and sys.version_info.major < 3:
continue
fullpath = os.path.join(test_case_path, filename)
with open(fullpath) as file_:
data = file_.read()
styles = data.splitlines()[0].lstrip("#").strip().split()
codes = _extract_expected_errors(data)
tree = ast.parse(data, fullpath)
for style_name in styles:
style_entry_point = lookup_entry_point(style_name)
test_cases.append((filename, tree, style_entry_point, codes))
return test_cases
def _checker(filename, tree, style_entry_point):
options = {
"application_import_names": [
"flake8_import_order",
"namespace.package_b",
"tests",
],
"application_package_names": ["localpackage"],
"import_order_style": style_entry_point,
}
checker = ImportOrderChecker(filename, tree)
checker.options = options
return checker
@pytest.mark.parametrize(
"filename, tree, style, expected_codes",
_load_test_cases(),
)
def test_styles(filename, tree, style, expected_codes):
checker = _checker(filename, tree, style)
codes = [
(filename, error.lineno, error.code)
for error in checker.check_order()
]
assert [i[2] for i in codes] == expected_codes, codes
def test_unknown_style():
with pytest.raises(LookupError):
lookup_entry_point("Unknown")
|
