Add type hints for rings.infinity

src/sage/geometry/cone.py

@@ -201,9 +201,11 @@
 # (at your option) any later version.
 #                  https://www.gnu.org/licenses/
 # ****************************************************************************
+from __future__ import annotations
 
 from collections.abc import Hashable, Iterable, Container
 from copy import copy
+from typing import TYPE_CHECKING, Literal
 from warnings import warn
 
 from sage.misc.lazy_import import lazy_import
@@ -241,6 +243,9 @@
 lazy_import('ppl', ['ray', 'point'], as_=['PPL_ray', 'PPL_point'],
                     feature=PythonModule("ppl", spkg='pplpy', type='standard'))
 
+if TYPE_CHECKING:
+    from sage.misc.sage_input import SageInputBuilder, SageInputExpression
+
 
 def Cone(rays, lattice=None, check=True, normalize=True):
     r"""
@@ -1481,7 +1486,7 @@ def __init__(self, rays=None, lattice=None,
         if PPL is not None:
             self._PPL_C_Polyhedron = PPL
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         """
         Return Sage command to reconstruct ``self``.
 

src/sage/geometry/fan.py

@@ -233,9 +233,11 @@
 # (at your option) any later version.
 #                  https://www.gnu.org/licenses/
 # ****************************************************************************
+from __future__ import annotations
 
 from collections.abc import Callable, Container
 from copy import copy
+from typing import TYPE_CHECKING, Literal
 from warnings import warn
 
 import sage.geometry.abc
@@ -262,6 +264,9 @@
 from sage.rings.integer_ring import ZZ
 from sage.rings.rational_field import QQ
 
