1  from pathlib import Path
 2  
 3  from clint.config import Config
 4  from clint.index import SymbolIndex
 5  from clint.linter import Position, Range, lint_file
 6  from clint.rules.pytest_mark_repeat import PytestMarkRepeat
 7  
 8  
 9  def test_pytest_mark_repeat(index: SymbolIndex) -> None:
10      code = """
11  import pytest
12  
13  @pytest.mark.repeat(10)
14  def test_flaky_function():
15      ...
16  """
17      config = Config(select={PytestMarkRepeat.name})
18      violations = lint_file(Path("test_pytest_mark_repeat.py"), code, config, index)
19      assert len(violations) == 1
20      assert all(isinstance(v.rule, PytestMarkRepeat) for v in violations)
21      assert violations[0].range == Range(Position(3, 1))