slightly improve doctests

sagemath · Dec 18, 2021 · ae80b6d · ae80b6d
1 parent 6a56bb5
commit ae80b6d
Showing 1 changed file with 7 additions and 12 deletions.
diff --git a/src/sage/rings/continued_fraction_gosper.py b/src/sage/rings/continued_fraction_gosper.py
@@ -102,15 +102,15 @@ def __iter__(self):
 
         TESTS::
 
-            sage: a = Integer(randint(-100,100)); b = Integer(randint(-100,100));
-            sage: c = Integer(randint(-100,100)); d = Integer(randint(-100,100));
             sage: from sage.rings.continued_fraction_gosper import gosper_iterator
-            sage: ig = iter(gosper_iterator(a,b,c,d,continued_fraction(pi))); icf = iter(continued_fraction((a*pi+b)/(c*pi+d)));
+            sage: a, b, c, d = (Integer(randint(-100,100)) for _ in range(4))
+            sage: ig = iter(gosper_iterator(a, b, c, d, continued_fraction(pi)))
+            sage: icf = iter(continued_fraction((a*pi + b) / (c*pi + d)));
             sage: for i in range(10):
             ....:     try:
             ....:         assert next(ig) == next(icf)
             ....:     except StopIteration:
-            ....:         pass
+            ....:         break
         """
         return self
 
@@ -120,15 +120,10 @@ def __next__(self):
 
         TESTS::
 
-            sage: a = Integer(randint(-100,100)); b = Integer(randint(-100,100));
-            sage: c = Integer(randint(-100,100)); d = Integer(randint(-100,100));
             sage: from sage.rings.continued_fraction_gosper import gosper_iterator
-            sage: ig = iter(gosper_iterator(a,b,c,d,continued_fraction(pi))); icf = iter(continued_fraction((a*pi+b)/(c*pi+d)));
-            sage: for i in range(10):
-            ....:     try:
-            ....:         assert next(ig) == next(icf)
-            ....:     except StopIteration:
-            ....:         pass
+            sage: it = gosper_iterator(1, 0, 0, 1, continued_fraction(pi))
+            sage: list(next(it) for _ in range(10))
+            [3, 7, 15, 1, 292, 1, 1, 1, 2, 1]
         """
         while True:
             if self.currently_read >= self.input_preperiod_length: