memmap.pyi 754 B

123456789101112131415161718192021
  1. from typing import Any
  2. import numpy as np
  3. from typing_extensions import assert_type
  4. memmap_obj: np.memmap[Any, np.dtype[np.str_]]
  5. assert_type(np.memmap.__array_priority__, float)
  6. assert_type(memmap_obj.__array_priority__, float)
  7. assert_type(memmap_obj.filename, str | None)
  8. assert_type(memmap_obj.offset, int)
  9. assert_type(memmap_obj.mode, str)
  10. assert_type(memmap_obj.flush(), None)
  11. assert_type(np.memmap("file.txt", offset=5), np.memmap[Any, np.dtype[np.uint8]])
  12. assert_type(np.memmap(b"file.txt", dtype=np.float64, shape=(10, 3)), np.memmap[Any, np.dtype[np.float64]])
  13. with open("file.txt", "rb") as f:
  14. assert_type(np.memmap(f, dtype=float, order="K"), np.memmap[Any, np.dtype[Any]])
  15. assert_type(memmap_obj.__array_finalize__(object()), None)