🚚 port numpy.ma.min

Co-Authored-By: Marco Edward Gorelli <33491632+marcogorelli@users.noreply.github.com>
Co-Authored-By: Joren Hammudoglu <jhammudoglu@gmail.com>

Author: 3 people

src/numpy-stubs/@test/static/accept/ma.pyi

@@ -1,6 +1,8 @@
+from typing import Any, TypeAlias, TypeVar
 from typing_extensions import assert_type
 
 import numpy as np
+from numpy._typing import _Shape
 
 m: np.ma.MaskedArray[tuple[int], np.dtype[np.float64]]
 
@@ -10,3 +12,14 @@ assert_type(m.dtype, np.dtype[np.float64])
 
 assert_type(int(m), int)
 assert_type(float(m), float)
+
+_ScalarType_co = TypeVar("_ScalarType_co", bound=np.generic, covariant=True)
+MaskedNDArray: TypeAlias = np.ma.MaskedArray[_Shape, np.dtype[_ScalarType_co]]
+
+class MaskedNDArraySubclass(MaskedNDArray[np.complex128]): ...
+
+MAR_b: MaskedNDArray[np.bool]
+MAR_f4: MaskedNDArray[np.float32]
+MAR_i8: MaskedNDArray[np.int64]
+MAR_subclass: MaskedNDArraySubclass
+MAR_1d: np.ma.MaskedArray[tuple[int], np.dtype[Any]]

src/numpy-stubs/@test/static/reject/ma.pyi

@@ -4,3 +4,8 @@ m: np.ma.MaskedArray[tuple[int], np.dtype[np.float64]]
 
 m.shape = (3, 1)  # type: ignore[assignment]
 m.dtype = np.bool  # type: ignore[assignment]  # pyright: ignore[reportAttributeAccessIssue]
+
+np.amin(m, axis=1.0)  # type: ignore[call-overload] # pyright: ignore[reportArgumentType, reportCallIssue]
+np.amin(m, keepdims=1.0)  # type: ignore[call-overload] # pyright: ignore[reportArgumentType, reportCallIssue]
+np.amin(m, out=1.0)  # type: ignore[call-overload] # pyright: ignore[reportArgumentType, reportCallIssue]
+np.amin(m, fill_value=lambda x: 27)  # type: ignore[call-overload] # pyright: ignore[reportCallIssue, reportUnknownLambdaType]

src/numpy-stubs/ma/core.pyi

@@ -5,7 +5,8 @@ from typing_extensions import Never, Self, TypeVar, deprecated, overload, overri
 import numpy as np
 from _numtype import Array, ToGeneric_0d, ToGeneric_1nd, ToGeneric_nd
 from numpy import _OrderACF, _OrderKACF, amax, amin, bool_, expand_dims  # noqa: ICN003
-from numpy._typing import _BoolCodes
+from numpy._globals import _NoValueType
+from numpy._typing import ArrayLike, _ArrayLike, _BoolCodes, _ScalarLike_co, _ShapeLike
 
 __all__ = [
     "MAError",
@@ -195,6 +196,8 @@ _DTypeT = TypeVar("_DTypeT", bound=np.dtype)
 _DTypeT_co = TypeVar("_DTypeT_co", bound=np.dtype, default=np.dtype, covariant=True)
 
 _DTypeLikeBool: TypeAlias = type[bool | np.bool] | np.dtype[np.bool] | _BoolCodes
+_ArrayType = TypeVar("_ArrayType", bound=np.ndarray[Any, Any])
+_SCT = TypeVar("_SCT", bound=np.generic)
 
 ###
 
@@ -818,13 +821,39 @@ def array(
 ) -> Incomplete: ...
 
 #
+@overload
 def min(
-    obj: Incomplete,
-    axis: Incomplete = ...,
-    out: Incomplete = ...,
-    fill_value: Incomplete = ...,
-    keepdims: Incomplete = ...,
-) -> Incomplete: ...
+    obj: _ArrayLike[_SCT],
+    axis: None = None,
+    out: None = None,
+    fill_value: _ScalarLike_co | None = None,
+    keepdims: L[False] | _NoValueType = ...,
+) -> _SCT: ...
+@overload
+def min(
+    obj: ArrayLike,
+    axis: _ShapeLike | None = None,
+    out: None = None,
+    fill_value: _ScalarLike_co | None = None,
+    keepdims: bool | _NoValueType = ...,
+) -> Any: ...
+@overload
+def min(
+    obj: ArrayLike,
+    axis: None,
+    out: _ArrayType,
+    fill_value: _ScalarLike_co | None = None,
+    keepdims: bool | _NoValueType = ...,
+) -> _ArrayType: ...
+@overload
+def min(
+    obj: ArrayLike,
+    axis: _ShapeLike | None = None,
+    *,
+    out: _ArrayType,
+    fill_value: _ScalarLike_co | None = None,
+    keepdims: bool | _NoValueType = ...,
+) -> _ArrayType: ...
 def max(
     obj: Incomplete,
     axis: Incomplete = ...,