Add .prec attribute to series

Adds .prec attribute to fmpz_series, fmpq_series, arb_series and acb_series.

Also fixes the doctest runner to run all tests and fixes some doctests.

Make sure that doctests use the correct precision.

Author: oscarbenjamin

src/flint/test/test_all.py

@@ -666,7 +666,11 @@ def test_fmpz_series():
     s2 = Z([1,2])
     s3 = Z([1,1])
     s4 = Z([1,2],11)
+    s5 = Z([1,2],3)
     p1 = Zp([1,2])
+    assert s1.prec == 10
+    assert s4.prec == 11
+    assert s5.prec == 3
     assert s1._equal_repr(s1) is True
     assert s1._equal_repr(s2) is True
     assert s1._equal_repr(s3) is False
@@ -1157,10 +1161,14 @@ def test_fmpq_series():
     s2 = Q([1,2])
     s3 = Q([1,1])
     s4 = Q([1,2],1,11)
+    s5 = Q([1,2],1,3)
     p1 = Qp([1,2])
     sz1 = Z([1,2])
     sz2 = Z([1,1])
     sz3 = Z([1,1],11)
+    assert s1.prec == 10
+    assert s4.prec == 11
+    assert s5.prec == 3
     assert s1._equal_repr(s1) is True
     assert s1._equal_repr(s2) is True
     assert s1._equal_repr(s3) is False

src/flint/test/test_docstrings.py

@@ -5,7 +5,7 @@
 
 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,
@@ -16,23 +16,36 @@
 def find_doctests(module):
     finder = doctest.DocTestFinder()
     tests = []
+    tests_seen = set()
     for module_info in pkgutil.walk_packages(module.__path__, flint.__name__ + "."):
         try:
             module = importlib.import_module(module_info.name)
 
             res = []
             for test in filter(lambda x: bool(x.examples), finder.find(module)):
+
                 m = dunder_test_regex.match(test.name)
                 if m is not None:
                     groups = m.groups()
-                    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)
+                    test.name = groups[0] + groups[1]
+                    test.lineno = int(groups[2])
+
+                    # Prefer the __test__ version of the test (remove the other)
+                    if test.name in tests_seen:
+                        n = len(res)
+                        res = [r for r in res if r.name != test.name]
+                        if len(res) != n - 1:
+                            raise Exception(f"Duplicate test name: {test.name}")
+                        tests_seen.remove(test.name)
+
+                # Doctests that fail without latest flint
+                if test.name in test_flint_at_least:
+                    if test_flint_at_least[test.name] > flint.__FLINT_RELEASE__:
+                        continue
+
+                if test.name not in tests_seen:
+                    tests_seen.add(test.name)
+                    res.append(test)
 
             tests.append((module_info.name, res))
 

src/flint/types/acb_series.pxd

@@ -4,6 +4,6 @@ from flint.flintlib.functions.acb_poly cimport acb_poly_t
 
 cdef class acb_series(flint_series):
     cdef acb_poly_t val
-    cdef long prec
+    cdef long _prec
     cpdef long length(self)
     cpdef valuation(self)