gh-101277: Isolate itertools, add group and _grouper types to module …

…state (#101302) Co-authored-by: Kumar Aditya <59607654+kumaraditya303@users.noreply.github.com>
python · Feb 1, 2023 · 2b3e02a · 2b3e02a
1 parent cc407b9
commit 2b3e02a
Show file tree

Hide file tree

Showing 3 changed files with 173 additions and 111 deletions.
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
@@ -1694,6 +1694,38 @@ def test_zip_longest_result_gc(self):
         gc.collect()
         self.assertTrue(gc.is_tracked(next(it)))
 
+    @support.cpython_only
+    def test_immutable_types(self):
+        from itertools import _grouper, _tee, _tee_dataobject
+        dataset = (
+            accumulate,
+            batched,
+            chain,
+            combinations,
+            combinations_with_replacement,
+            compress,
+            count,
+            cycle,
+            dropwhile,
+            filterfalse,
+            groupby,
+            _grouper,
+            islice,
+            pairwise,
+            permutations,
+            product,
+            repeat,
+            starmap,
+            takewhile,
+            _tee,
+            _tee_dataobject,
+            zip_longest,
+        )
+        for tp in dataset:
+            with self.subTest(tp=tp):
+                with self.assertRaisesRegex(TypeError, "immutable"):
+                    tp.foobar = 1
+
 
 class TestExamples(unittest.TestCase):
 

diff --git a/Modules/clinic/itertoolsmodule.c.h b/Modules/clinic/itertoolsmodule.c.h