28 lines
818 B
Python
28 lines
818 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 UseSysExecutable
|
|
|
|
|
|
def test_use_sys_executable(index: SymbolIndex) -> None:
|
|
code = """
|
|
import subprocess
|
|
import sys
|
|
|
|
# Bad
|
|
subprocess.run(["mlflow", "ui"])
|
|
subprocess.check_call(["mlflow", "ui"])
|
|
|
|
# Good
|
|
subprocess.run([sys.executable, "-m", "mlflow", "ui"])
|
|
subprocess.check_call([sys.executable, "-m", "mlflow", "ui"])
|
|
"""
|
|
config = Config(select={UseSysExecutable.name})
|
|
results = lint_file(Path("test.py"), code, config, index)
|
|
assert len(results) == 2
|
|
assert all(isinstance(r.rule, UseSysExecutable) for r in results)
|
|
assert results[0].range == Range(Position(5, 0))
|
|
assert results[1].range == Range(Position(6, 0))
|