diff options
Diffstat (limited to 'tests/primer')
| -rw-r--r-- | tests/primer/test_primer_stdlib.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/primer/test_primer_stdlib.py b/tests/primer/test_primer_stdlib.py index 6cae6fd36..7496e67c8 100644 --- a/tests/primer/test_primer_stdlib.py +++ b/tests/primer/test_primer_stdlib.py @@ -2,10 +2,13 @@ # For details: https://github.com/PyCQA/pylint/blob/main/LICENSE # Copyright (c) https://github.com/PyCQA/pylint/blob/main/CONTRIBUTORS.txt +from __future__ import annotations + import contextlib import io import os import sys +from collections.abc import Iterator import pytest from pytest import CaptureFixture @@ -22,7 +25,7 @@ def is_package(filename: str, location: str) -> bool: @contextlib.contextmanager -def _patch_stdout(out): +def _patch_stdout(out: io.StringIO) -> Iterator[None]: sys.stdout = out try: yield |