codekingpro/portable-devtools

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