如何为 pytest 设备指定 mypy 类型

Question

我正在尝试为我在测试项目中使用的 pytest 本机装置指定 mypy 类型提示，例如：

import pytest

def pytest_configure(config):
    # Do something useful here

config 夹具返回一个 _pytest.config.Config 对象。如果我尝试天真地建模：

import pytest

def pytest_configure(config: Config) -> None:
    # Do something useful here

我收到一个 mypy 错误： conftest.py:3: error: Name 'Config' is not defined [name-defined]

我可以做from _pytest.config import Config，但这似乎不是一个好方法，因为 _pytest 是私有的。 另一种选择是忽略带有# type: ignore 的类型。如果这是推荐的方式，我当然会这样做，但我想知道是否有更好的选择。

我使用的任何类型的 pytest 本机装置都有相同的问题，例如request 用于参数化夹具。这将是_pytest.fixtures.FixtureRequest。

Answer 1

正在导入from _pytest.config

由于pytest 当前不导出Config（从6.2 开始），输入的唯一方法是使用from _pytest.config import Config。这也是我输入config 的方式，例如可以看到在this question of mine:

from _pytest.config import Config

def pytest_configure(config: Config) -> None:
    ...

您可以在pytest 问题中跟踪打字进度：#7469。

自定义类型存根

您还可以引入一个隐藏重新导出的小型自定义类型存根。在这里是否有用值得怀疑，只值得一提替代解决方案。如果您创建一个文件_typeshed/pytest.pyi，其内容如下：

from typing import Any
from _pytest.config import Config as Config

def __getattr__(name: str) -> Any: ...  # incomplete

并使其可供mypy in mypy.ini 访问：

[mypy]
mypy_path = _typeshed

现在您至少可以在类型检查模式下导入from pytest import Config - 运行时导入仍然会失败。所以导入看起来像

from typing import Any, TYPE_CHECKING

if TYPE_CHECKING:
    from pytest import Config
else:
    Config = Any


def pytest_configure(config: Config) -> None:
    pass

该解决方案的唯一好处是现在隐藏了私有导入；不过我还是会选择私人进口。