23 lines
694 B
Python
23 lines
694 B
Python
from collections.abc import Mapping
|
|
from typing import Any, SupportsIndex
|
|
|
|
import numpy as np
|
|
import numpy.typing as npt
|
|
|
|
def mode_func(
|
|
ar: npt.NDArray[np.number[Any]],
|
|
width: tuple[int, int],
|
|
iaxis: SupportsIndex,
|
|
kwargs: Mapping[str, Any],
|
|
) -> None: ...
|
|
|
|
AR_i8: npt.NDArray[np.int64]
|
|
AR_f8: npt.NDArray[np.float64]
|
|
AR_LIKE: list[int]
|
|
|
|
reveal_type(np.pad(AR_i8, (2, 3), "constant")) # E: ndarray[Any, dtype[{int64}]]
|
|
reveal_type(np.pad(AR_LIKE, (2, 3), "constant")) # E: ndarray[Any, dtype[Any]]
|
|
|
|
reveal_type(np.pad(AR_f8, (2, 3), mode_func)) # E: ndarray[Any, dtype[{float64}]]
|
|
reveal_type(np.pad(AR_f8, (2, 3), mode_func, a=1, b=2)) # E: ndarray[Any, dtype[{float64}]]
|