X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/c75abed63ef284d7de54db87777951d6c668eefc..da8a5bb1895e5434695d9dc2d844119cd8f88524:/tests/test_primer.py

diff --git a/tests/test_primer.py b/tests/test_primer.py
index e7f99fd..dc30a7a 100644
--- a/tests/test_primer.py
+++ b/tests/test_primer.py
@@ -11,7 +11,7 @@ from pathlib import Path
 from platform import system
 from subprocess import CalledProcessError
 from tempfile import TemporaryDirectory, gettempdir
-from typing import Any, Callable, Generator, Iterator, Tuple
+from typing import Any, Callable, Iterator, Tuple
 from unittest.mock import Mock, patch
 
 from click.testing import CliRunner
@@ -44,7 +44,9 @@ FAKE_PROJECT_CONFIG = {
 
 
 @contextmanager
-def capture_stdout(command: Callable, *args: Any, **kwargs: Any) -> Generator:
+def capture_stdout(
+    command: Callable[..., Any], *args: Any, **kwargs: Any
+) -> Iterator[str]:
     old_stdout, sys.stdout = sys.stdout, StringIO()
     try:
         command(*args, **kwargs)