hookspec: type annotate parent argument to pytest_collect_file

This commit is contained in:
Ran Benita 2020-08-20 23:51:08 +03:00
parent 4b8e1a1771
commit 57aca11d4a
3 changed files with 10 additions and 5 deletions

View File

@ -32,6 +32,7 @@ from _pytest.compat import TYPE_CHECKING
from _pytest.config import Config
from _pytest.config.argparsing import Parser
from _pytest.fixtures import FixtureRequest
from _pytest.nodes import Collector
from _pytest.outcomes import OutcomeException
from _pytest.pathlib import import_path
from _pytest.python_api import approx
@ -118,7 +119,7 @@ def pytest_unconfigure() -> None:
def pytest_collect_file(
path: py.path.local, parent
path: py.path.local, parent: Collector,
) -> Optional[Union["DoctestModule", "DoctestTextfile"]]:
config = parent.config
if path.ext == ".py":

View File

@ -274,10 +274,12 @@ def pytest_ignore_collect(path: py.path.local, config: "Config") -> Optional[boo
"""
def pytest_collect_file(path: py.path.local, parent) -> "Optional[Collector]":
"""Return collection Node or None for the given path.
def pytest_collect_file(
path: py.path.local, parent: "Collector"
) -> "Optional[Collector]":
"""Create a Collector for the given path, or None if not relevant.
Any new node needs to have the specified ``parent`` as a parent.
The new node needs to have the specified ``parent`` as a parent.
:param py.path.local path: The path to collect.
"""

View File

@ -184,7 +184,9 @@ def pytest_pyfunc_call(pyfuncitem: "Function") -> Optional[object]:
return True
def pytest_collect_file(path: py.path.local, parent) -> Optional["Module"]:
def pytest_collect_file(
path: py.path.local, parent: nodes.Collector
) -> Optional["Module"]:
ext = path.ext
if ext == ".py":
if not parent.session.isinitpath(path):