codekingpro/portable-devtools

1import contextlib
2from collections.abc import Callable
3from typing import assert_type
4
5import numpy as np
6import numpy.typing as npt
7from numpy._core.arrayprint import _FormatOptions
8
9AR: npt.NDArray[np.int64]
10func_float: Callable[[np.floating], str]
11func_int: Callable[[np.integer], str]
12
13assert_type(np.get_printoptions(), _FormatOptions)
14assert_type(
15    np.array2string(AR, formatter={"float_kind": func_float, "int_kind": func_int}),
16    str,
17)
18assert_type(np.format_float_scientific(1.0), str)
19assert_type(np.format_float_positional(1), str)
20assert_type(np.array_repr(AR), str)
21assert_type(np.array_str(AR), str)
22
23assert_type(np.printoptions(), contextlib._GeneratorContextManager[_FormatOptions])
24with np.printoptions() as dct:
25    assert_type(dct, _FormatOptions)
26