typ: tweak from mypy --strict

Author: oscarbenjamin

src/flint/test/test_all.py

@@ -16,7 +16,7 @@
 
 ctx = flint.ctx
 
-def raises(f, exception):
+def raises(f, exception) -> bool:
     try:
         f()
     except exception:

src/flint/types/fmpq.pyi

@@ -10,11 +10,11 @@ ifmpq = int | fmpz | fmpq
 
 class fmpq(flint_scalar):
     @overload
-    def __init__(self): ...
+    def __init__(self) -> None: ...
     @overload
-    def __init__(self, arg: ifmpq | _str, /): ...
+    def __init__(self, arg: ifmpq | _str, /) -> None: ...
     @overload
-    def __init__(self, num: ifmpz, den: ifmpz, /): ...
+    def __init__(self, num: ifmpz, den: ifmpz, /) -> None: ...
 
     @property
     def p(self) -> fmpz: ...
@@ -53,6 +53,9 @@ class fmpq(flint_scalar):
     @overload
     def __round__(self) -> fmpz: ...
 
+    def __eq__(self, other: object, /) -> bool: ...
+    def __hash__(self) -> int: ...
+
     def __lt__(self, other: ifmpq, /) -> bool: ...
     def __le__(self, other: ifmpq, /) -> bool: ...
     def __gt__(self, other: ifmpq, /) -> bool: ...

src/flint/types/fmpq_poly.pyi

@@ -16,12 +16,12 @@ class fmpq_poly(flint_poly[fmpq]):
     """
 
     @overload
-    def __init__(self): ...
+    def __init__(self) -> None: ...
     @overload
-    def __init__(self, other: fmpq_poly | fmpz_poly | list[ifmpq] | ifmpq, /): ...
+    def __init__(self, other: fmpq_poly | fmpz_poly | list[ifmpq] | ifmpq, /) -> None: ...
     @overload
     def __init__(self, other: fmpq_poly | fmpz_poly | list[ifmpq] | ifmpq,
-                       den: ifmpz, /): ...
+                       den: ifmpz, /) -> None: ...
 
     def __len__(self) -> int: ...
     def length(self) -> int: ...

src/flint/types/fmpq_series.pyi

@@ -1,3 +1,4 @@
+from typing import Any
 from flint.flint_base.flint_base import flint_series
 from flint.types.fmpz import fmpz, ifmpz
 from flint.types.fmpq import fmpq, ifmpq
@@ -35,8 +36,8 @@ class fmpq_series(flint_series[fmpq]):
     def __getitem__(self, i: int, /) -> fmpq: ...
     def __setitem__(self, i: int, x: ifmpq, /) -> None: ...
 
-    def repr(self, **kwargs) -> str: ...
-    def str(self, **kwargs) -> str: ...
+    def repr(self, **kwargs: Any) -> str: ...
+    def str(self, **kwargs: Any) -> str: ...
 
     def __pos__(self) -> fmpq_series: ...
     def __neg__(self) -> fmpq_series: ...

src/flint/types/fmpz.pyi

@@ -32,6 +32,9 @@ class fmpz(flint_scalar):
     def __trunc__(self) -> fmpz: ...
     def __round__(self, ndigits: int | None = None) -> fmpz: ...
 
+    def __eq__(self, other: object, /) -> bool: ...
+    def __hash__(self) -> int: ...
+
     def __lt__(self, other: ifmpz, /) -> bool: ...
     def __le__(self, other: ifmpz, /) -> bool: ...
     def __gt__(self, other: ifmpz, /) -> bool: ...
@@ -72,7 +75,7 @@ class fmpz(flint_scalar):
     def gcd(self, other: ifmpz, /) -> fmpz: ...
     def lcm(self, other: ifmpz, /) -> fmpz: ...
     def factor(self, trial_limit: int | None = None) -> list[tuple[fmpz, int]]: ...
-    def factor_smooth(self, bits: int = 15, proved: int = -1): ...
+    def factor_smooth(self, bits: int = 15, proved: int = -1) -> list[tuple[fmpz, int]]: ...
 
     def is_prime(self) -> bool: ...
     def is_probable_prime(self) -> bool: ...

src/flint/types/fmpz_mod_mpoly.pyi

@@ -25,7 +25,7 @@ class fmpz_mod_mpoly_ctx(flint_mod_mpoly_context[fmpz_mod_mpoly, fmpz_mod, ifmpz
     def ordering(self) -> Ordering: ...
     def modulus(self) -> int: ...
     def gen(self, i: int, /) -> fmpz_mod_mpoly: ...
-    def constant(self, z) -> fmpz_mod_mpoly: ...
+    def constant(self, z: ifmpz_mod) -> fmpz_mod_mpoly: ...
     def from_dict(self, d: Mapping[tuple[int, ...], ifmpz_mod]) -> fmpz_mod_mpoly: ...
 
 

src/flint/types/fmpz_poly.pyi

@@ -15,16 +15,16 @@ class fmpz_poly(flint_poly[fmpz]):
     """
 
     @overload
-    def __init__(self): ...
+    def __init__(self) -> None: ...
     @overload
-    def __init__(self, arg: list[ifmpz] | fmpz | fmpz_poly, /): ...
+    def __init__(self, arg: list[ifmpz] | fmpz | fmpz_poly, /) -> None: ...
 
     def __len__(self) -> int: ...
     def length(self) -> int: ...
     def degree(self) -> int: ...
 
     def __getitem__(self, i: int, /) -> fmpz: ...
-    def __setitem__(self, i: int, x: ifmpz, /): ...
+    def __setitem__(self, i: int, x: ifmpz, /) -> None: ...
 
     def repr(self) -> str: ...
 
@@ -98,7 +98,7 @@ class fmpz_poly(flint_poly[fmpz]):
     def chebyshev_t(n: int, /) -> fmpz_poly: ...
 
     @staticmethod
-    def chebyshev_u(n) -> fmpz_poly: ...
+    def chebyshev_u(n: int, /) -> fmpz_poly: ...
 
     @staticmethod
     def swinnerton_dyer(n: int, /, use_arb: bool = True) -> fmpz_poly: ...

src/flint/types/fmpz_series.pyi

@@ -25,7 +25,7 @@ class fmpz_series(flint_series[fmpz]):
     def __setitem__(self, i: int, x: ifmpz, /) -> None: ...
 
     def repr(self, **kwargs: Any) -> str: ...
-    def str(self, **kwargs) -> str: ...
+    def str(self, **kwargs: Any) -> str: ...
 
     def __pos__(self) -> fmpz_series: ...
     def __neg__(self) -> fmpz_series: ...