p53209761
/
test_ok2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixtures: type annotate get_scope_package

This commit is contained in:
Ran Benita 2023-05-14 22:47:29 +03:00
parent ac699e7b25
commit 1716d3c9bf
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/_pytest/fixtures.py
View File

@ -113,16 +113,19 @@ def pytest_sessionstart(session: "Session") -> None:
session._fixturemanager = FixtureManager(session) session._fixturemanager = FixtureManager(session)
def get_scope_package(node, fixturedef: "FixtureDef[object]"): def get_scope_package(
node: nodes.Item,
fixturedef: "FixtureDef[object]",
) -> Optional[Union[nodes.Item, nodes.Collector]]:
import pytest import pytest
cls = pytest.Package cls = pytest.Package
current = node current: Optional[Union[nodes.Item, nodes.Collector]] = node
fixture_package_name = "{}/{}".format(fixturedef.baseid, "__init__.py") fixture_package_name = "{}/{}".format(fixturedef.baseid, "__init__.py")
while current and ( while current and (
type(current) is not cls or fixture_package_name != current.nodeid type(current) is not cls or fixture_package_name != current.nodeid
): ):
current = current.parent current = current.parent # type: ignore[assignment]
if current is None: if current is None:
return node.session return node.session
return current return current