如何告诉 PyCharm 异步夹具返回了一些东西

Question

例子：

import pytest


@pytest.fixture
async def phrase():
    return 'hello world'


@pytest.fixture
async def replaced(phrase):
    return phrase.replace('hello', 'goodbye')

方法.replace 是黄色的，警告说：

Unresolved attribute reference 'replace' for class 'Coroutine'

但是，这些装置正在工作。如果我从def phrase(): 中删除async，Pycharm 正在正确处理.replace，表明它是str 类的方法。有没有办法告诉 PyCharm phrase 在replaced 中使用时将是str 的一个实例，而不是Coroutine？最好不要为每个使用 phrase 的夹具重复代码。

Answer 1

这不是您的代码，而是 Pycharm 问题 - 它无法正确解析本机协程装置的返回类型。 Pycharm 将解决旧的基于生成器的协程夹具

@pytest.fixture
async def phrase():
    yield 'hello world'

作为Generator[str, Any, None] 并将参数映射到夹具的返回类型。但是，原生协程夹具

@pytest.fixture
async def phrase():
    return 'hello world'

是 Coroutine[Any, Any, str]，目前，Pycharm 不会将测试参数映射到其返回类型（使用 Pycharm CE 2019.1 测试）。因此，您有两种可能：

设置显式类型提示

既然你知道协程应该返回什么，设置 return 和 arg 类型，Pycharm 就会停止猜测。这是最直接、最可靠的方法：

@pytest.fixture
async def phrase() -> str:
    return 'hello world'


@pytest.fixture
async def replaced(phrase: str) -> str:
    return phrase.replace('hello', 'goodbye')

切换到基于生成器的协程装置

这意味着yielding 而不是我在 cmets 中建议的returning；但是，您是否应该更改明显正确的代码来解决 Pycharm 的问题，这取决于您。