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
|
import ast
import pycodestyle
import pytest
from flake8_import_order import STDLIB_NAMES
from flake8_import_order.checker import ImportOrderChecker
def _load_test_cases():
test_cases = []
for name in STDLIB_NAMES:
if not name.startswith("__"):
test_cases.append(name)
return test_cases
def _checker(data):
pycodestyle.stdin_get_value = lambda: data
tree = ast.parse(data)
checker = ImportOrderChecker(None, tree)
checker.options = {}
return checker
@pytest.mark.parametrize('import_name', _load_test_cases())
def test_styles(import_name):
data = "import {}\nimport zoneinfo\n".format(import_name)
checker = _checker(data)
codes = [error.code for error in checker.check_order()]
assert codes == []
|
