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
|
# mypy: ignore-errors
from __future__ import annotations
import sys
from pathlib import Path
from token import NAME
from tokenize import TokenInfo
from tools.linter.adapters._linter import PythonFile
from tools.linter.adapters.set_linter import PythonLines, SetLinter
_PARENT = Path(__file__).parent.absolute()
_PATH = [Path(p).absolute() for p in sys.path]
if _PARENT in _PATH:
from linter_test_case import LinterTestCase
else:
from .linter_test_case import LinterTestCase
TESTDATA = Path("tools/test/set_linter_testdata")
TESTFILE = TESTDATA / "python_code.py.txt"
INCLUDES_FILE = TESTDATA / "includes.py.txt"
INCLUDES_FILE2 = TESTDATA / "includes_doesnt_change.py.txt"
FILES = TESTFILE, INCLUDES_FILE, INCLUDES_FILE2
def python_lines(p: str | Path) -> PythonLines:
pf = PythonFile.make(SetLinter.linter_name, p)
return PythonLines(pf)
class TestSetLinter(LinterTestCase):
maxDiff = 10000000
LinterClass = SetLinter
def test_get_all_tokens(self) -> None:
self.assertEqual(EXPECTED_SETS, python_lines(TESTFILE).sets)
def test_omitted_lines(self) -> None:
actual = sorted(python_lines(TESTFILE).omitted.omitted)
expected = [3, 13]
self.assertEqual(expected, actual)
def test_linting(self) -> None:
for path in (TESTFILE, INCLUDES_FILE, INCLUDES_FILE2):
with self.subTest(path):
r = self.lint_fix_test(path, [])
self.assertEqual(r.name, "Suggested fixes for set_linter")
def test_bracket_pairs(self) -> None:
TESTS: tuple[tuple[str, dict[int, int]], ...] = (
("", {}),
("{}", {0: 1}),
("{1}", {0: 2}),
("{1, 2}", {0: 4}),
("{1: 2}", {0: 4}),
("{One()}", {0: 4, 2: 3}),
(
"{One({1: [2], 2: {3}, 3: {4: 5}})}",
{0: 25, 2: 24, 3: 23, 6: 8, 12: 14, 18: 22},
),
)
for i, (s, expected) in enumerate(TESTS):
pl = python_lines(s)
if s:
actual = pl.token_lines[0].bracket_pairs
else:
self.assertEqual(pl.token_lines, [])
actual = {}
self.assertEqual(actual, expected)
def test_match_braced_sets(self) -> None:
TESTS: tuple[tuple[str, int], ...] = (
("{cast(int, inst.offset): inst for inst in instructions}", 0),
("", 0),
("{}", 0),
("{1: 0}", 0),
("{1}", 1),
("{i for i in range(2, 3)}", 1),
("{1, 2}", 1),
("{One({'a': 1}), Two([{}, {2}, {1, 2}])}", 3),
)
for i, (s, expected) in enumerate(TESTS):
pl = python_lines(s)
actual = pl.token_lines and pl.token_lines[0].braced_sets
self.assertEqual(len(actual), expected)
EXPECTED_SETS = [
TokenInfo(NAME, "set", (4, 4), (4, 7), "a = set()\n"),
TokenInfo(NAME, "set", (6, 4), (6, 7), "c = set\n"),
TokenInfo(NAME, "set", (9, 3), (9, 6), " set(\n"),
]
|
