ufuncs.pyi 2.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. import numpy as np
  2. import numpy.typing as npt
  3. f8: np.float64
  4. AR_f8: npt.NDArray[np.float64]
  5. AR_i8: npt.NDArray[np.int64]
  6. reveal_type(np.absolute.__doc__) # E: str
  7. reveal_type(np.absolute.types) # E: builtins.list[builtins.str]
  8. reveal_type(np.absolute.__name__) # E: Literal['absolute']
  9. reveal_type(np.absolute.ntypes) # E: Literal[20]
  10. reveal_type(np.absolute.identity) # E: None
  11. reveal_type(np.absolute.nin) # E: Literal[1]
  12. reveal_type(np.absolute.nin) # E: Literal[1]
  13. reveal_type(np.absolute.nout) # E: Literal[1]
  14. reveal_type(np.absolute.nargs) # E: Literal[2]
  15. reveal_type(np.absolute.signature) # E: None
  16. reveal_type(np.absolute(f8)) # E: Any
  17. reveal_type(np.absolute(AR_f8)) # E: ndarray
  18. reveal_type(np.absolute.at(AR_f8, AR_i8)) # E: None
  19. reveal_type(np.add.__name__) # E: Literal['add']
  20. reveal_type(np.add.ntypes) # E: Literal[22]
  21. reveal_type(np.add.identity) # E: Literal[0]
  22. reveal_type(np.add.nin) # E: Literal[2]
  23. reveal_type(np.add.nout) # E: Literal[1]
  24. reveal_type(np.add.nargs) # E: Literal[3]
  25. reveal_type(np.add.signature) # E: None
  26. reveal_type(np.add(f8, f8)) # E: Any
  27. reveal_type(np.add(AR_f8, f8)) # E: ndarray
  28. reveal_type(np.add.at(AR_f8, AR_i8, f8)) # E: None
  29. reveal_type(np.add.reduce(AR_f8, axis=0)) # E: Any
  30. reveal_type(np.add.accumulate(AR_f8)) # E: ndarray
  31. reveal_type(np.add.reduceat(AR_f8, AR_i8)) # E: ndarray
  32. reveal_type(np.add.outer(f8, f8)) # E: Any
  33. reveal_type(np.add.outer(AR_f8, f8)) # E: ndarray
  34. reveal_type(np.frexp.__name__) # E: Literal['frexp']
  35. reveal_type(np.frexp.ntypes) # E: Literal[4]
  36. reveal_type(np.frexp.identity) # E: None
  37. reveal_type(np.frexp.nin) # E: Literal[1]
  38. reveal_type(np.frexp.nout) # E: Literal[2]
  39. reveal_type(np.frexp.nargs) # E: Literal[3]
  40. reveal_type(np.frexp.signature) # E: None
  41. reveal_type(np.frexp(f8)) # E: Tuple[Any, Any]
  42. reveal_type(np.frexp(AR_f8)) # E: Tuple[ndarray[Any, dtype[Any]], ndarray[Any, dtype[Any]]]
  43. reveal_type(np.divmod.__name__) # E: Literal['divmod']
  44. reveal_type(np.divmod.ntypes) # E: Literal[15]
  45. reveal_type(np.divmod.identity) # E: None
  46. reveal_type(np.divmod.nin) # E: Literal[2]
  47. reveal_type(np.divmod.nout) # E: Literal[2]
  48. reveal_type(np.divmod.nargs) # E: Literal[4]
  49. reveal_type(np.divmod.signature) # E: None
  50. reveal_type(np.divmod(f8, f8)) # E: Tuple[Any, Any]
  51. reveal_type(np.divmod(AR_f8, f8)) # E: Tuple[ndarray[Any, dtype[Any]], ndarray[Any, dtype[Any]]]
  52. reveal_type(np.matmul.__name__) # E: Literal['matmul']
  53. reveal_type(np.matmul.ntypes) # E: Literal[19]
  54. reveal_type(np.matmul.identity) # E: None
  55. reveal_type(np.matmul.nin) # E: Literal[2]
  56. reveal_type(np.matmul.nout) # E: Literal[1]
  57. reveal_type(np.matmul.nargs) # E: Literal[3]
  58. reveal_type(np.matmul.signature) # E: Literal['(n?,k),(k,m?)->(n?,m?)']
  59. reveal_type(np.matmul.identity) # E: None
  60. reveal_type(np.matmul(AR_f8, AR_f8)) # E: Any
  61. reveal_type(np.matmul(AR_f8, AR_f8, axes=[(0, 1), (0, 1), (0, 1)])) # E: Any