8000 TYP: add missing annotations for `numpy.object_.__new__` · numpy/numpy@414efe1 · GitHub
[go: up one dir, main page]
Skip to content

Commit 414efe1

Browse files
jorenhamjorenham
committed
TYP: add missing annotations for numpy.object_.__new__ 
1 parent b46446e commit 414efe1

File tree

2 files changed

+61
-5
lines changed

2 files changed

+61
-5
lines changed

‎numpy/__init__.pyi

Lines changed: 32 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2612,8 +2612,6 @@ _NBit1 = TypeVar("_NBit1", bound=NBitBase)
26122612
_NBit2 = TypeVar("_NBit2", bound=NBitBase)
26132613

26142614
class generic(_ArrayOrScalarCommon):
2615-
@abstractmethod
2616-
def __init__(self, *args: Any, **kwargs: Any) -> None: ...
26172615
@overload
26182616
def __array__(self: _ScalarType, dtype: None = ..., /) -> NDArray[_ScalarType]: ...
26192617
@overload
@@ -2839,12 +2837,41 @@ class bool(generic):
28392837

28402838
bool_ = bool
28412839

2840+
_StringType = TypeVar("_StringType", bound=str | bytes)
2841+
_StringType = TypeVar("_StringType", bound=str | bytes)
2842+
_ShapeType = TypeVar("_ShapeType", bound=Any)
2843+
_ObjectType = TypeVar("_ObjectType", bound=object)
2844+
2845+
# A sequence-like interface like `collections.abc.Sequence`, but without the
2846+
# irrelevant methods.
2847+
class _SimpleSequence(Protocol):
2848+
def __len__(self, /) -> int: ...
2849+
def __getitem__(self, index: int, /) -> Any: ...
2850+
2851+
# The `object_` constructor returns the passed object, so instances with type
2852+
# `object_` cannot exists (at runtime).
28422853
class object_(generic):
2843-
def __init__(self, value: object = ..., /) -> None: ...
2854+
@overload
2855+
def __new__(cls, nothing_to_see_here: None = ..., /) -> None: ...
2856+
@overload
2857+
def __new__(cls, stringy: _StringType, /) -> _StringType: ...
2858+
@overload
2859+
def __new__(
2860+
cls,
2861+
array: ndarray[_ShapeType, Any], /,
2862+
) -> ndarray[_ShapeType, dtype[object_]]: ...
2863+
@overload
2864+
def __new__(cls, sequence: _SimpleSequence, /) -> NDArray[object_]: ...
2865+
@overload
2866+
def __new__(cls, value: _ObjectType, /) -> _ObjectType: ...
2867+
# catch-all
2868+
@overload
2869+
def __new__(cls, value: Any = ..., /) -> object | NDArray[object_]: ...
2870+
28442871
@property
2845-
def real(self: _ArraySelf) -> _ArraySelf: ...
2872+
def real(self) -> object_: ...
28462873
@property
2847-
def imag(self: _ArraySelf) -> _ArraySelf: ...
2874+
def imag(self) -> object_: ...
28482875
# The 3 protocols below may or may not raise,
28492876
# depending on the underlying object
28502877
def __int__(self) -> int: ...

‎numpy/typing/tests/data/reveal/scalars.pyi

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,11 @@ m: np.timedelta64
1919
U: np.str_
2020
S: np.bytes_
2121
V: np.void
22+
O: np.object_ # cannot exists at runtime
23+
24+
array_nd: np.ndarray[Any, Any]
25+
array_0d: np.ndarray[tuple[()], Any]
26+
array_2d_2x2: np.ndarray[tuple[Literal[2], Literal[2]], Any]
2227

2328
assert_type(c8.real, np.float32)
2429
assert_type(c8.imag, np.float32)
@@ -156,3 +161,27 @@ assert_type(f8.__ceil__(), int)
156161
assert_type(f8.__floor__(), int)
157162

158163
assert_type(i8.is_integer(), Literal[True])
164+
165+
assert_type(O.real, np.object_)
166+
assert_type(O.imag, np.object_)
167+
assert_type(int(O), int)
168+
assert_type(float(O), float)
169+
assert_type(complex(O), complex)
170+
171+
# These fail fail because of a mypy __new__ bug:
172+
# https://github.com/python/mypy/issues/15182
173+
# According to the typing spec, the following statements are valid, see
174+
# https://typing.readthedocs.io/en/latest/spec/constructors.html#new-method
175+
176+
# assert_type(np.object_(), None)
177+
# assert_type(np.object_(None), None)
178+
# assert_type(np.object_(array_nd), np.ndarray[Any, np.dtype[np.object_]])
179+
# assert_type(np.object_([]), npt.NDArray[np.object_])
180+
# assert_type(np.object_(()), npt.NDArray[np.object_])
181+
# assert_type(np.object_(range(4)), npt.NDArray[np.object_])
182+
# assert_type(np.object_(+42), int)
183+
# assert_type(np.object_(1 / 137), float)
184+
# assert_type(np.object_('Developers! ' * (1 << 6)), str)
185+
# assert_type(np.object_(object()), object)
186+
# assert_type(np.object_({False, True, NotADirectoryError}), set[Any])
187+
# assert_type(np.object_({'spam': 'food', 'ham': 'food'}), dict[str, str])

0 commit comments

Comments
 (0)
0