fromnumeric.pyi 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. """Tests for :mod:`numpy._core.fromnumeric`."""
  2. import numpy as np
  3. import numpy.typing as npt
  4. A = np.array(True, ndmin=2, dtype=bool)
  5. A.setflags(write=False)
  6. AR_U: npt.NDArray[np.str_]
  7. AR_M: npt.NDArray[np.datetime64]
  8. a = np.bool(True)
  9. np.take(a, None) # E: No overload variant
  10. np.take(a, axis=1.0) # E: No overload variant
  11. np.take(A, out=1) # E: No overload variant
  12. np.take(A, mode="bob") # E: No overload variant
  13. np.reshape(a, None) # E: No overload variant
  14. np.reshape(A, 1, order="bob") # E: No overload variant
  15. np.choose(a, None) # E: No overload variant
  16. np.choose(a, out=1.0) # E: No overload variant
  17. np.choose(A, mode="bob") # E: No overload variant
  18. np.repeat(a, None) # E: No overload variant
  19. np.repeat(A, 1, axis=1.0) # E: No overload variant
  20. np.swapaxes(A, None, 1) # E: No overload variant
  21. np.swapaxes(A, 1, [0]) # E: No overload variant
  22. np.transpose(A, axes=1.0) # E: No overload variant
  23. np.partition(a, None) # E: No overload variant
  24. np.partition( # E: No overload variant
  25. a, 0, axis="bob"
  26. )
  27. np.partition( # E: No overload variant
  28. A, 0, kind="bob"
  29. )
  30. np.partition(
  31. A, 0, order=range(5) # E: Argument "order" to "partition" has incompatible type
  32. )
  33. np.argpartition(
  34. a, None # E: incompatible type
  35. )
  36. np.argpartition(
  37. a, 0, axis="bob" # E: incompatible type
  38. )
  39. np.argpartition(
  40. A, 0, kind="bob" # E: incompatible type
  41. )
  42. np.argpartition(
  43. A, 0, order=range(5) # E: Argument "order" to "argpartition" has incompatible type
  44. )
  45. np.sort(A, axis="bob") # E: No overload variant
  46. np.sort(A, kind="bob") # E: No overload variant
  47. np.sort(A, order=range(5)) # E: Argument "order" to "sort" has incompatible type
  48. np.argsort(A, axis="bob") # E: Argument "axis" to "argsort" has incompatible type
  49. np.argsort(A, kind="bob") # E: Argument "kind" to "argsort" has incompatible type
  50. np.argsort(A, order=range(5)) # E: Argument "order" to "argsort" has incompatible type
  51. np.argmax(A, axis="bob") # E: No overload variant of "argmax" matches argument type
  52. np.argmax(A, kind="bob") # E: No overload variant of "argmax" matches argument type
  53. np.argmin(A, axis="bob") # E: No overload variant of "argmin" matches argument type
  54. np.argmin(A, kind="bob") # E: No overload variant of "argmin" matches argument type
  55. np.searchsorted( # E: No overload variant of "searchsorted" matches argument type
  56. A[0], 0, side="bob"
  57. )
  58. np.searchsorted( # E: No overload variant of "searchsorted" matches argument type
  59. A[0], 0, sorter=1.0
  60. )
  61. np.resize(A, 1.0) # E: No overload variant
  62. np.squeeze(A, 1.0) # E: No overload variant of "squeeze" matches argument type
  63. np.diagonal(A, offset=None) # E: No overload variant
  64. np.diagonal(A, axis1="bob") # E: No overload variant
  65. np.diagonal(A, axis2=[]) # E: No overload variant
  66. np.trace(A, offset=None) # E: No overload variant
  67. np.trace(A, axis1="bob") # E: No overload variant
  68. np.trace(A, axis2=[]) # E: No overload variant
  69. np.ravel(a, order="bob") # E: No overload variant
  70. np.nonzero(0) # E: No overload variant
  71. np.compress( # E: No overload variant
  72. [True], A, axis=1.0
  73. )
  74. np.clip(a, 1, 2, out=1) # E: No overload variant of "clip" matches argument type
  75. np.sum(a, axis=1.0) # E: No overload variant
  76. np.sum(a, keepdims=1.0) # E: No overload variant
  77. np.sum(a, initial=[1]) # E: No overload variant
  78. np.all(a, axis=1.0) # E: No overload variant
  79. np.all(a, keepdims=1.0) # E: No overload variant
  80. np.all(a, out=1.0) # E: No overload variant
  81. np.any(a, axis=1.0) # E: No overload variant
  82. np.any(a, keepdims=1.0) # E: No overload variant
  83. np.any(a, out=1.0) # E: No overload variant
  84. np.cumsum(a, axis=1.0) # E: No overload variant
  85. np.cumsum(a, dtype=1.0) # E: No overload variant
  86. np.cumsum(a, out=1.0) # E: No overload variant
  87. np.ptp(a, axis=1.0) # E: No overload variant
  88. np.ptp(a, keepdims=1.0) # E: No overload variant
  89. np.ptp(a, out=1.0) # E: No overload variant
  90. np.amax(a, axis=1.0) # E: No overload variant
  91. np.amax(a, keepdims=1.0) # E: No overload variant
  92. np.amax(a, out=1.0) # E: No overload variant
  93. np.amax(a, initial=[1.0]) # E: No overload variant
  94. np.amax(a, where=[1.0]) # E: incompatible type
  95. np.amin(a, axis=1.0) # E: No overload variant
  96. np.amin(a, keepdims=1.0) # E: No overload variant
  97. np.amin(a, out=1.0) # E: No overload variant
  98. np.amin(a, initial=[1.0]) # E: No overload variant
  99. np.amin(a, where=[1.0]) # E: incompatible type
  100. np.prod(a, axis=1.0) # E: No overload variant
  101. np.prod(a, out=False) # E: No overload variant
  102. np.prod(a, keepdims=1.0) # E: No overload variant
  103. np.prod(a, initial=int) # E: No overload variant
  104. np.prod(a, where=1.0) # E: No overload variant
  105. np.prod(AR_U) # E: incompatible type
  106. np.cumprod(a, axis=1.0) # E: No overload variant
  107. np.cumprod(a, out=False) # E: No overload variant
  108. np.cumprod(AR_U) # E: incompatible type
  109. np.size(a, axis=1.0) # E: Argument "axis" to "size" has incompatible type
  110. np.around(a, decimals=1.0) # E: No overload variant
  111. np.around(a, out=type) # E: No overload variant
  112. np.around(AR_U) # E: incompatible type
  113. np.mean(a, axis=1.0) # E: No overload variant
  114. np.mean(a, out=False) # E: No overload variant
  115. np.mean(a, keepdims=1.0) # E: No overload variant
  116. np.mean(AR_U) # E: incompatible type
  117. np.mean(AR_M) # E: incompatible type
  118. np.std(a, axis=1.0) # E: No overload variant
  119. np.std(a, out=False) # E: No overload variant
  120. np.std(a, ddof='test') # E: No overload variant
  121. np.std(a, keepdims=1.0) # E: No overload variant
  122. np.std(AR_U) # E: incompatible type
  123. np.var(a, axis=1.0) # E: No overload variant
  124. np.var(a, out=False) # E: No overload variant
  125. np.var(a, ddof='test') # E: No overload variant
  126. np.var(a, keepdims=1.0) # E: No overload variant
  127. np.var(AR_U) # E: incompatible type