如何让Pylance自动为pytest夹具参数提供类型提示？

先看示例代码：

fixtures.py

@pytest.fixture
def hello_world():
    return "Hello world!"

basic_tests.py

def test_says_hello(hello_world):
    assert hello_world == "Hello world!"

问题描述 #

目前Pylance没法自动关联测试函数里的参数和对应夹具的返回类型——比如上面的hello_world参数，Pylance不知道它是str类型。简单类型手动加标注还好，但如果是大量返回复杂对象的夹具，手动写类型既麻烦又影响代码可读性。

解决方案 #

你不需要修改Pylance的核心代码，只需要给pytest的夹具补充类型信息，让Pylance能通过标准类型机制识别。下面是几种可行思路：

1. 给夹具直接加返回类型标注

这是最基础的做法，先给夹具函数补上返回类型：

# fixtures.py
@pytest.fixture
def hello_world() -> str:
    return "Hello world!"

如果是复杂类型，可以先定义类型别名再复用，避免重复写长类型：

from typing import TypeAlias

ComplexType: TypeAlias = dict[str, list[int]]

@pytest.fixture
def complex_fixture() -> ComplexType:
    return {"key": [1,2,3]}

2. 用自定义装饰器批量绑定类型

如果不想每个夹具都手动写返回类型，可以写个装饰器自动给夹具绑定类型，同时让Pylance能识别：

from typing import TypeVar, Any
import pytest

T = TypeVar("T")

def typed_fixture(return_type: type[T]) -> Any:
    def wrapper(func):
        # 给夹具函数添加返回类型注解
        func.__annotations__["return"] = return_type
        return pytest.fixture(func)
    return wrapper

# 使用示例
@typed_fixture(str)
def hello_world():
    return "Hello world!"

这样Pylance会读取夹具函数的__annotations__属性，自动推断测试参数的类型。

3. 写类型 stub 文件（无需修改原代码）

如果不想改动现有夹具代码，可以给fixtures文件写对应的.pyi stub文件：

# fixtures.pyi
import pytest

@pytest.fixture
def hello_world() -> str: ...

@pytest.fixture
def complex_fixture() -> dict[str, list[int]]: ...

Pylance会自动读取stub里的类型信息，不用修改原代码就能获得类型提示。

4. 扩展Pyright/Pylance的类型推断逻辑

如果你想更进一步，不用手动加任何注解就能让Pylance识别，可以写Pyright插件（Pylance基于Pyright）。核心逻辑是：

• 扫描所有被@pytest.fixture装饰的函数，分析它们的返回值类型
• 给测试函数中同名的参数自动添加对应的类型注解
• 通过Pyright的扩展接口把这些类型信息注入到类型检查流程里

关键结论 #

你不需要修改Pylance的源码，只要让夹具提供标准的类型注解（不管是直接写、用装饰器还是stub文件），Pylance就能自动识别测试参数的类型。如果要做全自动的类型推断，通过Pyright插件扩展是可行的，不用碰Pylance的核心代码。

内容的提问来源于stack exchange，提问作者Tomáš Zato