diff --git a/pytest_examples/config.py b/pytest_examples/config.py index 7470a7d..6b37a55 100644 --- a/pytest_examples/config.py +++ b/pytest_examples/config.py @@ -28,6 +28,7 @@ class ExamplesConfig: ruff_select: list[str] | None = None ruff_ignore: list[str] | None = None white_space_dot: bool = False + first_party: list[str] | None = None """If True, replace spaces with `ยท` in example diffs.""" def black_mode(self): @@ -69,10 +70,11 @@ def ruff_config(self) -> tuple[str, ...]: if self.upgrade: select.append('UP') - if self.isort: + if self.isort and 'I' not in select: select.append('I') - if self.ruff_ignore: + if self.first_party: + args.append(f'--first-party={",".join(self.first_party)}') ignore.extend(self.ruff_ignore) if select: