ufuncs.pyi 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. from typing import Literal, Any, NoReturn
  2. import numpy as np
  3. import numpy.typing as npt
  4. from typing_extensions import assert_type
  5. i8: np.int64
  6. f8: np.float64
  7. AR_f8: npt.NDArray[np.float64]
  8. AR_i8: npt.NDArray[np.int64]
  9. assert_type(np.absolute.__doc__, str)
  10. assert_type(np.absolute.types, list[str])
  11. assert_type(np.absolute.__name__, Literal["absolute"])
  12. assert_type(np.absolute.__qualname__, Literal["absolute"])
  13. assert_type(np.absolute.ntypes, Literal[20])
  14. assert_type(np.absolute.identity, None)
  15. assert_type(np.absolute.nin, Literal[1])
  16. assert_type(np.absolute.nin, Literal[1])
  17. assert_type(np.absolute.nout, Literal[1])
  18. assert_type(np.absolute.nargs, Literal[2])
  19. assert_type(np.absolute.signature, None)
  20. assert_type(np.absolute(f8), Any)
  21. assert_type(np.absolute(AR_f8), npt.NDArray[Any])
  22. assert_type(np.absolute.at(AR_f8, AR_i8), None)
  23. assert_type(np.add.__name__, Literal["add"])
  24. assert_type(np.add.__qualname__, Literal["add"])
  25. assert_type(np.add.ntypes, Literal[22])
  26. assert_type(np.add.identity, Literal[0])
  27. assert_type(np.add.nin, Literal[2])
  28. assert_type(np.add.nout, Literal[1])
  29. assert_type(np.add.nargs, Literal[3])
  30. assert_type(np.add.signature, None)
  31. assert_type(np.add(f8, f8), Any)
  32. assert_type(np.add(AR_f8, f8), npt.NDArray[Any])
  33. assert_type(np.add.at(AR_f8, AR_i8, f8), None)
  34. assert_type(np.add.reduce(AR_f8, axis=0), Any)
  35. assert_type(np.add.accumulate(AR_f8), npt.NDArray[Any])
  36. assert_type(np.add.reduceat(AR_f8, AR_i8), npt.NDArray[Any])
  37. assert_type(np.add.outer(f8, f8), Any)
  38. assert_type(np.add.outer(AR_f8, f8), npt.NDArray[Any])
  39. assert_type(np.frexp.__name__, Literal["frexp"])
  40. assert_type(np.frexp.__qualname__, Literal["frexp"])
  41. assert_type(np.frexp.ntypes, Literal[4])
  42. assert_type(np.frexp.identity, None)
  43. assert_type(np.frexp.nin, Literal[1])
  44. assert_type(np.frexp.nout, Literal[2])
  45. assert_type(np.frexp.nargs, Literal[3])
  46. assert_type(np.frexp.signature, None)
  47. assert_type(np.frexp(f8), tuple[Any, Any])
  48. assert_type(np.frexp(AR_f8), tuple[npt.NDArray[Any], npt.NDArray[Any]])
  49. assert_type(np.divmod.__name__, Literal["divmod"])
  50. assert_type(np.divmod.__qualname__, Literal["divmod"])
  51. assert_type(np.divmod.ntypes, Literal[15])
  52. assert_type(np.divmod.identity, None)
  53. assert_type(np.divmod.nin, Literal[2])
  54. assert_type(np.divmod.nout, Literal[2])
  55. assert_type(np.divmod.nargs, Literal[4])
  56. assert_type(np.divmod.signature, None)
  57. assert_type(np.divmod(f8, f8), tuple[Any, Any])
  58. assert_type(np.divmod(AR_f8, f8), tuple[npt.NDArray[Any], npt.NDArray[Any]])
  59. assert_type(np.matmul.__name__, Literal["matmul"])
  60. assert_type(np.matmul.__qualname__, Literal["matmul"])
  61. assert_type(np.matmul.ntypes, Literal[19])
  62. assert_type(np.matmul.identity, None)
  63. assert_type(np.matmul.nin, Literal[2])
  64. assert_type(np.matmul.nout, Literal[1])
  65. assert_type(np.matmul.nargs, Literal[3])
  66. assert_type(np.matmul.signature, Literal["(n?,k),(k,m?)->(n?,m?)"])
  67. assert_type(np.matmul.identity, None)
  68. assert_type(np.matmul(AR_f8, AR_f8), Any)
  69. assert_type(np.matmul(AR_f8, AR_f8, axes=[(0, 1), (0, 1), (0, 1)]), Any)
  70. assert_type(np.vecdot.__name__, Literal["vecdot"])
  71. assert_type(np.vecdot.__qualname__, Literal["vecdot"])
  72. assert_type(np.vecdot.ntypes, Literal[19])
  73. assert_type(np.vecdot.identity, None)
  74. assert_type(np.vecdot.nin, Literal[2])
  75. assert_type(np.vecdot.nout, Literal[1])
  76. assert_type(np.vecdot.nargs, Literal[3])
  77. assert_type(np.vecdot.signature, Literal["(n),(n)->()"])
  78. assert_type(np.vecdot.identity, None)
  79. assert_type(np.vecdot(AR_f8, AR_f8), Any)
  80. assert_type(np.bitwise_count.__name__, Literal["bitwise_count"])
  81. assert_type(np.bitwise_count.__qualname__, Literal["bitwise_count"])
  82. assert_type(np.bitwise_count.ntypes, Literal[11])
  83. assert_type(np.bitwise_count.identity, None)
  84. assert_type(np.bitwise_count.nin, Literal[1])
  85. assert_type(np.bitwise_count.nout, Literal[1])
  86. assert_type(np.bitwise_count.nargs, Literal[2])
  87. assert_type(np.bitwise_count.signature, None)
  88. assert_type(np.bitwise_count.identity, None)
  89. assert_type(np.bitwise_count(i8), Any)
  90. assert_type(np.bitwise_count(AR_i8), npt.NDArray[Any])
  91. assert_type(np.absolute.outer(), NoReturn)
  92. assert_type(np.frexp.outer(), NoReturn)
  93. assert_type(np.divmod.outer(), NoReturn)
  94. assert_type(np.matmul.outer(), NoReturn)
  95. assert_type(np.absolute.reduceat(), NoReturn)
  96. assert_type(np.frexp.reduceat(), NoReturn)
  97. assert_type(np.divmod.reduceat(), NoReturn)
  98. assert_type(np.matmul.reduceat(), NoReturn)
  99. assert_type(np.absolute.reduce(), NoReturn)
  100. assert_type(np.frexp.reduce(), NoReturn)
  101. assert_type(np.divmod.reduce(), NoReturn)
  102. assert_type(np.matmul.reduce(), NoReturn)
  103. assert_type(np.absolute.accumulate(), NoReturn)
  104. assert_type(np.frexp.accumulate(), NoReturn)
  105. assert_type(np.divmod.accumulate(), NoReturn)
  106. assert_type(np.matmul.accumulate(), NoReturn)
  107. assert_type(np.frexp.at(), NoReturn)
  108. assert_type(np.divmod.at(), NoReturn)
  109. assert_type(np.matmul.at(), NoReturn)