如何解决如何使用pytest参数化测试的解包参数在mypy上打字注释?
我在 my_module.py
上有一个带有签名的简单方法:
def my_method(value: float,extra: str = "something") -> str:
pass
然后我对 test_my_module.py
进行了参数化测试:
from typing import List
from typing import Union
import pytest
import my_module
@pytest.mark.parametrize(
"params,expected",[
([1.0],"result1"),([2.0],"result2"),([2.0,"extra"],"result3")
],)
def test_my_method(params: List[Union[int,str]],expected: str) -> None:
assert my_module.my_method(*params) == expected
当我运行诸如 typeguard
之类的动态类型测试时,它通过了。但是使用 mypy
我得到这些 [arg-type]
错误:
tests/test_my_module.py: note: In function "test_my_method":
tests/test_my_module.py:50:30: error: Argument 1 to "my_method" has incompatible type "*List[Union[int,str]]"; expected "float"
[arg-type]
assert my_module.my_method(*params) == expected
^
tests/test_my_module.py:50:30: error: Argument 1 to "my_method" has incompatible type "*List[Union[int,str]]"; expected "str"
[arg-type]
assert my_module.my_method(*params) == expected
^
不太确定我应该如何注释这些类型以使其通过。请注意,这是一个最小的示例,因此我想继续解压缩我创建的列表中的参数列表 params
,以对测试进行参数化。
有什么想法吗?
解决方法
您可以使用 Tuple
而不是 List
进行注释:
@pytest.mark.parametrize(
"params,expected",[
([1.0],"result1"),([2.0],"result2"),([2.0,"extra"],"result3")
],)
def test_my_method(params: Union[Tuple[float],Tuple[float,str]],expected: str) -> None:
assert my_module.my_method(*params) == expected
这承诺了固定数量的元素,因此静态类型检查器可以推断签名是兼容的。
,我建议将您的变量参数表示为元组而不是列表。类型当前不允许列表在特定索引处具有特定类型,但是固定大小的元组可以为其维度指定类型
您当前拥有的两个形状要么是单个浮点数:Tuple[float]
,要么是浮点数和字符串:Tuple[float,str]
-- 通过将您的参数化调用和参数转换为此,它应该可以很好地工作:>
from typing import Tuple
from typing import Union
import pytest
import my_module
@pytest.mark.parametrize(
"params,[
((1.0,),((2.0,"extra"),expected: str) -> None:
assert my_module.my_method(*params) == expected
拆分你的测试也可能是有意义的——你可能在这里过度使用了参数化,并且为你的固定参数设置一组测试并为你的可选参数设置另一组测试可能更清楚(而不是试图将所有这些测试与参数化结合在一起)——也许是这样的:
@pytest.mark.parametrize(
"arg,[
(1.0,(2.0,],)
def test_my_method_one_arg(arg: float,expected: str) -> None:
assert my_module.my_method(arg) == expected
@pytest.mark.parametrize(
"arg,extra,"extra","result3"),"result4"),)
def test_my_method_extra_arg(arg: float,extra: str,expected: str) -> None:
assert my_module.my_method(arg,extra) == expected
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。