From 9f000715aceb413e692b9918e2077b28f1fae96a Mon Sep 17 00:00:00 2001 From: Daniel Hahler Date: Fri, 10 Apr 2020 08:14:08 +0200 Subject: [PATCH] typing: python: pytest_pyfunc_call --- src/_pytest/python.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/_pytest/python.py b/src/_pytest/python.py index 45e078aac08..6ff314839e7 100644 --- a/src/_pytest/python.py +++ b/src/_pytest/python.py @@ -61,6 +61,8 @@ from _pytest.warning_types import PytestUnhandledCoroutineWarning if TYPE_CHECKING: + from typing_extensions import Literal + from _pytest._io import TerminalWriter @@ -183,7 +185,7 @@ def async_warn(pyfuncitem: "Function") -> None: @hookimpl(trylast=True) -def pytest_pyfunc_call(pyfuncitem: "Function"): +def pytest_pyfunc_call(pyfuncitem: "Function") -> "Literal[True]": testfunction = pyfuncitem.obj if iscoroutinefunction(testfunction) or ( sys.version_info >= (3, 6) and inspect.isasyncgenfunction(testfunction)