fromnumeric.py 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264
  1. """Tests for :mod:`numpy.core.fromnumeric`."""
  2. import numpy as np
  3. A = np.array(True, ndmin=2, dtype=bool)
  4. B = np.array(1.0, ndmin=2, dtype=np.float32)
  5. A.setflags(write=False)
  6. B.setflags(write=False)
  7. a = np.bool_(True)
  8. b = np.float32(1.0)
  9. c = 1.0
  10. d = np.array(1.0, dtype=np.float32) # writeable
  11. reveal_type(np.take(a, 0)) # E: Any
  12. reveal_type(np.take(b, 0)) # E: Any
  13. reveal_type(np.take(c, 0)) # E: Any
  14. reveal_type(np.take(A, 0)) # E: Any
  15. reveal_type(np.take(B, 0)) # E: Any
  16. reveal_type(np.take(A, [0])) # E: Any
  17. reveal_type(np.take(B, [0])) # E: Any
  18. reveal_type(np.reshape(a, 1)) # E: numpy.ndarray
  19. reveal_type(np.reshape(b, 1)) # E: numpy.ndarray
  20. reveal_type(np.reshape(c, 1)) # E: numpy.ndarray
  21. reveal_type(np.reshape(A, 1)) # E: numpy.ndarray
  22. reveal_type(np.reshape(B, 1)) # E: numpy.ndarray
  23. reveal_type(np.choose(a, [True, True])) # E: Any
  24. reveal_type(np.choose(A, [True, True])) # E: Any
  25. reveal_type(np.repeat(a, 1)) # E: numpy.ndarray
  26. reveal_type(np.repeat(b, 1)) # E: numpy.ndarray
  27. reveal_type(np.repeat(c, 1)) # E: numpy.ndarray
  28. reveal_type(np.repeat(A, 1)) # E: numpy.ndarray
  29. reveal_type(np.repeat(B, 1)) # E: numpy.ndarray
  30. # TODO: Add tests for np.put()
  31. reveal_type(np.swapaxes(A, 0, 0)) # E: numpy.ndarray
  32. reveal_type(np.swapaxes(B, 0, 0)) # E: numpy.ndarray
  33. reveal_type(np.transpose(a)) # E: numpy.ndarray
  34. reveal_type(np.transpose(b)) # E: numpy.ndarray
  35. reveal_type(np.transpose(c)) # E: numpy.ndarray
  36. reveal_type(np.transpose(A)) # E: numpy.ndarray
  37. reveal_type(np.transpose(B)) # E: numpy.ndarray
  38. reveal_type(np.partition(a, 0, axis=None)) # E: numpy.ndarray
  39. reveal_type(np.partition(b, 0, axis=None)) # E: numpy.ndarray
  40. reveal_type(np.partition(c, 0, axis=None)) # E: numpy.ndarray
  41. reveal_type(np.partition(A, 0)) # E: numpy.ndarray
  42. reveal_type(np.partition(B, 0)) # E: numpy.ndarray
  43. reveal_type(np.argpartition(a, 0)) # E: Any
  44. reveal_type(np.argpartition(b, 0)) # E: Any
  45. reveal_type(np.argpartition(c, 0)) # E: Any
  46. reveal_type(np.argpartition(A, 0)) # E: Any
  47. reveal_type(np.argpartition(B, 0)) # E: Any
  48. reveal_type(np.sort(A, 0)) # E: numpy.ndarray
  49. reveal_type(np.sort(B, 0)) # E: numpy.ndarray
  50. reveal_type(np.argsort(A, 0)) # E: numpy.ndarray
  51. reveal_type(np.argsort(B, 0)) # E: numpy.ndarray
  52. reveal_type(np.argmax(A)) # E: numpy.signedinteger[Any]
  53. reveal_type(np.argmax(B)) # E: numpy.signedinteger[Any]
  54. reveal_type(np.argmax(A, axis=0)) # E: Any
  55. reveal_type(np.argmax(B, axis=0)) # E: Any
  56. reveal_type(np.argmin(A)) # E: numpy.signedinteger[Any]
  57. reveal_type(np.argmin(B)) # E: numpy.signedinteger[Any]
  58. reveal_type(np.argmin(A, axis=0)) # E: Any
  59. reveal_type(np.argmin(B, axis=0)) # E: Any
  60. reveal_type(np.searchsorted(A[0], 0)) # E: numpy.signedinteger[Any]
  61. reveal_type(np.searchsorted(B[0], 0)) # E: numpy.signedinteger[Any]
  62. reveal_type(np.searchsorted(A[0], [0])) # E: numpy.ndarray
  63. reveal_type(np.searchsorted(B[0], [0])) # E: numpy.ndarray
  64. reveal_type(np.resize(a, (5, 5))) # E: numpy.ndarray
  65. reveal_type(np.resize(b, (5, 5))) # E: numpy.ndarray
  66. reveal_type(np.resize(c, (5, 5))) # E: numpy.ndarray
  67. reveal_type(np.resize(A, (5, 5))) # E: numpy.ndarray
  68. reveal_type(np.resize(B, (5, 5))) # E: numpy.ndarray
  69. reveal_type(np.squeeze(a)) # E: numpy.bool_
  70. reveal_type(np.squeeze(b)) # E: numpy.floating[numpy.typing._32Bit]
  71. reveal_type(np.squeeze(c)) # E: numpy.ndarray
  72. reveal_type(np.squeeze(A)) # E: numpy.ndarray
  73. reveal_type(np.squeeze(B)) # E: numpy.ndarray
  74. reveal_type(np.diagonal(A)) # E: numpy.ndarray
  75. reveal_type(np.diagonal(B)) # E: numpy.ndarray
  76. reveal_type(np.trace(A)) # E: Any
  77. reveal_type(np.trace(B)) # E: Any
  78. reveal_type(np.ravel(a)) # E: numpy.ndarray
  79. reveal_type(np.ravel(b)) # E: numpy.ndarray
  80. reveal_type(np.ravel(c)) # E: numpy.ndarray
  81. reveal_type(np.ravel(A)) # E: numpy.ndarray
  82. reveal_type(np.ravel(B)) # E: numpy.ndarray
  83. reveal_type(np.nonzero(a)) # E: tuple[numpy.ndarray]
  84. reveal_type(np.nonzero(b)) # E: tuple[numpy.ndarray]
  85. reveal_type(np.nonzero(c)) # E: tuple[numpy.ndarray]
  86. reveal_type(np.nonzero(A)) # E: tuple[numpy.ndarray]
  87. reveal_type(np.nonzero(B)) # E: tuple[numpy.ndarray]
  88. reveal_type(np.shape(a)) # E: tuple[builtins.int]
  89. reveal_type(np.shape(b)) # E: tuple[builtins.int]
  90. reveal_type(np.shape(c)) # E: tuple[builtins.int]
  91. reveal_type(np.shape(A)) # E: tuple[builtins.int]
  92. reveal_type(np.shape(B)) # E: tuple[builtins.int]
  93. reveal_type(np.compress([True], a)) # E: numpy.ndarray
  94. reveal_type(np.compress([True], b)) # E: numpy.ndarray
  95. reveal_type(np.compress([True], c)) # E: numpy.ndarray
  96. reveal_type(np.compress([True], A)) # E: numpy.ndarray
  97. reveal_type(np.compress([True], B)) # E: numpy.ndarray
  98. reveal_type(np.clip(a, 0, 1.0)) # E: Any
  99. reveal_type(np.clip(b, -1, 1)) # E: Any
  100. reveal_type(np.clip(c, 0, 1)) # E: Any
  101. reveal_type(np.clip(A, 0, 1)) # E: Any
  102. reveal_type(np.clip(B, 0, 1)) # E: Any
  103. reveal_type(np.sum(a)) # E: Any
  104. reveal_type(np.sum(b)) # E: Any
  105. reveal_type(np.sum(c)) # E: Any
  106. reveal_type(np.sum(A)) # E: Any
  107. reveal_type(np.sum(B)) # E: Any
  108. reveal_type(np.sum(A, axis=0)) # E: Any
  109. reveal_type(np.sum(B, axis=0)) # E: Any
  110. reveal_type(np.all(a)) # E: numpy.bool_
  111. reveal_type(np.all(b)) # E: numpy.bool_
  112. reveal_type(np.all(c)) # E: numpy.bool_
  113. reveal_type(np.all(A)) # E: numpy.bool_
  114. reveal_type(np.all(B)) # E: numpy.bool_
  115. reveal_type(np.all(A, axis=0)) # E: Any
  116. reveal_type(np.all(B, axis=0)) # E: Any
  117. reveal_type(np.all(A, keepdims=True)) # E: Any
  118. reveal_type(np.all(B, keepdims=True)) # E: Any
  119. reveal_type(np.any(a)) # E: numpy.bool_
  120. reveal_type(np.any(b)) # E: numpy.bool_
  121. reveal_type(np.any(c)) # E: numpy.bool_
  122. reveal_type(np.any(A)) # E: numpy.bool_
  123. reveal_type(np.any(B)) # E: numpy.bool_
  124. reveal_type(np.any(A, axis=0)) # E: Any
  125. reveal_type(np.any(B, axis=0)) # E: Any
  126. reveal_type(np.any(A, keepdims=True)) # E: Any
  127. reveal_type(np.any(B, keepdims=True)) # E: Any
  128. reveal_type(np.cumsum(a)) # E: numpy.ndarray
  129. reveal_type(np.cumsum(b)) # E: numpy.ndarray
  130. reveal_type(np.cumsum(c)) # E: numpy.ndarray
  131. reveal_type(np.cumsum(A)) # E: numpy.ndarray
  132. reveal_type(np.cumsum(B)) # E: numpy.ndarray
  133. reveal_type(np.ptp(a)) # E: Any
  134. reveal_type(np.ptp(b)) # E: Any
  135. reveal_type(np.ptp(c)) # E: Any
  136. reveal_type(np.ptp(A)) # E: Any
  137. reveal_type(np.ptp(B)) # E: Any
  138. reveal_type(np.ptp(A, axis=0)) # E: Any
  139. reveal_type(np.ptp(B, axis=0)) # E: Any
  140. reveal_type(np.ptp(A, keepdims=True)) # E: Any
  141. reveal_type(np.ptp(B, keepdims=True)) # E: Any
  142. reveal_type(np.amax(a)) # E: Any
  143. reveal_type(np.amax(b)) # E: Any
  144. reveal_type(np.amax(c)) # E: Any
  145. reveal_type(np.amax(A)) # E: Any
  146. reveal_type(np.amax(B)) # E: Any
  147. reveal_type(np.amax(A, axis=0)) # E: Any
  148. reveal_type(np.amax(B, axis=0)) # E: Any
  149. reveal_type(np.amax(A, keepdims=True)) # E: Any
  150. reveal_type(np.amax(B, keepdims=True)) # E: Any
  151. reveal_type(np.amin(a)) # E: Any
  152. reveal_type(np.amin(b)) # E: Any
  153. reveal_type(np.amin(c)) # E: Any
  154. reveal_type(np.amin(A)) # E: Any
  155. reveal_type(np.amin(B)) # E: Any
  156. reveal_type(np.amin(A, axis=0)) # E: Any
  157. reveal_type(np.amin(B, axis=0)) # E: Any
  158. reveal_type(np.amin(A, keepdims=True)) # E: Any
  159. reveal_type(np.amin(B, keepdims=True)) # E: Any
  160. reveal_type(np.prod(a)) # E: Any
  161. reveal_type(np.prod(b)) # E: Any
  162. reveal_type(np.prod(c)) # E: Any
  163. reveal_type(np.prod(A)) # E: Any
  164. reveal_type(np.prod(B)) # E: Any
  165. reveal_type(np.prod(A, axis=0)) # E: Any
  166. reveal_type(np.prod(B, axis=0)) # E: Any
  167. reveal_type(np.prod(A, keepdims=True)) # E: Any
  168. reveal_type(np.prod(B, keepdims=True)) # E: Any
  169. reveal_type(np.prod(b, out=d)) # E: Any
  170. reveal_type(np.prod(B, out=d)) # E: Any
  171. reveal_type(np.cumprod(a)) # E: numpy.ndarray
  172. reveal_type(np.cumprod(b)) # E: numpy.ndarray
  173. reveal_type(np.cumprod(c)) # E: numpy.ndarray
  174. reveal_type(np.cumprod(A)) # E: numpy.ndarray
  175. reveal_type(np.cumprod(B)) # E: numpy.ndarray
  176. reveal_type(np.ndim(a)) # E: int
  177. reveal_type(np.ndim(b)) # E: int
  178. reveal_type(np.ndim(c)) # E: int
  179. reveal_type(np.ndim(A)) # E: int
  180. reveal_type(np.ndim(B)) # E: int
  181. reveal_type(np.size(a)) # E: int
  182. reveal_type(np.size(b)) # E: int
  183. reveal_type(np.size(c)) # E: int
  184. reveal_type(np.size(A)) # E: int
  185. reveal_type(np.size(B)) # E: int
  186. reveal_type(np.around(a)) # E: Any
  187. reveal_type(np.around(b)) # E: Any
  188. reveal_type(np.around(c)) # E: Any
  189. reveal_type(np.around(A)) # E: Any
  190. reveal_type(np.around(B)) # E: Any
  191. reveal_type(np.mean(a)) # E: Any
  192. reveal_type(np.mean(b)) # E: Any
  193. reveal_type(np.mean(c)) # E: Any
  194. reveal_type(np.mean(A)) # E: Any
  195. reveal_type(np.mean(B)) # E: Any
  196. reveal_type(np.mean(A, axis=0)) # E: Any
  197. reveal_type(np.mean(B, axis=0)) # E: Any
  198. reveal_type(np.mean(A, keepdims=True)) # E: Any
  199. reveal_type(np.mean(B, keepdims=True)) # E: Any
  200. reveal_type(np.mean(b, out=d)) # E: Any
  201. reveal_type(np.mean(B, out=d)) # E: Any
  202. reveal_type(np.std(a)) # E: Any
  203. reveal_type(np.std(b)) # E: Any
  204. reveal_type(np.std(c)) # E: Any
  205. reveal_type(np.std(A)) # E: Any
  206. reveal_type(np.std(B)) # E: Any
  207. reveal_type(np.std(A, axis=0)) # E: Any
  208. reveal_type(np.std(B, axis=0)) # E: Any
  209. reveal_type(np.std(A, keepdims=True)) # E: Any
  210. reveal_type(np.std(B, keepdims=True)) # E: Any
  211. reveal_type(np.std(b, out=d)) # E: Any
  212. reveal_type(np.std(B, out=d)) # E: Any
  213. reveal_type(np.var(a)) # E: Any
  214. reveal_type(np.var(b)) # E: Any
  215. reveal_type(np.var(c)) # E: Any
  216. reveal_type(np.var(A)) # E: Any
  217. reveal_type(np.var(B)) # E: Any
  218. reveal_type(np.var(A, axis=0)) # E: Any
  219. reveal_type(np.var(B, axis=0)) # E: Any
  220. reveal_type(np.var(A, keepdims=True)) # E: Any
  221. reveal_type(np.var(B, keepdims=True)) # E: Any
  222. reveal_type(np.var(b, out=d)) # E: Any
  223. reveal_type(np.var(B, out=d)) # E: Any