flatiter.pyi 819 B

1234567891011121314151617181920212223
  1. from typing import Any
  2. import numpy as np
  3. a: np.flatiter[np.ndarray[Any, np.dtype[np.str_]]]
  4. reveal_type(a.base) # E: ndarray[Any, dtype[str_]]
  5. reveal_type(a.copy()) # E: ndarray[Any, dtype[str_]]
  6. reveal_type(a.coords) # E: tuple[builtins.int, ...]
  7. reveal_type(a.index) # E: int
  8. reveal_type(iter(a)) # E: Any
  9. reveal_type(next(a)) # E: str_
  10. reveal_type(a[0]) # E: str_
  11. reveal_type(a[[0, 1, 2]]) # E: ndarray[Any, dtype[str_]]
  12. reveal_type(a[...]) # E: ndarray[Any, dtype[str_]]
  13. reveal_type(a[:]) # E: ndarray[Any, dtype[str_]]
  14. reveal_type(a[(...,)]) # E: ndarray[Any, dtype[str_]]
  15. reveal_type(a[(0,)]) # E: str_
  16. reveal_type(a.__array__()) # E: ndarray[Any, dtype[str_]]
  17. reveal_type(a.__array__(np.dtype(np.float64))) # E: ndarray[Any, dtype[{float64}]]
  18. a[0] = "a"
  19. a[:5] = "a"
  20. a[...] = "a"
  21. a[(...,)] = "a"