codekingpro/portable-devtools
114k
1from typing import Any, assert_type
2
3import numpy as np
4import numpy.typing as npt
5
6AR_i8: npt.NDArray[np.int64]
7AR_f8: npt.NDArray[np.float64]
8
9assert_type(np.histogram_bin_edges(AR_i8, bins="auto"), npt.NDArray[Any])
10assert_type(np.histogram_bin_edges(AR_i8, bins="rice", range=(0, 3)), npt.NDArray[Any])
11assert_type(np.histogram_bin_edges(AR_i8, bins="scott", weights=AR_f8), npt.NDArray[Any])
12
13assert_type(np.histogram(AR_i8, bins="auto"), tuple[npt.NDArray[Any], npt.NDArray[Any]])
14assert_type(np.histogram(AR_i8, bins="rice", range=(0, 3)), tuple[npt.NDArray[Any], npt.NDArray[Any]])
15assert_type(np.histogram(AR_i8, bins="scott", weights=AR_f8), tuple[npt.NDArray[Any], npt.NDArray[Any]])
16assert_type(np.histogram(AR_f8, bins=1, density=True), tuple[npt.NDArray[Any], npt.NDArray[Any]])
17
18assert_type(np.histogramdd(AR_i8, bins=[1]),
19 tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])
20assert_type(np.histogramdd(AR_i8, range=[(0, 3)]),
21 tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])
22assert_type(np.histogramdd(AR_i8, weights=AR_f8),
23 tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])
24assert_type(np.histogramdd(AR_f8, density=True),
25 tuple[npt.NDArray[Any], tuple[npt.NDArray[Any], ...]])
26 