fmpz_mat: add type stubs

Author: oscarbenjamin

src/flint/test/test_all.py

@@ -1099,6 +1099,9 @@ def test_fmpq_mat():
     assert (Q(1,1,[1]) != 1) is True
     assert (1 == Q(1,1,[1])) is False
     assert (1 != Q(1,1,[1])) is True
+    marker = object()
+    assert (Q(1,1,[1]) == marker) is False
+    assert (Q(1,1,[1]) != marker) is True
     assert Q(1,2,[3,4]) * 2 == Q(1,2,[6,8])
     assert Q(1,2,[3,4]) * flint.fmpq(1,3) == Q(1,2,[1,flint.fmpq(4,3)])
     assert Q(1,2,[3,4]) * flint.fmpq(5,3) == Q(1,2,[5,flint.fmpq(20,3)])
@@ -1110,6 +1113,7 @@ def test_fmpq_mat():
     assert M ** 1 == M
     assert M ** 2 == Q([[7,10],[15,22]])
     assert M ** 12 == Q([[138067399, 201223170],[301834755, 439902154]])
+    assert raises(lambda: pow(M, 2, 3), TypeError)
     M = Q([[1,2],[2,4]])
     assert raises(lambda: M ** -1, ZeroDivisionError)
     assert Q(1,2,[3,4]) / 2 == Q(1,2,[flint.fmpq(3,2),2])
@@ -1169,6 +1173,9 @@ def set_bad(i):
     X = Q([[1,2],[3,4]])
     B = A*X
     assert A.solve(B) == X
+    I25 = Q([[1 if i == j else 0 for j in range(25)] for i in range(25)])
+    X25 = Q.hilbert(25, 2)
+    assert I25.solve(X25) == X25
     for algorithm in None, "fflu", "dixon":
         assert A.solve(B, algorithm=algorithm) == X
     assert raises(lambda: A.solve(B, algorithm="invalid"), ValueError)

src/flint/types/fmpq_mat.pyi

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+from collections.abc import Iterable, Iterator, Sequence
+from typing import Literal, overload
+
+from flint.flint_base.flint_base import flint_mat
+from flint.types.fmpq import fmpq, ifmpq
+from flint.types.fmpq_poly import fmpq_poly
+from flint.types.fmpz import fmpz
+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: ...
+    @overload
+    def __init__(self, m: int, n: int, /) -> None: ...
+    @overload
+    def __init__(self, m: int, n: int, entries: Iterable[ifmpq], /) -> None: ...
+
+    def __bool__(self) -> bool: ...
+    def __eq__(self, other: object, /) -> bool: ...
+    def __ne__(self, other: object, /) -> bool: ...
+
+    def __getitem__(self, index: tuple[int, int], /) -> fmpq: ...
+    def __setitem__(self, index: tuple[int, int], value: ifmpq, /) -> None: ...
+    def __iter__(self) -> Iterator[fmpq]: ...
+    def entries(self) -> list[fmpq]: ...
+    def table(self) -> list[list[fmpq]]: ...
+    def tolist(self) -> list[list[fmpq]]: ...
+
+    def det(self) -> fmpq: ...
+    def __pos__(self) -> fmpq_mat: ...
+    def __neg__(self) -> fmpq_mat: ...
+    def __add__(self, other: fmpq_mat | fmpz_mat, /) -> fmpq_mat: ...
+    def __sub__(self, other: fmpq_mat | fmpz_mat, /) -> fmpq_mat: ...
+    def __mul__(self, other: fmpq_mat | fmpz_mat | ifmpq, /) -> fmpq_mat: ...
+    def __rmul__(self, other: fmpz_mat | ifmpq, /) -> fmpq_mat: ...
+    def __truediv__(self, other: ifmpq, /) -> fmpq_mat: ...
+
+    def inv(self) -> fmpq_mat: ...
+    def transpose(self) -> fmpq_mat: ...
+    def solve(
+        self,
+        other: fmpq_mat | fmpz_mat,
+        algorithm: Literal["fflu", "dixon"] | None = None,
+        /,
+    ) -> fmpq_mat: ...
+    def rref(self, inplace: bool = False, /) -> tuple[fmpq_mat, int]: ...
+    def rank(self) -> int: ...
+    @classmethod
+    def hilbert(cls, n: int, m: int, /) -> fmpq_mat: ...
+    def numer_denom(self) -> tuple[fmpz_mat, fmpz]: ...
+    def charpoly(self) -> fmpq_poly: ...
+    def minpoly(self) -> fmpq_poly: ...
+    def __pow__(self, exponent: int, modulo: None = None, /) -> fmpq_mat: ...
+
+    def str(self, *args: object, **kwargs: object) -> _str: ...
+    def repr(self) -> _str: ...
+    def __str__(self) -> _str: ...
+    def __repr__(self) -> _str: ...

src/flint/types/fmpq_mat.pyx

@@ -100,9 +100,6 @@ cdef class fmpq_mat(flint_mat):
         cdef bint r
         if op != 2 and op != 3:
             raise TypeError("matrices cannot be ordered")
-        s = any_as_fmpq_mat(s)
-        if t is NotImplemented:
-            return s
         t = any_as_fmpq_mat(t)
         if t is NotImplemented:
             return t
@@ -269,9 +266,6 @@ cdef class fmpq_mat(flint_mat):
     def __truediv__(s, t):
         return fmpq_mat._div_(s, t)
 
-    def __div__(s, t):
-        return fmpq_mat._div_(s, t)
-
     def inv(self):
         """
         Returns the inverse matrix of *self*.

src/flint/types/meson.build

@@ -14,7 +14,7 @@ pyfiles = [
   'fmpq_poly.pyi',
   'fmpq_series.pyi',
   'fmpq_mpoly.pyi',
-  # 'fmpq_mat.pyi',
+  'fmpq_mat.pyi',
   'fmpq_vec.pyi',
 
   'nmod.pyi',