fmpz_mod_mat: add type stubs

Author: oscarbenjamin

src/flint/types/fmpq_mat.pyi

@@ -11,12 +11,9 @@ from flint.types.fmpz_mat import fmpz_mat
 
 _str = str
 
-ifmpq_mat = fmpq_mat | fmpz_mat | Sequence[Sequence[ifmpq]]
-
-
 class fmpq_mat(flint_mat[fmpq]):
     @overload
-    def __init__(self, val: ifmpq_mat, /) -> None: ...
+    def __init__(self, val: fmpq_mat | fmpz_mat | Sequence[Sequence[ifmpq]], /) -> None: ...
     @overload
     def __init__(self, m: int, n: int, /) -> None: ...
     @overload

src/flint/types/fmpz_mat.pyi

@@ -10,12 +10,9 @@ from flint.types.fmpz_poly import fmpz_poly
 
 _str = str
 
-ifmpz_mat = fmpz_mat | Sequence[Sequence[ifmpz]]
-
-
 class fmpz_mat(flint_mat[fmpz]):
     @overload
-    def __init__(self, val: ifmpz_mat, /) -> None: ...
+    def __init__(self, val: fmpz_mat | Sequence[Sequence[ifmpz]], /) -> None: ...
     @overload
     def __init__(self, m: int, n: int, /) -> None: ...
     @overload

src/flint/types/fmpz_mod_mat.pyi

@@ -0,0 +1,55 @@
+from __future__ import annotations
+
+from collections.abc import Sequence
+from typing import overload
+
+from flint.flint_base.flint_base import flint_mat
+from flint.types.fmpz import fmpz
+from flint.types.fmpz_mat import fmpz_mat
+from flint.types.fmpz_mod import fmpz_mod, fmpz_mod_ctx, ifmpz_mod
+from flint.types.fmpz_mod_poly import fmpz_mod_poly
+from flint.types.nmod_mat import nmod_mat
+
+class fmpz_mod_mat(flint_mat[fmpz_mod]):
+    @overload
+    def __init__(self, m: int, n: int, entries: Sequence[ifmpz_mod], ctx: fmpz_mod_ctx, /) -> None: ...
+    @overload
+    def __init__(self, m: int, n: int, ctx: fmpz_mod_ctx, /) -> None: ...
+    @overload
+    def __init__(self, val: Sequence[Sequence[ifmpz_mod]], ctx: fmpz_mod_ctx, /) -> None: ...
+    @overload
+    def __init__(self, val: fmpz_mod_mat | fmpz_mat | nmod_mat, ctx: fmpz_mod_ctx, /) -> None: ...
+    @overload
+    def __init__(self, val: fmpz_mod_mat | nmod_mat, /) -> None: ...
+
+    def nrows(self) -> int: ...
+    def ncols(self) -> int: ...
+    def modulus(self) -> fmpz: ...
+    def repr(self) -> str: ...
+    def entries(self) -> list[fmpz_mod]: ...
+    def __getitem__(self, index: tuple[int, int], /) -> fmpz_mod: ...
+    def __setitem__(self, index: tuple[int, int], value: ifmpz_mod, /) -> None: ...
+
+    def __bool__(self) -> bool: ...
+    def __eq__(self, other: object, /) -> bool: ...
+    def __ne__(self, other: object, /) -> bool: ...
+    def __pos__(self) -> fmpz_mod_mat: ...
+    def __neg__(self) -> fmpz_mod_mat: ...
+
+    def __add__(self, other: fmpz_mod_mat | fmpz_mat | nmod_mat, /) -> fmpz_mod_mat: ...
+    def __radd__(self, other: fmpz_mod_mat | fmpz_mat | nmod_mat, /) -> fmpz_mod_mat: ...
+    def __sub__(self, other: fmpz_mod_mat | fmpz_mat | nmod_mat, /) -> fmpz_mod_mat: ...
+    def __rsub__(self, other: fmpz_mod_mat | fmpz_mat | nmod_mat, /) -> fmpz_mod_mat: ...
+    def __mul__(self, other: fmpz_mod_mat | fmpz_mat | nmod_mat | ifmpz_mod, /) -> fmpz_mod_mat: ...
+    def __rmul__(self, other: fmpz_mod_mat | fmpz_mat | nmod_mat | ifmpz_mod, /) -> fmpz_mod_mat: ...
+    def __pow__(self, other: int, /) -> fmpz_mod_mat: ...
+    def __truediv__(self, other: ifmpz_mod, /) -> fmpz_mod_mat: ...
+
+    def inv(self) -> fmpz_mod_mat: ...
+    def det(self) -> fmpz_mod: ...
+    def charpoly(self) -> fmpz_mod_poly: ...
+    def minpoly(self) -> fmpz_mod_poly: ...
+    def transpose(self) -> fmpz_mod_mat: ...
+    def solve(self, rhs: fmpz_mod_mat | nmod_mat, /) -> fmpz_mod_mat: ...
+    def rank(self) -> int: ...
+    def rref(self, inplace: bool = False, /) -> tuple[fmpz_mod_mat, int]: ...

