arraypad.pyi 688 B

123456789101112131415161718192021222324
  1. from collections.abc import Mapping
  2. from typing import Any, SupportsIndex
  3. import numpy as np
  4. import numpy.typing as npt
  5. from typing_extensions import assert_type
  6. def mode_func(
  7. ar: npt.NDArray[np.number[Any]],
  8. width: tuple[int, int],
  9. iaxis: SupportsIndex,
  10. kwargs: Mapping[str, Any],
  11. ) -> None: ...
  12. AR_i8: npt.NDArray[np.int64]
  13. AR_f8: npt.NDArray[np.float64]
  14. AR_LIKE: list[int]
  15. assert_type(np.pad(AR_i8, (2, 3), "constant"), npt.NDArray[np.int64])
  16. assert_type(np.pad(AR_LIKE, (2, 3), "constant"), npt.NDArray[Any])
  17. assert_type(np.pad(AR_f8, (2, 3), mode_func), npt.NDArray[np.float64])
  18. assert_type(np.pad(AR_f8, (2, 3), mode_func, a=1, b=2), npt.NDArray[np.float64])