mirror of
https://github.com/astral-sh/ruff.git
synced 2025-10-21 07:41:53 +00:00
[ty] Add precise inference for unpacking a TypeVar if the TypeVar has an upper bound with a precise tuple spec (#19985)
This commit is contained in:
parent
c82e255ca8
commit
662d18bd05
3 changed files with 94 additions and 2 deletions
|
@ -455,3 +455,45 @@ def overloaded_outer(t: T | None = None) -> None:
|
|||
if t is not None:
|
||||
inner(t)
|
||||
```
|
||||
|
||||
## Unpacking a TypeVar
|
||||
|
||||
We can infer precise heterogeneous types from the result of an unpacking operation applied to a type
|
||||
variable if the type variable's upper bound is a type with a precise tuple spec:
|
||||
|
||||
```py
|
||||
from dataclasses import dataclass
|
||||
from typing import NamedTuple, Final, TypeVar, Generic
|
||||
|
||||
T = TypeVar("T", bound=tuple[int, str])
|
||||
|
||||
def f(x: T) -> T:
|
||||
a, b = x
|
||||
reveal_type(a) # revealed: int
|
||||
reveal_type(b) # revealed: str
|
||||
return x
|
||||
|
||||
@dataclass
|
||||
class Team(Generic[T]):
|
||||
employees: list[T]
|
||||
|
||||
def x(team: Team[T]) -> Team[T]:
|
||||
age, name = team.employees[0]
|
||||
reveal_type(age) # revealed: int
|
||||
reveal_type(name) # revealed: str
|
||||
return team
|
||||
|
||||
class Age(int): ...
|
||||
class Name(str): ...
|
||||
|
||||
class Employee(NamedTuple):
|
||||
age: Age
|
||||
name: Name
|
||||
|
||||
EMPLOYEES: Final = (Employee(name=Name("alice"), age=Age(42)),)
|
||||
team = Team(employees=list(EMPLOYEES))
|
||||
reveal_type(team.employees) # revealed: list[Employee]
|
||||
age, name = team.employees[0]
|
||||
reveal_type(age) # revealed: Age
|
||||
reveal_type(name) # revealed: Name
|
||||
```
|
||||
|
|
|
@ -454,3 +454,43 @@ def overloaded_outer[T](t: T | None = None) -> None:
|
|||
if t is not None:
|
||||
inner(t)
|
||||
```
|
||||
|
||||
## Unpacking a TypeVar
|
||||
|
||||
We can infer precise heterogeneous types from the result of an unpacking operation applied to a
|
||||
TypeVar if the TypeVar's upper bound is a type with a precise tuple spec:
|
||||
|
||||
```py
|
||||
from dataclasses import dataclass
|
||||
from typing import NamedTuple, Final
|
||||
|
||||
def f[T: tuple[int, str]](x: T) -> T:
|
||||
a, b = x
|
||||
reveal_type(a) # revealed: int
|
||||
reveal_type(b) # revealed: str
|
||||
return x
|
||||
|
||||
@dataclass
|
||||
class Team[T: tuple[int, str]]:
|
||||
employees: list[T]
|
||||
|
||||
def x[T: tuple[int, str]](team: Team[T]) -> Team[T]:
|
||||
age, name = team.employees[0]
|
||||
reveal_type(age) # revealed: int
|
||||
reveal_type(name) # revealed: str
|
||||
return team
|
||||
|
||||
class Age(int): ...
|
||||
class Name(str): ...
|
||||
|
||||
class Employee(NamedTuple):
|
||||
age: Age
|
||||
name: Name
|
||||
|
||||
EMPLOYEES: Final = (Employee(name=Name("alice"), age=Age(42)),)
|
||||
team = Team(employees=list(EMPLOYEES))
|
||||
reveal_type(team.employees) # revealed: list[Employee]
|
||||
age, name = team.employees[0]
|
||||
reveal_type(age) # revealed: Age
|
||||
reveal_type(name) # revealed: Name
|
||||
```
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue