🚚 port numpy.ma.min

Co-Authored-By: Marco Edward Gorelli <[email protected]>

Author: guan404ming

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

@@ -0,0 +1,8 @@
+import numpy as np
+
+m: np.ma.MaskedArray[tuple[int], np.dtype[np.float64]]
+
+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)
 
 ###
 
@@ -824,13 +827,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 = ...,