typ: annotations for truncate, left/right_shift

Author: oscarbenjamin

src/flint/types/fmpq_poly.pyi

@@ -68,6 +68,10 @@ class fmpq_poly(flint_poly[fmpq]):
     def __rdivmod__(self, other: ifmpq_poly, /) -> tuple[fmpq_poly, fmpq_poly]: ...
     def __pow__(self, exp: int | ifmpz, /) -> fmpq_poly: ...
 
+    def left_shift(self, n: int, /) -> fmpq_poly: ...
+    def right_shift(self, n: int, /) -> fmpq_poly: ...
+    def truncate(self, n: int, /) -> fmpq_poly: ...
+
     def gcd(self, other: ifmpq_poly, /) -> fmpq_poly: ...
     def resultant(self, other: ifmpq_poly, /) -> fmpq: ...
     def xgcd(self, other: ifmpq_poly, /) -> tuple[fmpq_poly, fmpq_poly, fmpq_poly]: ...

src/flint/types/fmpz_poly.pyi

@@ -55,6 +55,11 @@ class fmpz_poly(flint_poly[fmpz]):
     def __divmod__(self, other: ifmpz_poly, /) -> tuple[fmpz_poly, fmpz_poly]: ...
     def __rdivmod__(self, other: ifmpz, /) -> tuple[fmpz_poly, fmpz_poly]: ...
     def __pow__(self, other: int, /) -> fmpz_poly: ...
+
+    def left_shift(self, n: int, /) -> fmpz_poly: ...
+    def right_shift(self, n: int, /) -> fmpz_poly: ...
+    def truncate(self, n: int, /) -> fmpz_poly: ...
+
     def gcd(self, other: ifmpz_poly, /) -> fmpz_poly: ...
     def content(self) -> fmpz: ...
     def resultant(self, other: ifmpz_poly, /) -> fmpz: ...

src/flint/typing.py

@@ -119,6 +119,11 @@ def __rmod__(self, other: _Tscalar | ifmpz, /) -> Self: ...
     def __divmod__(self, other: _Tscalar | ifmpz | Self, /) -> tuple[Self, Self]: ...
     def __rdivmod__(self, other: _Tscalar | ifmpz, /) -> tuple[Self, Self]: ...
     def __pow__(self, other: int, /) -> Self: ...
+
+    def left_shift(self, other: int, /) -> Self: ...
+    def right_shift(self, other: int, /) -> Self: ...
+    def truncate(self, n: int, /) -> Self: ...
+
     def is_zero(self) -> bool: ...
     def is_one(self) -> bool: ...
     def is_constant(self) -> bool: ...