Hammer on that run_tests.py thing again.

[python_utils.git] / list_utils.py

diff --git a/list_utils.py b/list_utils.py
index 7f28ade922a4b733ee29ea0c68bc957ff587c51b..1141af2fe9b6bb8861dfd25c08dab735bfd4ceaf 100644 (file)
--- a/list_utils.py
+++ b/list_utils.py
@@ -1,10 +1,12 @@
 #!/usr/bin/env python3
 
+# © Copyright 2021-2022, Scott Gasch
+
 """Some useful(?) utilities for dealing with Lists."""
 
 import random
 from collections import Counter
-from itertools import islice
+from itertools import chain, combinations, islice
 from typing import Any, Iterator, List, MutableSequence, Sequence, Tuple
 
 
@@ -305,6 +307,23 @@ def _binary_search(lst: Sequence[Any], target: Any, low: int, high: int) -> Tupl
         return (False, low)
 
 
+def powerset(lst: Sequence[Any]) -> Iterator[Sequence[Any]]:
+    """Returns the powerset of the items in the input sequence.
+
+    >>> for x in powerset([1, 2, 3]):
+    ...     print(x)
+    ()
+    (1,)
+    (2,)
+    (3,)
+    (1, 2)
+    (1, 3)
+    (2, 3)
+    (1, 2, 3)
+    """
+    return chain.from_iterable(combinations(lst, r) for r in range(len(lst) + 1))
+
+
 if __name__ == '__main__':
     import doctest