提示复合假设策略的类型

Question

我正在使用

hypothesis

库，我想用类型提示来注释我的代码。 docs 提到

hypothesis.strategies.SearchStrategy

作为所有搜索策略的类型。

举这个例子：

@composite
def int_strategy(draw: DrawFn) -> hypothesis.strategies.SearchStrategy[int]:
    ... # some computation here resulting in ``x`` being an ``int``

    return x

运行

mypy

将（正确地）导致以下错误：

error: Returning Any from function declared to return "SearchStrategy[Any]"  [no-any-return]

我的意思是，我实际上返回的是

int

，而不是

SearchStrategy

。

我该如何输入注释我的

hypothesis

策略？

Answer 1

您的函数需要返回 SearchStrategy 实例而不是实际值。

def int_strategy() -> hypothesis.strategies.SearchStrategy[int]:
    ... # some computation here resulting in ``x`` being an ``int``

    return st.integers()

Answer 2

用

@composite

修饰的函数应该像平常一样进行类型提示：

@composite
def int_strategy(draw: DrawFn) -> int:
    ...

@composite

然后会自动将其转换为：

# As if it doesn't have the `draw` parameter and that it returns a `SearchStrategy`
def int_strategy() -> SearchStrategy[int]:
    ...

这与其他装饰器相同：函数的最终类型由其原始类型提示及其所有

@decorator

决定。在

composite()

的情况下，这就是 的定义方式（至少在类型检查时）：

# Takes a function whose first argument is `DrawFn`
# and returns a function without that argument,
# returning a `SearchStrategy` that will output
# values of the same type as the original's return type.
def composite(
    f: Callable[Concatenate[DrawFn, P], Ex]
) -> Callable[P, SearchStrategy[Ex]]:
    ...