numeric.py 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. """
  2. Tests for :mod:`numpy.core.numeric`.
  3. Does not include tests which fall under ``array_constructors``.
  4. """
  5. from typing import List
  6. import numpy as np
  7. class SubClass(np.ndarray):
  8. ...
  9. i8: np.int64
  10. A: np.ndarray
  11. B: List[int]
  12. C: SubClass
  13. reveal_type(np.count_nonzero(i8)) # E: int
  14. reveal_type(np.count_nonzero(A)) # E: int
  15. reveal_type(np.count_nonzero(B)) # E: int
  16. reveal_type(np.count_nonzero(A, keepdims=True)) # E: Any
  17. reveal_type(np.count_nonzero(A, axis=0)) # E: Any
  18. reveal_type(np.isfortran(i8)) # E: bool
  19. reveal_type(np.isfortran(A)) # E: bool
  20. reveal_type(np.argwhere(i8)) # E: numpy.ndarray
  21. reveal_type(np.argwhere(A)) # E: numpy.ndarray
  22. reveal_type(np.flatnonzero(i8)) # E: numpy.ndarray
  23. reveal_type(np.flatnonzero(A)) # E: numpy.ndarray
  24. reveal_type(np.correlate(B, A, mode="valid")) # E: numpy.ndarray
  25. reveal_type(np.correlate(A, A, mode="same")) # E: numpy.ndarray
  26. reveal_type(np.convolve(B, A, mode="valid")) # E: numpy.ndarray
  27. reveal_type(np.convolve(A, A, mode="same")) # E: numpy.ndarray
  28. reveal_type(np.outer(i8, A)) # E: numpy.ndarray
  29. reveal_type(np.outer(B, A)) # E: numpy.ndarray
  30. reveal_type(np.outer(A, A)) # E: numpy.ndarray
  31. reveal_type(np.outer(A, A, out=C)) # E: SubClass
  32. reveal_type(np.tensordot(B, A)) # E: numpy.ndarray
  33. reveal_type(np.tensordot(A, A)) # E: numpy.ndarray
  34. reveal_type(np.tensordot(A, A, axes=0)) # E: numpy.ndarray
  35. reveal_type(np.tensordot(A, A, axes=(0, 1))) # E: numpy.ndarray
  36. reveal_type(np.isscalar(i8)) # E: bool
  37. reveal_type(np.isscalar(A)) # E: bool
  38. reveal_type(np.isscalar(B)) # E: bool
  39. reveal_type(np.roll(A, 1)) # E: numpy.ndarray
  40. reveal_type(np.roll(A, (1, 2))) # E: numpy.ndarray
  41. reveal_type(np.roll(B, 1)) # E: numpy.ndarray
  42. reveal_type(np.rollaxis(A, 0, 1)) # E: numpy.ndarray
  43. reveal_type(np.moveaxis(A, 0, 1)) # E: numpy.ndarray
  44. reveal_type(np.moveaxis(A, (0, 1), (1, 2))) # E: numpy.ndarray
  45. reveal_type(np.cross(B, A)) # E: numpy.ndarray
  46. reveal_type(np.cross(A, A)) # E: numpy.ndarray
  47. reveal_type(np.indices([0, 1, 2])) # E: numpy.ndarray
  48. reveal_type(np.indices([0, 1, 2], sparse=False)) # E: numpy.ndarray
  49. reveal_type(np.indices([0, 1, 2], sparse=True)) # E: tuple[numpy.ndarray]
  50. reveal_type(np.binary_repr(1)) # E: str
  51. reveal_type(np.base_repr(1)) # E: str
  52. reveal_type(np.allclose(i8, A)) # E: bool
  53. reveal_type(np.allclose(B, A)) # E: bool
  54. reveal_type(np.allclose(A, A)) # E: bool
  55. reveal_type(np.isclose(i8, A)) # E: Any
  56. reveal_type(np.isclose(B, A)) # E: Any
  57. reveal_type(np.isclose(A, A)) # E: Any
  58. reveal_type(np.array_equal(i8, A)) # E: bool
  59. reveal_type(np.array_equal(B, A)) # E: bool
  60. reveal_type(np.array_equal(A, A)) # E: bool
  61. reveal_type(np.array_equiv(i8, A)) # E: bool
  62. reveal_type(np.array_equiv(B, A)) # E: bool
  63. reveal_type(np.array_equiv(A, A)) # E: bool