snappy 3.0.3__cp38-cp38-macosx_11_0_arm64.whl → 3.2__cp38-cp38-macosx_11_0_arm64.whl

snappy/sage_helper.py

@@ -9,15 +9,10 @@ than the usual ">>>".
 """
 
 try:
-    import sage.all
+    import sage.structure.sage_object
     _within_sage = True
-except:
+except ImportError:
     _within_sage = False
-    import decorator
-
-import sys, doctest, re, types
-
-from .numeric_output_checker import NumericOutputChecker
 
 class SageNotAvailable(Exception):
     pass
@@ -27,23 +22,86 @@ if _within_sage:
         function._sage_method = True
         return function
 
-
-    try: # Sage >= 9.3, see https://trac.sagemath.org/ticket/24483
+    try:
+        # Monolithic Sage library
+        from sage.all import RealField, RealDoubleElement, gcd, xgcd, prod, powerset
+        from sage.all import MatrixSpace, matrix, vector, ZZ
+        from sage.all import Integer, Rational, QQ, RR, CC
+        from sage.all import sqrt
+        from sage.all import I, Infinity
+        from sage.all import arccosh
+        from sage.all import RIF, CIF
+        from sage.all import (cached_method, real_part, imag_part, round, ceil, floor, log,
+                              CDF, ComplexDoubleField, ComplexField, CyclotomicField, NumberField, PolynomialRing, identity_matrix)
+        from sage.all import FiniteField as GF
+        from sage.all import VectorSpace, ChainComplex
+        from sage.all import ComplexBallField, exp, sin, block_matrix, prime_range, det
+        from sage.all import LaurentPolynomialRing, AbelianGroup, GroupAlgebra
+        from sage.all import is_prime
+    except ImportError:
+        # Modularized Sage library
+        from sage.algebras.group_algebra import GroupAlgebra
+        from sage.arith.misc import gcd, xgcd, is_prime
+        from sage.combinat.subset import powerset
+        from sage.functions.hyperbolic import arccosh
+        from sage.functions.log import exp
+        from sage.functions.other import (real as real_part,
+                                          imag as imag_part,
+                                          ceil,
+                                          floor)
+        from sage.functions.trig import sin
+        from sage.groups.abelian_gps.abelian_group import AbelianGroup
+        from sage.homology.chain_complex import ChainComplex
+        from sage.matrix.constructor import Matrix as matrix
+        from sage.matrix.matrix_space import MatrixSpace
+        from sage.matrix.special import block_matrix, identity_matrix
+        from sage.misc.cachefunc import cached_method
+        from sage.misc.functional import det, log, round, sqrt
+        from sage.misc.misc_c import prod
+        from sage.modules.free_module import VectorSpace
+        from sage.modules.free_module_element import free_module_element as vector
+        from sage.rings.cc import CC
+        from sage.rings.cif import CIF
+        from sage.rings.complex_arb import ComplexBallField
+        from sage.rings.complex_double import CDF, ComplexDoubleField
+        from sage.rings.complex_mpfr import ComplexField
+        from sage.rings.fast_arith import prime_range
+        from sage.rings.finite_rings.finite_field_constructor import FiniteField as GF
+        from sage.rings.imaginary_unit import I
+        from sage.rings.infinity import Infinity
+        from sage.rings.integer import Integer
+        from sage.rings.integer_ring import ZZ
+        from sage.rings.polynomial.laurent_polynomial_ring import LaurentPolynomialRing
+        from sage.rings.number_field.number_field import CyclotomicField, NumberField
+        from sage.rings.polynomial.polynomial_ring_constructor import PolynomialRing
+        from sage.rings.rational import Rational
+        from sage.rings.rational_field import QQ
+        from sage.rings.real_double import RealDoubleElement
+        from sage.rings.real_mpfi import RIF
+        from sage.rings.real_mpfr import RealField, RealNumber, RR
+
+    from sage.rings.complex_interval_field import ComplexIntervalField
+    from sage.rings.real_mpfi import is_RealIntervalFieldElement, RealIntervalField
+    from sage.rings.real_mpfr import RealNumber, RealField_class
+    from sage.structure.sage_object import SageObject
+
+    try:  # Sage >= 9.3, see https://trac.sagemath.org/ticket/24483
         from sage.rings.complex_mpfr import (ComplexField,
                                              ComplexField_class,
                                              create_ComplexNumber)
-    except ModuleNotFoundError:
+    except ImportError:
         from sage.rings.complex_field import ComplexField, ComplexField_class
         from sage.rings.complex_number import create_ComplexNumber
 
 else:
+    import decorator
+
     def _sage_method(function, *args, **kw):
         raise SageNotAvailable('Sorry, this feature requires using SnapPy inside Sage.')
 
     def sage_method(function):
         return decorator.decorator(_sage_method, function)
 
-
 # Not currently used, but could be exploited by an interpreter to hide
 # sage_methods when in plain Python.
 
@@ -57,115 +115,3 @@ def sage_methods(obj):
         except AttributeError:
             pass
     return ans
-
-# Used for doctesting
-_gui_status = {}
-
-try:
-    from snappy.gui import Tk_
-    _gui_status['tk'] = True
-except ImportError:
-    _gui_status['tk'] = False
-if _gui_status['tk']:
-    try:
-        import snappy.CyOpenGL
-        _gui_status['cyopengl'] = True
-    except:
-        _gui_status['cyopengl'] = False
-else:
-    _gui_status['cyopengl'] = False
-_gui_status['fake_root'] = False
-
-def cyopengl_works():
-    if not _gui_status['cyopengl']:
-        return False
-    # if we are running the tests from the snappy app the default root will
-    # already exist -- it will be the tkterminal window.  Otherwise, we open
-    # a root window here to serve as the master of all of the GUI windows
-    # which get created during testing.
-    if _gui_status['tk'] and not Tk_._default_root:
-        try:
-            root = Tk_.Tk()
-            if sys.platform not in ('linux', 'linux2'):
-                root.withdraw()
-        except:
-            # tkinter loads OK but is not able to get a display.
-            _gui_status['tk'] = _gui_status['cyopengl'] = False
-    return _gui_status['cyopengl']
-
-def tk_root():
-    if _gui_status['tk']:
-        return Tk_._default_root
-    else:
-        return None
-
-def root_is_fake():
-    return _gui_status['fake_root']
-
-class DocTestParser(doctest.DocTestParser):
-    _use_cyopengl_initialized = False
-    _use_cyopengl = False
-    use_modernopengl = True
-    use_sage = False
-
-    def parse(self, string, name='<string>'):
-        string = re.subn(
-            r'#doctest: \+CYOPENGL',
-            '' if DocTestParser._use_cyopengl else '#doctest: +SKIP',
-            string)[0]
-
-        string = re.subn(
-            r'#doctest: \+CYMODERNOPENGL',
-            (''
-             if (DocTestParser._use_cyopengl and
-                 DocTestParser.use_modernopengl)
-             else '#doctest: +SKIP'),
-            string)[0]
-        
-        if DocTestParser.use_sage:
-            string = re.subn(r'(\n\s*)sage:|(\A\s*)sage:',
-                             r'\g<1>>>>',
-                             string)[0]
-        return doctest.DocTestParser.parse(self, string, name)
-
-DocTestParser.use_sage = _within_sage
-
-if _within_sage:
-    globs = {'PSL':sage.all.PSL, 'BraidGroup':sage.all.BraidGroup}
-else:
-    globs = { }
-
-def print_results(module, results):
-    root = tk_root()
-    # Platform specific hacks to make running the tests work.
-    if root and (sys.version_info.major < 3 or not root_is_fake()):
-        if sys.platform in ('linux', 'linux2'):
-            root.deiconify()
-            root.update_idletasks()
-        else:
-            root.update()
-    print(module.__name__ + ':')
-    print('   %s failures out of %s tests.' %  (results.failed, results.attempted))
-
-def doctest_modules(modules, verbose=False, print_info=True, extraglobs=dict()):
-    finder = doctest.DocTestFinder(parser=DocTestParser())
-    #full_extraglobals = dict(globs.items() + extraglobs.items())
-    full_extraglobals = globs.copy()
-    full_extraglobals.update(extraglobs)
-    failed, attempted = 0, 0
-    for module in modules:
-        if isinstance(module, types.ModuleType):
-            runner = doctest.DocTestRunner(checker = NumericOutputChecker(), verbose=verbose)
-            for test in finder.find(module, extraglobs=full_extraglobals):
-                runner.run(test)
-            result = runner.summarize()
-        else:
-            result = module(verbose=verbose)
-        failed += result.failed
-        attempted += result.attempted
-        if print_info:
-            print_results(module, result)
-
-    if print_info:
-        print('\nAll doctests:\n   %s failures out of %s tests.' % (failed, attempted))
-    return doctest.TestResults(failed, attempted)