wehub-resource-sync
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
from clint.config import Config
|
|
from clint.index import SymbolIndex
|
|
from clint.linter import lint_file
|
|
from clint.rules import PreferDictUnion
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"code",
|
|
[
|
|
pytest.param("{**dict1, **dict2}", id="two_dict_unpacks"),
|
|
pytest.param("{**dict1, **dict2, **dict3}", id="three_dict_unpacks"),
|
|
pytest.param("{**a, **b, **c, **d}", id="four_dict_unpacks"),
|
|
pytest.param("{**obj.attr, **other}", id="attribute_and_name"),
|
|
pytest.param("{**a.x, **b.y}", id="two_attributes"),
|
|
pytest.param("{**a.b.c, **d}", id="chained_attribute"),
|
|
],
|
|
)
|
|
def test_flag(index: SymbolIndex, code: str) -> None:
|
|
config = Config(select={PreferDictUnion.name})
|
|
results = lint_file(Path("test.py"), code, config, index)
|
|
assert len(results) == 1
|
|
assert isinstance(results[0].rule, PreferDictUnion)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"code",
|
|
[
|
|
pytest.param("{**dict1}", id="single_dict_unpack"),
|
|
pytest.param('{**dict1, "extra_key": "value"}', id="unpack_then_literal"),
|
|
pytest.param('{"key": "value", **dict1}', id="literal_then_unpack"),
|
|
pytest.param('{"key1": "value1", "key2": "value2"}', id="only_literals"),
|
|
pytest.param("{}", id="empty_dict"),
|
|
pytest.param('{**dict1, "k1": "v1", **dict2, "k2": "v2"}', id="mixed_unpacks_and_literals"),
|
|
pytest.param('{**dict1, **dict2, "override": "value"}', id="unpacks_with_trailing_literal"),
|
|
pytest.param("{**data[0], **other}", id="subscript_access"),
|
|
pytest.param('{**configs, **{"key": "value"}}', id="dict_literal_unpack"),
|
|
pytest.param("{**func(), **other}", id="function_call"),
|
|
pytest.param("{**a,\n**b}", id="multi_line"),
|
|
],
|
|
)
|
|
def test_no_flag(index: SymbolIndex, code: str) -> None:
|
|
config = Config(select={PreferDictUnion.name})
|
|
results = lint_file(Path("test.py"), code, config, index)
|
|
assert len(results) == 0
|