numeric.py 1.5 KB

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