1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
| """Based on the `if __name__ == "__main__"` test code in `lib/_user_array_impl.py`."""
|
| from __future__ import annotations
|
| import numpy as np
| from numpy.lib.user_array import container
|
| N = 10_000
| W = H = int(N**0.5)
|
| a: np.ndarray[tuple[int, int], np.dtype[np.int32]]
| ua: container[tuple[int, int], np.dtype[np.int32]]
|
| a = np.arange(N, dtype=np.int32).reshape(W, H)
| ua = container(a)
|
| ua_small: container[tuple[int, int], np.dtype[np.int32]] = ua[:3, :5]
| ua_small[0, 0] = 10
|
| ua_bool: container[tuple[int, int], np.dtype[np.bool]] = ua_small > 1
|
| # shape: tuple[int, int] = np.shape(ua)
|
|
