fromnumeric.pyi 5.5 KB

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