Python тип намекающих союзов

Question

У меня есть функция с двумя аргументами:

def same_type_params(param1: Union[str, int], param2: Union[str, int]):
    pass

Как ограничить, чтобы типы param1 и param2 были равны? то есть либо оба str, либо оба int

Answer 1

Используйте переменную типа :

from typing import TypeVar
T = TypeVar('T', str, int)
def same_type_params(param1: T, param2: T) -> None:
    pass