codekingpro/portable-devtools

1from typing import Any, assert_type
2
3import numpy as np
4import numpy.typing as npt
5
6AR_f8: npt.NDArray[np.float64]
7AR_LIKE_f: list[float]
8interface_dict: dict[str, Any]
9
10assert_type(np.lib.stride_tricks.as_strided(AR_f8), npt.NDArray[np.float64])
11assert_type(np.lib.stride_tricks.as_strided(AR_LIKE_f), npt.NDArray[Any])
12assert_type(np.lib.stride_tricks.as_strided(AR_f8, strides=(1, 5)), npt.NDArray[np.float64])
13assert_type(np.lib.stride_tricks.as_strided(AR_f8, shape=[9, 20]), npt.NDArray[np.float64])
14
15assert_type(np.lib.stride_tricks.sliding_window_view(AR_f8, 5), npt.NDArray[np.float64])
16assert_type(np.lib.stride_tricks.sliding_window_view(AR_LIKE_f, (1, 5)), npt.NDArray[Any])
17assert_type(np.lib.stride_tricks.sliding_window_view(AR_f8, [9], axis=1), npt.NDArray[np.float64])
18
19assert_type(np.broadcast_to(AR_f8, 5), npt.NDArray[np.float64])
20assert_type(np.broadcast_to(AR_LIKE_f, (1, 5)), npt.NDArray[Any])
21assert_type(np.broadcast_to(AR_f8, [4, 6], subok=True), npt.NDArray[np.float64])
22
23assert_type(np.broadcast_shapes((1, 2), [3, 1], (3, 2)), tuple[Any, ...])
24assert_type(np.broadcast_shapes((6, 7), (5, 6, 1), 7, (5, 1, 7)), tuple[Any, ...])
25
26assert_type(np.broadcast_arrays(AR_f8, AR_f8), tuple[npt.NDArray[Any], ...])
27assert_type(np.broadcast_arrays(AR_f8, AR_LIKE_f), tuple[npt.NDArray[Any], ...])
28