🏷️ fix stubtest errors in numpy.polynomial.legendre

Author: jorenham

src/numpy-stubs/polynomial/legendre.pyi

@@ -1,32 +1,46 @@
-from typing import Final, Literal as L
+from typing import Final, Literal as L, SupportsIndex as CanIndex, overload
 
 import numpy as np
-from _numtype import Array_1d
+from _numtype import (
+    Array,
+    Array_1d,
+    CoComplex_1d,
+    CoFloating_1d,
+    CoFloating_nd,
+    CoInteger_1d,
+    ToComplex128_1d,
+    ToComplex_1d,
+    ToComplex_nd,
+    ToFloat64_1d,
+    ToObject_1d,
+    ToObject_nd,
+)
 
 from ._polybase import ABCPolyBase
-from ._polytypes import (
-    _FuncBinOp,
-    _FuncCompanion,
-    _FuncDer,
-    _FuncFit,
-    _FuncFromRoots,
-    _FuncGauss,
-    _FuncInteg,
-    _FuncLine,
-    _FuncPoly2Ortho,
-    _FuncPow,
-    _FuncRoots,
-    _FuncUnOp,
-    _FuncVal,
-    _FuncVal2D,
-    _FuncVal3D,
-    _FuncValFromRoots,
-    _FuncVander,
-    _FuncVander2D,
-    _FuncVander3D,
-    _FuncWeight,
+from .polynomial import (
+    polyadd as legadd,
+    polycompanion as legcompanion,
+    polyder as legder,
+    polydiv as legdiv,
+    polyfit as legfit,
+    polyfromroots as legfromroots,
+    polygrid2d as leggrid2d,
+    polygrid3d as leggrid3d,
+    polyint as legint,
+    polyline as legline,
+    polymul as legmul,
+    polymulx as legmulx,
+    polypow as legpow,
+    polyroots as legroots,
+    polysub as legsub,
+    polytrim as legtrim,
+    polyval as legval,
+    polyval2d as legval2d,
+    polyval3d as legval3d,
+    polyvander as legvander,
+    polyvander2d as legvander2d,
+    polyvander3d as legvander3d,
 )
-from .polyutils import trimcoef as legtrim
 
 __all__ = [
     "Legendre",
@@ -62,39 +76,55 @@ __all__ = [
     "poly2leg",
 ]
 
+###
+
 legdomain: Final[Array_1d[np.float64]] = ...
 legzero: Final[Array_1d[np.int_]] = ...
 legone: Final[Array_1d[np.int_]] = ...
 legx: Final[Array_1d[np.int_]] = ...
 
-poly2leg: Final[_FuncPoly2Ortho[L["poly2leg"]]] = ...
-leg2poly: Final[_FuncUnOp[L["leg2poly"]]] = ...
-legline: Final[_FuncLine[L["legline"]]] = ...
-legfromroots: Final[_FuncFromRoots[L["legfromroots"]]] = ...
-legadd: Final[_FuncBinOp[L["legadd"]]] = ...
-legsub: Final[_FuncBinOp[L["legsub"]]] = ...
-legmulx: Final[_FuncUnOp[L["legmulx"]]] = ...
-legmul: Final[_FuncBinOp[L["legmul"]]] = ...
-legdiv: Final[_FuncBinOp[L["legdiv"]]] = ...
-legpow: Final[_FuncPow[L["legpow"]]] = ...
-legder: Final[_FuncDer[L["legder"]]] = ...
-legint: Final[_FuncInteg[L["legint"]]] = ...
-legval: Final[_FuncVal[L["legval"]]] = ...
-legval2d: Final[_FuncVal2D[L["legval2d"]]] = ...
-legval3d: Final[_FuncVal3D[L["legval3d"]]] = ...
-legvalfromroots: Final[_FuncValFromRoots[L["legvalfromroots"]]] = ...
-leggrid2d: Final[_FuncVal2D[L["leggrid2d"]]] = ...
-leggrid3d: Final[_FuncVal3D[L["leggrid3d"]]] = ...
-legvander: Final[_FuncVander[L["legvander"]]] = ...
-legvander2d: Final[_FuncVander2D[L["legvander2d"]]] = ...
-legvander3d: Final[_FuncVander3D[L["legvander3d"]]] = ...
-legfit: Final[_FuncFit[L["legfit"]]] = ...
-legcompanion: Final[_FuncCompanion[L["legcompanion"]]] = ...
-legroots: Final[_FuncRoots[L["legroots"]]] = ...
-leggauss: Final[_FuncGauss[L["leggauss"]]] = ...
-legweight: Final[_FuncWeight[L["legweight"]]] = ...
+###
 
 class Legendre(ABCPolyBase):
     domain: Array_1d[np.float64] = ...  # pyright: ignore[reportIncompatibleMethodOverride]
     window: Array_1d[np.float64] = ...  # pyright: ignore[reportIncompatibleMethodOverride]
     basis_name: L["P"] = "P"  # pyright: ignore[reportIncompatibleMethodOverride]
+
+###
+@overload
+def poly2leg(pol: ToFloat64_1d | CoInteger_1d) -> Array_1d[np.float64]: ...
+@overload
+def poly2leg(pol: CoFloating_1d) -> Array_1d[np.floating]: ...
+@overload
+def poly2leg(pol: ToComplex128_1d) -> Array_1d[np.complex128]: ...
+@overload
+def poly2leg(pol: ToComplex_1d) -> Array_1d[np.complexfloating]: ...
+@overload
+def poly2leg(pol: CoComplex_1d) -> Array_1d[np.inexact]: ...
+@overload
+def poly2leg(pol: ToObject_1d) -> Array_1d[np.object_]: ...
+
+#
+@overload
+def leg2poly(c: ToFloat64_1d | CoInteger_1d) -> Array_1d[np.float64]: ...
+@overload
+def leg2poly(c: CoFloating_1d) -> Array_1d[np.floating]: ...
+@overload
+def leg2poly(c: ToComplex128_1d) -> Array_1d[np.complex128]: ...
+@overload
+def leg2poly(c: ToComplex_1d) -> Array_1d[np.complexfloating]: ...
+@overload
+def leg2poly(c: CoComplex_1d) -> Array_1d[np.inexact]: ...
+@overload
+def leg2poly(c: ToObject_1d) -> Array_1d[np.object_]: ...
+
+#
+@overload
+def legweight(x: CoFloating_nd) -> Array[np.float64]: ...
+@overload
+def legweight(x: ToComplex_nd) -> Array[np.complex128]: ...
+@overload
+def legweight(x: ToObject_nd) -> Array[np.object_]: ...
+
+#
+def leggauss(deg: CanIndex) -> tuple[Array_1d[np.float64], Array_1d[np.float64]]: ...

tool/.mypyignore-todo

@@ -138,30 +138,3 @@ numpy.polynomial.laguerre.lagvander2d
 numpy.polynomial.laguerre.lagvander3d
 numpy.polynomial.laguerre.lagweight
 numpy.polynomial.laguerre.poly2lag
-
-numpy.polynomial.legendre.leg2poly
-numpy.polynomial.legendre.legadd
-numpy.polynomial.legendre.legcompanion
-numpy.polynomial.legendre.legder
-numpy.polynomial.legendre.legdiv
-numpy.polynomial.legendre.legfit
-numpy.polynomial.legendre.legfromroots
-numpy.polynomial.legendre.leggauss
-numpy.polynomial.legendre.leggrid2d
-numpy.polynomial.legendre.leggrid3d
-numpy.polynomial.legendre.legint
-numpy.polynomial.legendre.legline
-numpy.polynomial.legendre.legmul
-numpy.polynomial.legendre.legmulx
-numpy.polynomial.legendre.legpow
-numpy.polynomial.legendre.legroots
-numpy.polynomial.legendre.legsub
-numpy.polynomial.legendre.legval
-numpy.polynomial.legendre.legval2d
-numpy.polynomial.legendre.legval3d
-numpy.polynomial.legendre.legvalfromroots
-numpy.polynomial.legendre.legvander
-numpy.polynomial.legendre.legvander2d
-numpy.polynomial.legendre.legvander3d
-numpy.polynomial.legendre.legweight
-numpy.polynomial.legendre.poly2leg