Python mypy 无法从联合返回类型推断类型

Question

这里是示例代码

from typing import Dict, Union, Tuple


def select_range(data: Dict[str, Union[str, int]]) -> Tuple[int, int]:
    if data['start'] and data['end']:
        return data['start'], data['end']
    return 1, 1

select_range({})

Mypy 输出：

mypy different_return.py
different_return.py:6: error: Incompatible return value type (got 
"Tuple[Union[str, int], Union[str, int]]", expected "Tuple[int, int]")

即使字典值之一是 int，mypy 也无法推断。

Answer 1

即使字典值之一是 int，mypy 也无法推断。

Mypy 是正确的。您的代码有错误，而 mypy 正确地标记了它。在您的代码中无法保证 data['start'] 和 data['end'] 始终是整数。

您的data 签名是Dict[str, Union[str, int]]，因此值的类型为Union[str, int]。 Mypy 必须假定传入{'start': '2018-07-12', 'end': -42}总是正确的，所以返回值必须是Tuple[Union[str, int], Union[str, int]]。您声称该函数返回 Tuple[int, int] 与此冲突。

在运行时实际上发生了什么并不重要。这不是重点。 mypy 是一个静态类型检查器，旨在帮助您保持运行时行为无错误。这里重要的是，根据类型提示，可能为start 和end 传递非整数值，因此类型检查器无法保护您免受未来的错误您的代码不小心为这两个键中的任何一个设置了字符串值。

如果您在字典中传递结构化数据，您将始终不得不为此与 mypy 抗争，因为字典确实是错误的结构。你真的想在这里使用命名元组或a dataclass。

我在这里使用名称FooBar，但对于您的特定应用程序，我相信您传递的数据结构会有更好的名称：

from typing import NamedTuple

class FooBar(NamedTuple):
    start: int
    end: int
    # other fields, perhaps with defaults and Optionals


def select_range(data: FooBar) -> Tuple[int, int]:
    if data.start and data.end:
        return data.start, data.end
    return 1, 1

Answer 2

代码一切都很好。你忘了给你函数参数：

from typing import Dict, Union, Tuple


def select_range(data: Dict[str, Union[str, int]]) -> Tuple[int, int]:
    print (data)
    if data['start'] and data['end']:
        return data['start'], data['end']
    return 1, 1

print (select_range({"start":[1,5], 'end':[2,6]}))

用它来查找一些参数：

from typing import Dict, Union, Tuple

def select_range(data: Dict[str, Union[str, int]]) -> Tuple[int, int]:
#    print (data)
    if 'start' in data and 'end' in data:
        return data['start'], data['end']
    return 1, 1

print (select_range({"start":[5], 'end':[6]}))
#select_range({})

([5], [6]) 是工作的结果