Mypy：dict 值的类型不会随着赋值而改变，而变量的类型会改变

Question

在下面的例子中，为什么当变量foo的值设置为整数时，它的类型会改变，而bar["foo"]的类型不会发生同样的情况？

import typing as tp

foo: tp.Union[float, int]
D = tp.TypedDict("D", {"foo": tp.Union[float, int]})

foo = 123.45
bar = D(foo=123.45)

reveal_type(foo)  # main.py:9: note: Revealed type is 'builtins.float'
reveal_type(bar["foo"])  # main.py:10: note: Revealed type is 'builtins.float'

foo = int(foo)
bar["foo"] = int(bar["foo"])

reveal_type(foo)  # main.py:15: note: Revealed type is 'builtins.int'
reveal_type(bar["foo"])  # main.py:16: note: Revealed type is 'builtins.float'

Answer 1

这是由于一些mypy功能：

• 为Union键入缩小。

初始声明之后的任何赋值都会缩小类型

x: Union[int, float]
x = 1.1
reveal_type(x)  # Revealed type is 'builtins.float'

但是mypy不会缩小赋值中的类型：

x: Union[int, float] = 1.1
reveal_type(x)  # Revealed type is 'Union[builtins.int, builtins.float]'

• 所谓的promotion of types由mypy 。

例如， int不是运行时float的子类型，但mypy认为它是。 在需要float任何地方，都可以传递int （有时可能不明显），尽管它不是它的子类型。

• 简化类属性的Union

class A:
    x: Union[bool, int, float]  

reveal_type(A.x)  # Revealed type is 'builtins.float'

这个结果是因为bool是int的子类型，而int被mypy提升为float 。

在您的TypedDict示例中，有一个Union简化，如下所示

from typing import Union, TypedDict

D = TypedDict("D", {"x": Union[int, float]}) 
d: D
y: Union[int, float]

reveal_type(d["x"])  # Revealed type is 'builtins.float'
reveal_type(y)  # Revealed type is 'Union[builtins.int, builtins.float]'