Use match/case in grouper() recipe (pythongh-113059)

Use match/case in grouper() reciper (cherry picked from commit 2111795) Co-authored-by: Raymond Hettinger <[email protected]>
miss-islington · Dec 15, 2023 · 3f18633 · 3f18633
1 parent 85ee49c
commit 3f18633
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/Doc/library/itertools.rst b/Doc/library/itertools.rst
@@ -914,14 +914,15 @@ which incur interpreter overhead.
        # grouper('ABCDEFG', 3, incomplete='strict') --> ABC DEF ValueError
        # grouper('ABCDEFG', 3, incomplete='ignore') --> ABC DEF
        args = [iter(iterable)] * n
-       if incomplete == 'fill':
-           return zip_longest(*args, fillvalue=fillvalue)
-       elif incomplete == 'strict':
-           return zip(*args, strict=True)
-       elif incomplete == 'ignore':
-           return zip(*args)
-       else:
-           raise ValueError('Expected fill, strict, or ignore')
+       match incomplete:
+           case 'fill':
+               return zip_longest(*args, fillvalue=fillvalue)
+           case 'strict':
+               return zip(*args, strict=True)
+           case 'ignore':
+               return zip(*args)
+           case _:
+               raise ValueError('Expected fill, strict, or ignore')
 
    def sliding_window(iterable, n):
        # sliding_window('ABCDEFG', 4) --> ABCD BCDE CDEF DEFG