+if TYPE_CHECKING:
+    from sage.misc.sage_input import SageInputBuilder, SageInputExpression
+
 
 def Fan(cones, rays=None, lattice=None, check=True, normalize=True,
         is_complete=None, virtual_rays=None, discard_faces=False,
@@ -1187,7 +1192,7 @@ def __init__(self, cones, rays, lattice,
         if virtual_rays is not None:
             self._virtual_rays = PointCollection(virtual_rays, self.lattice())
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         """
         Return Sage command to reconstruct ``self``.
 

src/sage/geometry/lattice_polytope.py

@@ -120,42 +120,43 @@
 # (at your option) any later version.
 #                  https://www.gnu.org/licenses/
 # ****************************************************************************
+from __future__ import annotations
+
+import os
+import shlex
 from collections.abc import Hashable
 from copyreg import constructor as copyreg_constructor
 from functools import reduce
 from io import IOBase, StringIO
-from subprocess import Popen, PIPE
-from warnings import warn
-import os
-import shlex
+from subprocess import PIPE, Popen
+from typing import Literal
 
+import sage.geometry.abc
 from sage.arith.misc import GCD as gcd
 from sage.categories.monoids import Monoids
 from sage.features import PythonModule
-from sage.features.palp import PalpExecutable
 from sage.features.databases import DatabaseReflexivePolytopes
+from sage.features.palp import PalpExecutable
 from sage.geometry.cone import _ambient_space_point, integral_length
-from sage.geometry.point_collection import (PointCollection,
-                                            read_palp_point_collection)
-from sage.geometry.toric_lattice import ToricLattice, ToricLattice_generic
 from sage.geometry.convex_set import ConvexSet_compact
+from sage.geometry.point_collection import PointCollection, read_palp_point_collection
+from sage.geometry.toric_lattice import ToricLattice, ToricLattice_generic
 from sage.matrix.constructor import matrix
 from sage.misc.cachefunc import cached_method
 from sage.misc.flatten import flatten
 from sage.misc.lazy_import import lazy_import
+from sage.misc.sage_input import SageInputBuilder, SageInputExpression
 from sage.misc.temporary_file import tmp_filename
 from sage.modules.free_module_element import vector
 from sage.rings.integer import Integer
 from sage.rings.integer_ring import ZZ
 from sage.rings.rational_field import QQ
-from sage.structure.element import Matrix, Element
-from sage.structure.richcmp import richcmp_method, richcmp
-from sage.structure.parent import Parent
+from sage.structure.element import Element, Matrix
+from sage.structure.parent import Parent, Set_generic
+from sage.structure.richcmp import richcmp, richcmp_method
 from sage.structure.sage_object import SageObject
 from sage.structure.sequence import Sequence
 from sage.structure.unique_representation import UniqueRepresentation
-import sage.geometry.abc
-
 
 lazy_import("sage.combinat.posets.posets", 'FinitePoset')
 lazy_import("sage.geometry.hasse_diagram", 'lattice_from_incidences')
@@ -170,6 +171,34 @@
             feature=PythonModule("ppl", spkg='pplpy', type='standard'))
 
 
+class SetOfAllLatticePolytopesClass(Set_generic):
+    def _repr_(self) -> str:
+        r"""
+        Return a string representation.
+
+        TESTS::
+
+            sage: lattice_polytope.SetOfAllLatticePolytopesClass()._repr_()
+            'Set of all Lattice Polytopes'
+        """
+        return "Set of all Lattice Polytopes"
+
+    def __call__(self, x):
+        r"""
+        TESTS::
+
+            sage: o = lattice_polytope.cross_polytope(3)
+            sage: lattice_polytope.SetOfAllLatticePolytopesClass().__call__(o)
+            3-d reflexive polytope in 3-d lattice M
+        """
+        if isinstance(x, LatticePolytopeClass):
+            return x
+        raise TypeError
+
+
+SetOfAllLatticePolytopes = SetOfAllLatticePolytopesClass()
+
+
 def LatticePolytope(data, compute_vertices=True, n=0, lattice=None):
     r"""
     Construct a lattice polytope.
@@ -517,7 +546,7 @@
             self._vertices = ambient.vertices(self._ambient_vertex_indices)
         Element.__init__(self, parent)
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         """
         Return Sage command to reconstruct ``self``.
 
@@ -4469,7 +4498,7 @@
             pass
         return result
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool) -> SageInputExpression:
         """
         Return Sage command to reconstruct ``self``.
 

src/sage/geometry/point_collection.pyx

@@ -83,6 +83,8 @@ from sage.geometry.toric_lattice import ToricLattice
 from sage.matrix.constructor import matrix
 from sage.misc.latex import latex
 
+from sage.misc.sage_input import SageInputBuilder, SageInputExpression
+
 
 def is_PointCollection(x):
     r"""
@@ -172,7 +174,7 @@ cdef class PointCollection(SageObject):
         self._points = tuple(points)
         self._module = self._points[0].parent() if module is None else module
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         r"""
         Return Sage command to reconstruct ``self``.
 

src/sage/geometry/polyhedron/base0.py

@@ -29,12 +29,17 @@
 # (at your option) any later version.
 #                  https://www.gnu.org/licenses/
 # ****************************************************************************
+from __future__ import annotations
 
+from typing import TYPE_CHECKING, Literal
 from sage.misc.cachefunc import cached_method
 from sage.misc.abstract_method import abstract_method
 from sage.structure.element import Element
 import sage.geometry.abc
 
+if TYPE_CHECKING:
+    from sage.misc.sage_input import SageInputBuilder, SageInputExpression
+
 
 class Polyhedron_base0(Element, sage.geometry.abc.Polyhedron):
     """
@@ -296,7 +301,7 @@ def _delete(self):
         """
         self.parent().recycle(self)
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         """
         Return Sage command to reconstruct ``self``.
 

src/sage/geometry/toric_lattice.py

@@ -144,7 +144,9 @@
 #
 #                  https://www.gnu.org/licenses/
 # ****************************************************************************
+from __future__ import annotations
 
+from typing import TYPE_CHECKING, Literal
 from sage.geometry.toric_lattice_element import ToricLatticeElement
 from sage.misc.lazy_import import lazy_import
 lazy_import('sage.geometry.toric_plotter', 'ToricPlotter')
@@ -162,6 +164,9 @@
 from sage.rings.rational_field import QQ
 from sage.structure.factory import UniqueFactory
 
+if TYPE_CHECKING:
+    from sage.misc.sage_input import SageInputBuilder, SageInputExpression
+
 
 class ToricLatticeFactory(UniqueFactory):
     r"""
@@ -825,7 +830,7 @@ def __init__(self, rank, name, dual_name, latex_name, latex_dual_name):
         self._latex_name = latex_name
         self._latex_dual_name = latex_dual_name
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         r"""
         Return Sage command to reconstruct ``self``.
 

src/sage/matrix/matrix1.pyx

@@ -24,6 +24,7 @@ from cpython.sequence cimport PySequence_Fast
 import sage.modules.free_module
 from sage.structure.coerce cimport coercion_model
 
+from sage.misc.sage_input import SageInputBuilder, SageInputExpression
 
 cdef class Matrix(Matrix0):
     ###################################################
@@ -622,7 +623,7 @@ cdef class Matrix(Matrix0):
             matrix._sage_object = self
         return matrix
 
-    def _sage_input_(self, sib, coerce):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         r"""
         Produce an expression which will reproduce this value when evaluated.
 

src/sage/misc/explain_pickle.py

@@ -152,13 +152,14 @@
 # (at your option) any later version.
 #                  http://www.gnu.org/licenses/
 # *****************************************************************************
-
+from __future__ import annotations
 
 import pickletools
 import re
 import sys
 import types
 
+from typing import TYPE_CHECKING, Literal
 import zlib as comp
 import bz2 as comp_other
 
@@ -169,6 +170,9 @@
 from sage.misc.persist import (unpickle_override, unpickle_global, dumps,
                                register_unpickle_override, SageUnpickler)
 
+if TYPE_CHECKING:
+    from sage.misc.sage_input import SageInputBuilder, SageInputExpression
+
 
 # Python 3 does not have a "ClassType". Instead, we ensure that
 # isinstance(foo, ClassType) will always return False.
@@ -358,7 +362,7 @@
         self.expression = expression
         self.immutable = False
 
-    def _sage_input_(self, sib, coerced):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         r"""
         Extracts the expression from a PickleObject, and sets the immutable
         flag.

src/sage/misc/sage_input.py

@@ -172,6 +172,9 @@
 #                  https://www.gnu.org/licenses/
 # ****************************************************************************
 
+from __future__ import annotations
+from typing import Literal
+
 from sage.misc.lazy_import import lazy_import
 
 lazy_import('sage.rings.real_mpfi', 'RealIntervalFieldElement')
@@ -342,7 +345,7 @@ def __init__(self, allow_locals=False, preparse=True):
         self._next_local = 1
         self._locals = {}
 
-    def __call__(self, x, coerced=False):
+    def __call__(self, x, coerced: bool | Literal[2] = False):
         r"""
         Try to convert an arbitrary value ``x`` into a
         :class:`SageInputExpression` (an SIE).
@@ -460,6 +463,7 @@ def __call__(self, x, coerced=False):
             return x
 
         if hasattr(x, '_sage_input_'):
+            #print(coerced, type(coerced))
             return x._sage_input_(self, coerced)
 
         if x is None:
@@ -592,7 +596,7 @@ def float_str(self, n):
         """
         return SIE_literal_stringrep(self, n)
 
-    def name(self, n):
+    def name(self, n) -> SageInputExpression:
         r"""
         Given a string representing a Python name,
         produces a :class:`SageInputExpression` for that name.

src/sage/modules/free_module_element.pyx

@@ -129,6 +129,7 @@ from sage.rings.abc import RealDoubleField, ComplexDoubleField
 
 from sage.rings.integer cimport Integer, smallInteger
 from sage.arith.numerical_approx cimport digits_to_bits
+from sage.misc.sage_input import SageInputBuilder, SageInputExpression
 
 # For the norm function, we cache Sage integers 1 and 2
 __one__ = smallInteger(1)
@@ -1259,7 +1260,7 @@ cdef class FreeModuleElement(Vector):   # abstract base class
             return self
         return self.change_ring(R)
 
-    def _sage_input_(self, sib, coerce):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         r"""
         Produce an expression which will reproduce this value when evaluated.
 

src/sage/rings/complex_interval.pyx

@@ -64,6 +64,7 @@ from sage.libs.flint.fmpz cimport *
 from sage.libs.mpfr cimport MPFR_RNDU
 from sage.arith.constants cimport LOG_TEN_TWO_PLUS_EPSILON
 
+from sage.misc.sage_input import SageInputBuilder, SageInputExpression
 from sage.structure.element cimport FieldElement
 from sage.structure.parent cimport Parent
 from sage.rings.complex_mpfr cimport ComplexNumber
@@ -1002,7 +1003,7 @@ cdef class ComplexIntervalFieldElement(FieldElement):
         """
         raise TypeError
 
-    def _sage_input_(self, sib, coerce):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         r"""
         Produce an expression which will reproduce this value when evaluated.
 

src/sage/rings/complex_interval_field.py

@@ -34,8 +34,9 @@
 # (at your option) any later version.
 #                  https://www.gnu.org/licenses/
 # ****************************************************************************
+from __future__ import annotations
 
-
+from typing import TYPE_CHECKING, Literal
 import weakref
 
 import sage.rings.abc
@@ -48,6 +49,9 @@
 from sage.rings.ring import Field
 from sage.structure.parent import Parent
 
+if TYPE_CHECKING:
+    from sage.misc.sage_input import SageInputBuilder, SageInputExpression
+
 cache = {}
 
 
@@ -295,7 +299,7 @@ def _magma_init_(self, magma):
         """
         return "ComplexField(%s : Bits := true)" % self.prec()
 
-    def _sage_input_(self, sib, coerce):
+    def _sage_input_(self, sib: SageInputBuilder, coerced: bool | Literal[2]) -> SageInputExpression:
         r"""
         Produce an expression which will reproduce this value when evaluated.