diff --git a/src/pytest_loguru/plugin.py b/src/pytest_loguru/plugin.py index 3ea4882..c0a3013 100644 --- a/src/pytest_loguru/plugin.py +++ b/src/pytest_loguru/plugin.py @@ -1,10 +1,12 @@ +from typing import Iterator # pragma: no cover + import pytest # pragma: no cover from _pytest.logging import LogCaptureFixture # pragma: no cover from loguru import logger # pragma: no cover @pytest.fixture -def caplog(caplog: LogCaptureFixture) -> None: +def caplog(caplog: LogCaptureFixture) -> Iterator[LogCaptureFixture]: """Emitting logs from loguru's logger.log means that they will not show up in caplog which only works with Python's standard logging. This adds the same LogCaptureHandler being used by caplog to hook into loguru. @@ -12,8 +14,8 @@ def caplog(caplog: LogCaptureFixture) -> None: Args: caplog (LogCaptureFixture): caplog fixture - Returns: - None + Yields: + LogCaptureFixture """ def filter_(record):