src/flint/types/meson.build

@@ -26,7 +26,7 @@ pyfiles = [
   'fmpz_mod.pyi',
   'fmpz_mod_poly.pyi',
   'fmpz_mod_mpoly.pyi',
-  # 'fmpz_mod_mat.pyi',
+  'fmpz_mod_mat.pyi',
 
   'fq_default.pyi',
   'fq_default_poly.pyi',

src/flint/types/nmod_mat.pyi

@@ -11,9 +11,6 @@ from flint.types.nmod_poly import nmod_poly
 
 _str = str
 
-inmod_mat = nmod_mat | fmpz_mat | Sequence[Sequence[inmod]]
-
-
 class nmod_mat(flint_mat[nmod]):
     @overload
     def __init__(self, val: nmod_mat, /) -> None: ...
@@ -47,19 +44,19 @@ class nmod_mat(flint_mat[nmod]):
 
     def __pos__(self) -> nmod_mat: ...
     def __neg__(self) -> nmod_mat: ...
-    def __add__(self, other: inmod_mat, /) -> nmod_mat: ...
-    def __radd__(self, other: inmod_mat, /) -> nmod_mat: ...
-    def __sub__(self, other: inmod_mat, /) -> nmod_mat: ...
-    def __rsub__(self, other: inmod_mat, /) -> nmod_mat: ...
-    def __mul__(self, other: inmod_mat | inmod, /) -> nmod_mat: ...
-    def __rmul__(self, other: inmod_mat | inmod, /) -> nmod_mat: ...
+    def __add__(self, other: nmod_mat | fmpz_mat, /) -> nmod_mat: ...
+    def __radd__(self, other: nmod_mat | fmpz_mat, /) -> nmod_mat: ...
+    def __sub__(self, other: nmod_mat | fmpz_mat, /) -> nmod_mat: ...
+    def __rsub__(self, other: nmod_mat | fmpz_mat, /) -> nmod_mat: ...
+    def __mul__(self, other: nmod_mat | fmpz_mat | inmod, /) -> nmod_mat: ...
+    def __rmul__(self, other: nmod_mat | fmpz_mat | inmod, /) -> nmod_mat: ...
     def __truediv__(self, other: inmod, /) -> nmod_mat: ...
     def __pow__(self, exponent: int, modulo: None = None, /) -> nmod_mat: ...
 
     def det(self) -> nmod: ...
     def inv(self) -> nmod_mat: ...
     def transpose(self) -> nmod_mat: ...
-    def solve(self, other: inmod_mat, /) -> nmod_mat: ...
+    def solve(self, other: nmod_mat | fmpz_mat, /) -> nmod_mat: ...
     def rref(self, inplace: bool = False, /) -> tuple[nmod_mat, int]: ...
     def rank(self) -> int: ...
     def nullspace(self) -> tuple[nmod_mat, int]: ...