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