char.pyi 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. import numpy as np
  2. import numpy.typing as npt
  3. from collections.abc import Sequence
  4. AR_U: npt.NDArray[np.str_]
  5. AR_S: npt.NDArray[np.bytes_]
  6. reveal_type(np.char.equal(AR_U, AR_U)) # E: ndarray[Any, dtype[bool_]]
  7. reveal_type(np.char.equal(AR_S, AR_S)) # E: ndarray[Any, dtype[bool_]]
  8. reveal_type(np.char.not_equal(AR_U, AR_U)) # E: ndarray[Any, dtype[bool_]]
  9. reveal_type(np.char.not_equal(AR_S, AR_S)) # E: ndarray[Any, dtype[bool_]]
  10. reveal_type(np.char.greater_equal(AR_U, AR_U)) # E: ndarray[Any, dtype[bool_]]
  11. reveal_type(np.char.greater_equal(AR_S, AR_S)) # E: ndarray[Any, dtype[bool_]]
  12. reveal_type(np.char.less_equal(AR_U, AR_U)) # E: ndarray[Any, dtype[bool_]]
  13. reveal_type(np.char.less_equal(AR_S, AR_S)) # E: ndarray[Any, dtype[bool_]]
  14. reveal_type(np.char.greater(AR_U, AR_U)) # E: ndarray[Any, dtype[bool_]]
  15. reveal_type(np.char.greater(AR_S, AR_S)) # E: ndarray[Any, dtype[bool_]]
  16. reveal_type(np.char.less(AR_U, AR_U)) # E: ndarray[Any, dtype[bool_]]
  17. reveal_type(np.char.less(AR_S, AR_S)) # E: ndarray[Any, dtype[bool_]]
  18. reveal_type(np.char.multiply(AR_U, 5)) # E: ndarray[Any, dtype[str_]]
  19. reveal_type(np.char.multiply(AR_S, [5, 4, 3])) # E: ndarray[Any, dtype[bytes_]]
  20. reveal_type(np.char.mod(AR_U, "test")) # E: ndarray[Any, dtype[str_]]
  21. reveal_type(np.char.mod(AR_S, "test")) # E: ndarray[Any, dtype[bytes_]]
  22. reveal_type(np.char.capitalize(AR_U)) # E: ndarray[Any, dtype[str_]]
  23. reveal_type(np.char.capitalize(AR_S)) # E: ndarray[Any, dtype[bytes_]]
  24. reveal_type(np.char.center(AR_U, 5)) # E: ndarray[Any, dtype[str_]]
  25. reveal_type(np.char.center(AR_S, [2, 3, 4], b"a")) # E: ndarray[Any, dtype[bytes_]]
  26. reveal_type(np.char.encode(AR_U)) # E: ndarray[Any, dtype[bytes_]]
  27. reveal_type(np.char.decode(AR_S)) # E: ndarray[Any, dtype[str_]]
  28. reveal_type(np.char.expandtabs(AR_U)) # E: ndarray[Any, dtype[str_]]
  29. reveal_type(np.char.expandtabs(AR_S, tabsize=4)) # E: ndarray[Any, dtype[bytes_]]
  30. reveal_type(np.char.join(AR_U, "_")) # E: ndarray[Any, dtype[str_]]
  31. reveal_type(np.char.join(AR_S, [b"_", b""])) # E: ndarray[Any, dtype[bytes_]]
  32. reveal_type(np.char.ljust(AR_U, 5)) # E: ndarray[Any, dtype[str_]]
  33. reveal_type(np.char.ljust(AR_S, [4, 3, 1], fillchar=[b"a", b"b", b"c"])) # E: ndarray[Any, dtype[bytes_]]
  34. reveal_type(np.char.rjust(AR_U, 5)) # E: ndarray[Any, dtype[str_]]
  35. reveal_type(np.char.rjust(AR_S, [4, 3, 1], fillchar=[b"a", b"b", b"c"])) # E: ndarray[Any, dtype[bytes_]]
  36. reveal_type(np.char.lstrip(AR_U)) # E: ndarray[Any, dtype[str_]]
  37. reveal_type(np.char.lstrip(AR_S, chars=b"_")) # E: ndarray[Any, dtype[bytes_]]
  38. reveal_type(np.char.rstrip(AR_U)) # E: ndarray[Any, dtype[str_]]
  39. reveal_type(np.char.rstrip(AR_S, chars=b"_")) # E: ndarray[Any, dtype[bytes_]]
  40. reveal_type(np.char.strip(AR_U)) # E: ndarray[Any, dtype[str_]]
  41. reveal_type(np.char.strip(AR_S, chars=b"_")) # E: ndarray[Any, dtype[bytes_]]
  42. reveal_type(np.char.partition(AR_U, "\n")) # E: ndarray[Any, dtype[str_]]
  43. reveal_type(np.char.partition(AR_S, [b"a", b"b", b"c"])) # E: ndarray[Any, dtype[bytes_]]
  44. reveal_type(np.char.rpartition(AR_U, "\n")) # E: ndarray[Any, dtype[str_]]
  45. reveal_type(np.char.rpartition(AR_S, [b"a", b"b", b"c"])) # E: ndarray[Any, dtype[bytes_]]
  46. reveal_type(np.char.replace(AR_U, "_", "-")) # E: ndarray[Any, dtype[str_]]
  47. reveal_type(np.char.replace(AR_S, [b"_", b""], [b"a", b"b"])) # E: ndarray[Any, dtype[bytes_]]
  48. reveal_type(np.char.split(AR_U, "_")) # E: ndarray[Any, dtype[object_]]
  49. reveal_type(np.char.split(AR_S, maxsplit=[1, 2, 3])) # E: ndarray[Any, dtype[object_]]
  50. reveal_type(np.char.rsplit(AR_U, "_")) # E: ndarray[Any, dtype[object_]]
  51. reveal_type(np.char.rsplit(AR_S, maxsplit=[1, 2, 3])) # E: ndarray[Any, dtype[object_]]
  52. reveal_type(np.char.splitlines(AR_U)) # E: ndarray[Any, dtype[object_]]
  53. reveal_type(np.char.splitlines(AR_S, keepends=[True, True, False])) # E: ndarray[Any, dtype[object_]]
  54. reveal_type(np.char.swapcase(AR_U)) # E: ndarray[Any, dtype[str_]]
  55. reveal_type(np.char.swapcase(AR_S)) # E: ndarray[Any, dtype[bytes_]]
  56. reveal_type(np.char.title(AR_U)) # E: ndarray[Any, dtype[str_]]
  57. reveal_type(np.char.title(AR_S)) # E: ndarray[Any, dtype[bytes_]]
  58. reveal_type(np.char.upper(AR_U)) # E: ndarray[Any, dtype[str_]]
  59. reveal_type(np.char.upper(AR_S)) # E: ndarray[Any, dtype[bytes_]]
  60. reveal_type(np.char.zfill(AR_U, 5)) # E: ndarray[Any, dtype[str_]]
  61. reveal_type(np.char.zfill(AR_S, [2, 3, 4])) # E: ndarray[Any, dtype[bytes_]]
  62. reveal_type(np.char.count(AR_U, "a", start=[1, 2, 3])) # E: ndarray[Any, dtype[{int_}]]
  63. reveal_type(np.char.count(AR_S, [b"a", b"b", b"c"], end=9)) # E: ndarray[Any, dtype[{int_}]]
  64. reveal_type(np.char.endswith(AR_U, "a", start=[1, 2, 3])) # E: ndarray[Any, dtype[bool_]]
  65. reveal_type(np.char.endswith(AR_S, [b"a", b"b", b"c"], end=9)) # E: ndarray[Any, dtype[bool_]]
  66. reveal_type(np.char.startswith(AR_U, "a", start=[1, 2, 3])) # E: ndarray[Any, dtype[bool_]]
  67. reveal_type(np.char.startswith(AR_S, [b"a", b"b", b"c"], end=9)) # E: ndarray[Any, dtype[bool_]]
  68. reveal_type(np.char.find(AR_U, "a", start=[1, 2, 3])) # E: ndarray[Any, dtype[{int_}]]
  69. reveal_type(np.char.find(AR_S, [b"a", b"b", b"c"], end=9)) # E: ndarray[Any, dtype[{int_}]]
  70. reveal_type(np.char.rfind(AR_U, "a", start=[1, 2, 3])) # E: ndarray[Any, dtype[{int_}]]
  71. reveal_type(np.char.rfind(AR_S, [b"a", b"b", b"c"], end=9)) # E: ndarray[Any, dtype[{int_}]]
  72. reveal_type(np.char.index(AR_U, "a", start=[1, 2, 3])) # E: ndarray[Any, dtype[{int_}]]
  73. reveal_type(np.char.index(AR_S, [b"a", b"b", b"c"], end=9)) # E: ndarray[Any, dtype[{int_}]]
  74. reveal_type(np.char.rindex(AR_U, "a", start=[1, 2, 3])) # E: ndarray[Any, dtype[{int_}]]
  75. reveal_type(np.char.rindex(AR_S, [b"a", b"b", b"c"], end=9)) # E: ndarray[Any, dtype[{int_}]]
  76. reveal_type(np.char.isalpha(AR_U)) # E: ndarray[Any, dtype[bool_]]
  77. reveal_type(np.char.isalpha(AR_S)) # E: ndarray[Any, dtype[bool_]]
  78. reveal_type(np.char.isalnum(AR_U)) # E: ndarray[Any, dtype[bool_]]
  79. reveal_type(np.char.isalnum(AR_S)) # E: ndarray[Any, dtype[bool_]]
  80. reveal_type(np.char.isdecimal(AR_U)) # E: ndarray[Any, dtype[bool_]]
  81. reveal_type(np.char.isdecimal(AR_S)) # E: ndarray[Any, dtype[bool_]]
  82. reveal_type(np.char.isdigit(AR_U)) # E: ndarray[Any, dtype[bool_]]
  83. reveal_type(np.char.isdigit(AR_S)) # E: ndarray[Any, dtype[bool_]]
  84. reveal_type(np.char.islower(AR_U)) # E: ndarray[Any, dtype[bool_]]
  85. reveal_type(np.char.islower(AR_S)) # E: ndarray[Any, dtype[bool_]]
  86. reveal_type(np.char.isnumeric(AR_U)) # E: ndarray[Any, dtype[bool_]]
  87. reveal_type(np.char.isnumeric(AR_S)) # E: ndarray[Any, dtype[bool_]]
  88. reveal_type(np.char.isspace(AR_U)) # E: ndarray[Any, dtype[bool_]]
  89. reveal_type(np.char.isspace(AR_S)) # E: ndarray[Any, dtype[bool_]]
  90. reveal_type(np.char.istitle(AR_U)) # E: ndarray[Any, dtype[bool_]]
  91. reveal_type(np.char.istitle(AR_S)) # E: ndarray[Any, dtype[bool_]]
  92. reveal_type(np.char.isupper(AR_U)) # E: ndarray[Any, dtype[bool_]]
  93. reveal_type(np.char.isupper(AR_S)) # E: ndarray[Any, dtype[bool_]]
  94. reveal_type(np.char.str_len(AR_U)) # E: ndarray[Any, dtype[{int_}]]
  95. reveal_type(np.char.str_len(AR_S)) # E: ndarray[Any, dtype[{int_}]]
  96. reveal_type(np.char.array(AR_U)) # E: chararray[Any, dtype[str_]]
  97. reveal_type(np.char.array(AR_S, order="K")) # E: chararray[Any, dtype[bytes_]]
  98. reveal_type(np.char.array("bob", copy=True)) # E: chararray[Any, dtype[str_]]
  99. reveal_type(np.char.array(b"bob", itemsize=5)) # E: chararray[Any, dtype[bytes_]]
  100. reveal_type(np.char.array(1, unicode=False)) # E: chararray[Any, dtype[bytes_]]
  101. reveal_type(np.char.array(1, unicode=True)) # E: chararray[Any, dtype[str_]]
  102. reveal_type(np.char.asarray(AR_U)) # E: chararray[Any, dtype[str_]]
  103. reveal_type(np.char.asarray(AR_S, order="K")) # E: chararray[Any, dtype[bytes_]]
  104. reveal_type(np.char.asarray("bob")) # E: chararray[Any, dtype[str_]]
  105. reveal_type(np.char.asarray(b"bob", itemsize=5)) # E: chararray[Any, dtype[bytes_]]
  106. reveal_type(np.char.asarray(1, unicode=False)) # E: chararray[Any, dtype[bytes_]]
  107. reveal_type(np.char.asarray(1, unicode=True)) # E: chararray[Any, dtype[str_]]