Files
mlflow--mlflow/dev/clint/tests/rules/test_unnamed_thread_pool.py
2026-07-13 13:22:34 +08:00

24 lines
657 B
Python

from pathlib import Path
from clint.config import Config
from clint.index import SymbolIndex
from clint.linter import Position, Range, lint_file
from clint.rules import UnnamedThreadPool
def test_thread_pool_executor(index: SymbolIndex) -> None:
code = """
from concurrent.futures import ThreadPoolExecutor
# Bad
ThreadPoolExecutor()
# Good
ThreadPoolExecutor(thread_name_prefix="worker")
"""
config = Config(select={UnnamedThreadPool.name})
results = lint_file(Path("test.py"), code, config, index)
assert len(results) == 1
assert isinstance(results[0].rule, UnnamedThreadPool)
assert results[0].range == Range(Position(4, 0))