matrix.pyi 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. from typing import Any
  2. import numpy as np
  3. import numpy.typing as npt
  4. mat: np.matrix[Any, np.dtype[np.int64]]
  5. ar_f8: npt.NDArray[np.float64]
  6. reveal_type(mat * 5) # E: matrix[Any, Any]
  7. reveal_type(5 * mat) # E: matrix[Any, Any]
  8. mat *= 5
  9. reveal_type(mat**5) # E: matrix[Any, Any]
  10. mat **= 5
  11. reveal_type(mat.sum()) # E: Any
  12. reveal_type(mat.mean()) # E: Any
  13. reveal_type(mat.std()) # E: Any
  14. reveal_type(mat.var()) # E: Any
  15. reveal_type(mat.prod()) # E: Any
  16. reveal_type(mat.any()) # E: bool_
  17. reveal_type(mat.all()) # E: bool_
  18. reveal_type(mat.max()) # E: {int64}
  19. reveal_type(mat.min()) # E: {int64}
  20. reveal_type(mat.argmax()) # E: {intp}
  21. reveal_type(mat.argmin()) # E: {intp}
  22. reveal_type(mat.ptp()) # E: {int64}
  23. reveal_type(mat.sum(axis=0)) # E: matrix[Any, Any]
  24. reveal_type(mat.mean(axis=0)) # E: matrix[Any, Any]
  25. reveal_type(mat.std(axis=0)) # E: matrix[Any, Any]
  26. reveal_type(mat.var(axis=0)) # E: matrix[Any, Any]
  27. reveal_type(mat.prod(axis=0)) # E: matrix[Any, Any]
  28. reveal_type(mat.any(axis=0)) # E: matrix[Any, dtype[bool_]]
  29. reveal_type(mat.all(axis=0)) # E: matrix[Any, dtype[bool_]]
  30. reveal_type(mat.max(axis=0)) # E: matrix[Any, dtype[{int64}]]
  31. reveal_type(mat.min(axis=0)) # E: matrix[Any, dtype[{int64}]]
  32. reveal_type(mat.argmax(axis=0)) # E: matrix[Any, dtype[{intp}]]
  33. reveal_type(mat.argmin(axis=0)) # E: matrix[Any, dtype[{intp}]]
  34. reveal_type(mat.ptp(axis=0)) # E: matrix[Any, dtype[{int64}]]
  35. reveal_type(mat.sum(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  36. reveal_type(mat.mean(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  37. reveal_type(mat.std(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  38. reveal_type(mat.var(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  39. reveal_type(mat.prod(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  40. reveal_type(mat.any(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  41. reveal_type(mat.all(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  42. reveal_type(mat.max(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  43. reveal_type(mat.min(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  44. reveal_type(mat.argmax(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  45. reveal_type(mat.argmin(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  46. reveal_type(mat.ptp(out=ar_f8)) # E: ndarray[Any, dtype[{float64}]]
  47. reveal_type(mat.T) # E: matrix[Any, dtype[{int64}]]
  48. reveal_type(mat.I) # E: matrix[Any, Any]
  49. reveal_type(mat.A) # E: ndarray[Any, dtype[{int64}]]
  50. reveal_type(mat.A1) # E: ndarray[Any, dtype[{int64}]]
  51. reveal_type(mat.H) # E: matrix[Any, dtype[{int64}]]
  52. reveal_type(mat.getT()) # E: matrix[Any, dtype[{int64}]]
  53. reveal_type(mat.getI()) # E: matrix[Any, Any]
  54. reveal_type(mat.getA()) # E: ndarray[Any, dtype[{int64}]]
  55. reveal_type(mat.getA1()) # E: ndarray[Any, dtype[{int64}]]
  56. reveal_type(mat.getH()) # E: matrix[Any, dtype[{int64}]]
  57. reveal_type(np.bmat(ar_f8)) # E: matrix[Any, Any]
  58. reveal_type(np.bmat([[0, 1, 2]])) # E: matrix[Any, Any]
  59. reveal_type(np.bmat("mat")) # E: matrix[Any, Any]
  60. reveal_type(np.asmatrix(ar_f8, dtype=np.int64)) # E: matrix[Any, Any]