ufunclike.pyi 1.2 KB

123456789101112131415161718192021222324252627282930313233
  1. from typing import Any
  2. import numpy as np
  3. import numpy.typing as npt
  4. from typing_extensions import assert_type
  5. AR_LIKE_b: list[bool]
  6. AR_LIKE_u: list[np.uint32]
  7. AR_LIKE_i: list[int]
  8. AR_LIKE_f: list[float]
  9. AR_LIKE_O: list[np.object_]
  10. AR_U: npt.NDArray[np.str_]
  11. assert_type(np.fix(AR_LIKE_b), npt.NDArray[np.floating[Any]])
  12. assert_type(np.fix(AR_LIKE_u), npt.NDArray[np.floating[Any]])
  13. assert_type(np.fix(AR_LIKE_i), npt.NDArray[np.floating[Any]])
  14. assert_type(np.fix(AR_LIKE_f), npt.NDArray[np.floating[Any]])
  15. assert_type(np.fix(AR_LIKE_O), npt.NDArray[np.object_])
  16. assert_type(np.fix(AR_LIKE_f, out=AR_U), npt.NDArray[np.str_])
  17. assert_type(np.isposinf(AR_LIKE_b), npt.NDArray[np.bool])
  18. assert_type(np.isposinf(AR_LIKE_u), npt.NDArray[np.bool])
  19. assert_type(np.isposinf(AR_LIKE_i), npt.NDArray[np.bool])
  20. assert_type(np.isposinf(AR_LIKE_f), npt.NDArray[np.bool])
  21. assert_type(np.isposinf(AR_LIKE_f, out=AR_U), npt.NDArray[np.str_])
  22. assert_type(np.isneginf(AR_LIKE_b), npt.NDArray[np.bool])
  23. assert_type(np.isneginf(AR_LIKE_u), npt.NDArray[np.bool])
  24. assert_type(np.isneginf(AR_LIKE_i), npt.NDArray[np.bool])
  25. assert_type(np.isneginf(AR_LIKE_f), npt.NDArray[np.bool])
  26. assert_type(np.isneginf(AR_LIKE_f, out=AR_U), npt.NDArray[np.str_])