Merge pull request #255 from Tom-Hubrecht/gr-ctx-methods

gr: Add context methods
flintlib · Jan 31, 2025 · e50cda6 · e50cda6
2 parents 7786fc0 + 8b7f0c6
commit e50cda6
Show file tree

Hide file tree

Showing 4 changed files with 1,096 additions and 24 deletions.
diff --git a/src/flint/flintlib/types/gr.pxd b/src/flint/flintlib/types/gr.pxd
@@ -5,6 +5,26 @@ from flint.flintlib.types.flint cimport (
     flint_bitcnt_t,
 )
 
+cdef extern from *:
+    """
+    /*
+     * The following functions were introduced in FLINT 3.2.0
+     */
+
+    #if __FLINT_RELEASE < 30200
+    #define gr_min(res, x, y, ctx) GR_UNABLE
+    #define gr_max(res, x, y, ctx) GR_UNABLE
+    #define gr_le(x, y, ctx) T_UNKNOWN
+    #define gr_lt(x, y, ctx) T_UNKNOWN
+    #define gr_ge(x, y, ctx) T_UNKNOWN
+    #define gr_gt(x, y, ctx) T_UNKNOWN
+    #define gr_abs_le(x, y, ctx) T_UNKNOWN
+    #define gr_abs_lt(x, y, ctx) T_UNKNOWN
+    #define gr_abs_ge(x, y, ctx) T_UNKNOWN
+    #define gr_abs_gt(x, y, ctx) T_UNKNOWN
+    #endif
+    """
+
 
 cdef extern from "flint/gr.h":
 

diff --git a/src/flint/test/test_docstrings.py b/src/flint/test/test_docstrings.py
@@ -5,7 +5,12 @@
 
 import flint
 
-dunder_test_regex = re.compile(r'^(.*?)__test__\..*?\.(.*) \(line (\d+)\)$')
+dunder_test_regex = re.compile(r'^(.*?)__test__\.(.*?\.)(.*) \(line (\d+)\)$')
+
+test_flint_at_least = {
+    "flint.types._gr.gr_ctx.gens": 30100,
+    "flint.types._gr.gr_ctx.neg": 30100,
+}
 
 
 def find_doctests(module):
@@ -20,9 +25,14 @@ def find_doctests(module):
                 m = dunder_test_regex.match(test.name)
                 if m is not None:
                     groups = m.groups()
-                    test.name = groups[0] + groups[1]
-                    test.lineno = int(groups[2])
-                    res.append(test)
+                    test.name = groups[0] + groups[2]
+                    test.lineno = int(groups[3])
+
+                    if (
+                        test_flint_at_least.get("".join(groups[:3]), flint.__FLINT_RELEASE__)
+                        <= flint.__FLINT_RELEASE__
+                    ):
+                        res.append(test)
 
             tests.append((module_info.name, res))