mypy 数值类型的返回值类型不兼容

Question

考虑这个例子

from typing import TypeVar

Even = TypeVar("Even",bound=int)
Odd  = TypeVar("Odd",bound=int)


def make_even(n:int) -> Even:
    return n*2
    
def make_odd(n:int) -> Odd:
    return n*2+1

def make_both(n:int) -> tuple[Even,Odd]:
    return make_even(n), make_odd(n)

help(make_both)

它的输出很好，就像我想要的一样

Help on function make_both in module __main__:

make_both(n: int) -> tuple[~Even, ~Odd]

但 mypy 不喜欢它。

test5.py:8: error: Incompatible return value type (got "int", expected "Even")
test5.py:11: error: Incompatible return value type (got "int", expected "Odd")
Found 2 errors in 1 file (checked 1 source file)

所以我的问题是如何取悦 mypy 并仍然保持帮助的良好输出？ 除了#type: ignore

Answer 1

要进行此传递类型检查，请改用NewType ：

from typing import NewType

Even = NewType('Even', int)
Odd  = NewType('Odd', int)


def make_even(n: int) -> Even:
    return Even(n*2)
    
def make_odd(n : int) -> Odd:
    return Odd(n*2 + 1)

def make_both(n: int) -> tuple[Even, Odd]:
    return make_even(n), make_odd(n)

但是现在help打印完全限定的名称（ __main__.Even在解释器中）。 要解决此问题，您可以在未来使用字符串文字或annotations ：

...

def make_both(n: int) -> 'tuple[Even, Odd]':
    return make_even(n), make_odd(n)

或者

from __future__ import annotations
...

def make_both(n: int) -> tuple[Even, Odd]:
    return make_even(n), make_odd(n)

在第一种情况下， help输出是

Help on function make_both in module __main__:

make_both(n: int) -> 'tuple[Even, Odd]'

在第二

Help on function make_both in module __main__:

make_both(n: 'int') -> 'tuple[Even, Odd]'