flatiter.py 471 B

1234567891011121314
  1. import numpy as np
  2. a: "np.flatiter[np.ndarray]"
  3. reveal_type(a.base) # E: numpy.ndarray*
  4. reveal_type(a.copy()) # E: numpy.ndarray*
  5. reveal_type(a.coords) # E: tuple[builtins.int]
  6. reveal_type(a.index) # E: int
  7. reveal_type(iter(a)) # E: Iterator[numpy.generic*]
  8. reveal_type(next(a)) # E: numpy.generic
  9. reveal_type(a[0]) # E: numpy.generic
  10. reveal_type(a[[0, 1, 2]]) # E: numpy.ndarray*
  11. reveal_type(a[...]) # E: numpy.ndarray*
  12. reveal_type(a[:]) # E: numpy.ndarray*