Add a test

python · kumaraditya303 · Mar 4, 2023 · Feb 14, 2023 · Feb 14, 2023 · Feb 14, 2023
commit 0c631e3413b0203cfb2335fdeec2363f3e8a31c4
diff --git a/Lib/test/test_iter.py b/Lib/test/test_iter.py
@@ -267,6 +267,22 @@ def spam(state=[0]):
             return i
         self.check_iterator(iter(spam, 20), list(range(10)), pickle=False)
 
+    # Test two-argument iter() with function that empties its associated
+    # iterator with list() (or anything else but next()) then returns
+    # a non-sentinel value.
+    def test_iter_function_concealing_reentrant_exhaustion(self):
+        HAS_MORE = 1
+        NO_MORE = 2
+        def spam():
+            if not spam.is_reentrant:
+                spam.is_reentrant = True
+                list(spam.iterator)
+                return HAS_MORE
+            return NO_MORE
+        spam.is_reentrant = False
+        spam.iterator = iter(spam, NO_MORE)
+        next(spam.iterator)
+
     # Test exception propagation through function iterator
     def test_exception_function(self):
         def spam(state=[0]):