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

22 lines
655 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.pytest_mark_repeat import PytestMarkRepeat
def test_pytest_mark_repeat(index: SymbolIndex) -> None:
code = """
import pytest
@pytest.mark.repeat(10)
def test_flaky_function():
...
"""
config = Config(select={PytestMarkRepeat.name})
violations = lint_file(Path("test_pytest_mark_repeat.py"), code, config, index)
assert len(violations) == 1
assert all(isinstance(v.rule, PytestMarkRepeat) for v in violations)
assert violations[0].range == Range(Position(3, 1))