pytest 在不返回对象的情况下执行操作的装置的较窄类型？

Question

考虑像这样的夹具

@pytest.fixture
def mock_database(monkeypatch: MonkeyPatch) -> None:
    ...

它在测试中使用

def test_with_mock_database(mock_database: None) -> None:
    ...

测试参数中mock_database的类型是什么？ 它在运行时似乎实际上只是一个None ，但这使得它有点难以缩小范围，如果我想区分例如SetupFixture和TeardownFixture ，因为没有 class Fixture（None）和投射黑客喜欢

class _FixtureType:
    pass

FixtureType = cast(_FixtureType, None)

@pytest.fixture
def mock_database(monkeypatch: MonkeyPatch) -> FixtureType:
    ...

有点休息，因为我现在必须return一些东西。

还有其他更优雅的解决方案吗？

Answer 1

您可以使用typing.TypeAlias在视觉上区分灯具。 您是否可以有效地浏览代码库或正确地 lint 是另一个问题。

"""my_project_types.py"""
import typing as t

# Note: `TypeAlias` is not suitable for `is` comparisons (don't write `if my_value is SetupFixture`)
SetupFixture: t.TypeAlias = None

from my_project_types import SetupFixture

@pytest.fixture
def mock_database(monkeypatch: MonkeyPatch) -> SetupFixture:
    """No explicit return or yield required here"""

from my_project_types import SetupFixture

def test_with_mock_database(mock_database: SetupFixture) -> None:
    ...