projektAI/venv/Lib/site-packages/numpy/typing/tests/data/reveal/dtype.py

53 lines
2.2 KiB
Python
Raw Normal View History

2021-06-06 22:13:05 +02:00
import numpy as np
dtype_obj: np.dtype[np.str_]
void_dtype_obj: np.dtype[np.void]
reveal_type(np.dtype(np.float64)) # E: numpy.dtype[numpy.floating[numpy.typing._64Bit]]
reveal_type(np.dtype(np.int64)) # E: numpy.dtype[numpy.signedinteger[numpy.typing._64Bit]]
# String aliases
reveal_type(np.dtype("float64")) # E: numpy.dtype[numpy.floating[numpy.typing._64Bit]]
reveal_type(np.dtype("float32")) # E: numpy.dtype[numpy.floating[numpy.typing._32Bit]]
reveal_type(np.dtype("int64")) # E: numpy.dtype[numpy.signedinteger[numpy.typing._64Bit]]
reveal_type(np.dtype("int32")) # E: numpy.dtype[numpy.signedinteger[numpy.typing._32Bit]]
reveal_type(np.dtype("bool")) # E: numpy.dtype[numpy.bool_]
reveal_type(np.dtype("bytes")) # E: numpy.dtype[numpy.bytes_]
reveal_type(np.dtype("str")) # E: numpy.dtype[numpy.str_]
# Python types
reveal_type(np.dtype(complex)) # E: numpy.dtype[numpy.complexfloating[numpy.typing._64Bit, numpy.typing._64Bit]]
reveal_type(np.dtype(float)) # E: numpy.dtype[numpy.floating[numpy.typing._64Bit]]
reveal_type(np.dtype(int)) # E: numpy.dtype
reveal_type(np.dtype(bool)) # E: numpy.dtype[numpy.bool_]
reveal_type(np.dtype(str)) # E: numpy.dtype[numpy.str_]
reveal_type(np.dtype(bytes)) # E: numpy.dtype[numpy.bytes_]
# Special case for None
reveal_type(np.dtype(None)) # E: numpy.dtype[numpy.floating[numpy.typing._64Bit]]
# Dtypes of dtypes
reveal_type(np.dtype(np.dtype(np.float64))) # E: numpy.dtype[numpy.floating[numpy.typing._64Bit]]
# Parameterized dtypes
reveal_type(np.dtype("S8")) # E: numpy.dtype
# Void
reveal_type(np.dtype(("U", 10))) # E: numpy.dtype[numpy.void]
reveal_type(dtype_obj.name) # E: str
reveal_type(dtype_obj.names) # E: Union[builtins.tuple[builtins.str], None]
reveal_type(dtype_obj * 0) # E: None
reveal_type(dtype_obj * 1) # E: numpy.dtype[numpy.str_]
reveal_type(dtype_obj * 2) # E: numpy.dtype[numpy.void]
reveal_type(0 * dtype_obj) # E: Any
reveal_type(1 * dtype_obj) # E: Any
reveal_type(2 * dtype_obj) # E: Any
reveal_type(void_dtype_obj["f0"]) # E: numpy.dtype[Any]
reveal_type(void_dtype_obj[0]) # E: numpy.dtype[Any]
reveal_type(void_dtype_obj[["f0", "f1"]]) # E: numpy.dtype[numpy.void]
reveal_type(void_dtype_obj[["f0"]]) # E: numpy.dtype[numpy.void]