chararray.pyi 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. from typing import Any
  2. import numpy as np
  3. import numpy.typing as npt
  4. from typing_extensions import assert_type
  5. AR_U: np.char.chararray[tuple[int, ...], np.dtype[np.str_]]
  6. AR_S: np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]]
  7. assert_type(AR_U == AR_U, npt.NDArray[np.bool])
  8. assert_type(AR_S == AR_S, npt.NDArray[np.bool])
  9. assert_type(AR_U != AR_U, npt.NDArray[np.bool])
  10. assert_type(AR_S != AR_S, npt.NDArray[np.bool])
  11. assert_type(AR_U >= AR_U, npt.NDArray[np.bool])
  12. assert_type(AR_S >= AR_S, npt.NDArray[np.bool])
  13. assert_type(AR_U <= AR_U, npt.NDArray[np.bool])
  14. assert_type(AR_S <= AR_S, npt.NDArray[np.bool])
  15. assert_type(AR_U > AR_U, npt.NDArray[np.bool])
  16. assert_type(AR_S > AR_S, npt.NDArray[np.bool])
  17. assert_type(AR_U < AR_U, npt.NDArray[np.bool])
  18. assert_type(AR_S < AR_S, npt.NDArray[np.bool])
  19. assert_type(AR_U * 5, np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  20. assert_type(AR_S * [5], np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  21. assert_type(AR_U % "test", np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  22. assert_type(AR_S % b"test", np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  23. assert_type(AR_U.capitalize(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  24. assert_type(AR_S.capitalize(), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  25. assert_type(AR_U.center(5), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  26. assert_type(AR_S.center([2, 3, 4], b"a"), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  27. assert_type(AR_U.encode(), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  28. assert_type(AR_S.decode(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  29. assert_type(AR_U.expandtabs(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  30. assert_type(AR_S.expandtabs(tabsize=4), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  31. assert_type(AR_U.join("_"), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  32. assert_type(AR_S.join([b"_", b""]), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  33. assert_type(AR_U.ljust(5), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  34. assert_type(AR_S.ljust([4, 3, 1], fillchar=[b"a", b"b", b"c"]), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  35. assert_type(AR_U.rjust(5), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  36. assert_type(AR_S.rjust([4, 3, 1], fillchar=[b"a", b"b", b"c"]), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  37. assert_type(AR_U.lstrip(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  38. assert_type(AR_S.lstrip(chars=b"_"), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  39. assert_type(AR_U.rstrip(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  40. assert_type(AR_S.rstrip(chars=b"_"), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  41. assert_type(AR_U.strip(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  42. assert_type(AR_S.strip(chars=b"_"), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  43. assert_type(AR_U.partition("\n"), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  44. assert_type(AR_S.partition([b"a", b"b", b"c"]), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  45. assert_type(AR_U.rpartition("\n"), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  46. assert_type(AR_S.rpartition([b"a", b"b", b"c"]), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  47. assert_type(AR_U.replace("_", "-"), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  48. assert_type(AR_S.replace([b"_", b""], [b"a", b"b"]), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  49. assert_type(AR_U.split("_"), npt.NDArray[np.object_])
  50. assert_type(AR_S.split(maxsplit=[1, 2, 3]), npt.NDArray[np.object_])
  51. assert_type(AR_U.rsplit("_"), npt.NDArray[np.object_])
  52. assert_type(AR_S.rsplit(maxsplit=[1, 2, 3]), npt.NDArray[np.object_])
  53. assert_type(AR_U.splitlines(), npt.NDArray[np.object_])
  54. assert_type(AR_S.splitlines(keepends=[True, True, False]), npt.NDArray[np.object_])
  55. assert_type(AR_U.swapcase(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  56. assert_type(AR_S.swapcase(), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  57. assert_type(AR_U.title(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  58. assert_type(AR_S.title(), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  59. assert_type(AR_U.upper(), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  60. assert_type(AR_S.upper(), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  61. assert_type(AR_U.zfill(5), np.char.chararray[tuple[int, ...], np.dtype[np.str_]])
  62. assert_type(AR_S.zfill([2, 3, 4]), np.char.chararray[tuple[int, ...], np.dtype[np.bytes_]])
  63. assert_type(AR_U.count("a", start=[1, 2, 3]), npt.NDArray[np.int_])
  64. assert_type(AR_S.count([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
  65. assert_type(AR_U.endswith("a", start=[1, 2, 3]), npt.NDArray[np.bool])
  66. assert_type(AR_S.endswith([b"a", b"b", b"c"], end=9), npt.NDArray[np.bool])
  67. assert_type(AR_U.startswith("a", start=[1, 2, 3]), npt.NDArray[np.bool])
  68. assert_type(AR_S.startswith([b"a", b"b", b"c"], end=9), npt.NDArray[np.bool])
  69. assert_type(AR_U.find("a", start=[1, 2, 3]), npt.NDArray[np.int_])
  70. assert_type(AR_S.find([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
  71. assert_type(AR_U.rfind("a", start=[1, 2, 3]), npt.NDArray[np.int_])
  72. assert_type(AR_S.rfind([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
  73. assert_type(AR_U.index("a", start=[1, 2, 3]), npt.NDArray[np.int_])
  74. assert_type(AR_S.index([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
  75. assert_type(AR_U.rindex("a", start=[1, 2, 3]), npt.NDArray[np.int_])
  76. assert_type(AR_S.rindex([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
  77. assert_type(AR_U.isalpha(), npt.NDArray[np.bool])
  78. assert_type(AR_S.isalpha(), npt.NDArray[np.bool])
  79. assert_type(AR_U.isalnum(), npt.NDArray[np.bool])
  80. assert_type(AR_S.isalnum(), npt.NDArray[np.bool])
  81. assert_type(AR_U.isdecimal(), npt.NDArray[np.bool])
  82. assert_type(AR_S.isdecimal(), npt.NDArray[np.bool])
  83. assert_type(AR_U.isdigit(), npt.NDArray[np.bool])
  84. assert_type(AR_S.isdigit(), npt.NDArray[np.bool])
  85. assert_type(AR_U.islower(), npt.NDArray[np.bool])
  86. assert_type(AR_S.islower(), npt.NDArray[np.bool])
  87. assert_type(AR_U.isnumeric(), npt.NDArray[np.bool])
  88. assert_type(AR_S.isnumeric(), npt.NDArray[np.bool])
  89. assert_type(AR_U.isspace(), npt.NDArray[np.bool])
  90. assert_type(AR_S.isspace(), npt.NDArray[np.bool])
  91. assert_type(AR_U.istitle(), npt.NDArray[np.bool])
  92. assert_type(AR_S.istitle(), npt.NDArray[np.bool])
  93. assert_type(AR_U.isupper(), npt.NDArray[np.bool])
  94. assert_type(AR_S.isupper(), npt.NDArray[np.bool])
  95. assert_type(AR_U.__array_finalize__(object()), None)
  96. assert_type(AR_S.__array_finalize__(object()), None)