fromnumeric.py 3.7 KB

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