1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
| from typing import TypeVar, assert_type
|
| import numpy as np
| import numpy.typing as npt
| from numpy._typing import _32Bit, _64Bit
|
| T1 = TypeVar("T1", bound=npt.NBitBase) # type: ignore[deprecated] # pyright: ignore[reportDeprecated]
| T2 = TypeVar("T2", bound=npt.NBitBase) # type: ignore[deprecated] # pyright: ignore[reportDeprecated]
|
| def add(a: np.floating[T1], b: np.integer[T2]) -> np.floating[T1 | T2]:
| return a + b
|
| i8: np.int64
| i4: np.int32
| f8: np.float64
| f4: np.float32
|
| assert_type(add(f8, i8), np.floating[_64Bit])
| assert_type(add(f4, i8), np.floating[_32Bit | _64Bit])
| assert_type(add(f8, i4), np.floating[_32Bit | _64Bit])
| assert_type(add(f4, i4), np.floating[_32Bit])
|
|
