Optionally surface exceptions that happen under executors by reading

[python_utils.git] / list_utils.py

diff --git a/list_utils.py b/list_utils.py
index 992f1ae4207228711c17573bb779cc0aaae2d0f8..b65ab0d69fab36196a94b4ef3327fae63b88f48d 100644 (file)
--- a/list_utils.py
+++ b/list_utils.py
@@ -197,7 +197,7 @@ def ngrams(lst: Sequence[Any], n):
     ['an', 'awesome', 'test']
     """
     for i in range(len(lst) - n + 1):
-        yield lst[i:i + n]
+        yield lst[i : i + n]
 
 
 def permute(seq: Sequence[Any]):
@@ -216,6 +216,7 @@ def permute(seq: Sequence[Any]):
     """
     yield from _permute(seq, "")
 
+
 def _permute(seq: Sequence[Any], path):
     if len(seq) == 0:
         yield path
@@ -223,12 +224,14 @@ def _permute(seq: Sequence[Any], path):
     for i in range(len(seq)):
         car = seq[i]
         left = seq[0:i]
-        right = seq[i + 1:]
+        right = seq[i + 1 :]
         cdr = left + right
         yield from _permute(cdr, path + car)
 
 
-def binary_search(lst: Sequence[Any], target:Any) -> Tuple[bool, int]:
+def binary_search(
+    lst: Sequence[Any], target: Any, *, sanity_check=False
+) -> Tuple[bool, int]:
     """Performs a binary search on lst (which must already be sorted).
     Returns a Tuple composed of a bool which indicates whether the
     target was found and an int which indicates the index closest to
@@ -247,11 +250,25 @@ def binary_search(lst: Sequence[Any], target:Any) -> Tuple[bool, int]:
     >>> binary_search(a, 2)
     (False, 1)
 
+    >>> a.append(9)
+    >>> binary_search(a, 4, sanity_check=True)
+    Traceback (most recent call last):
+    ...
+    AssertionError
+
     """
+    if sanity_check:
+        last = None
+        for x in lst:
+            if last is not None:
+                assert x >= last  # This asserts iff the list isn't sorted
+            last = x  # in ascending order.
     return _binary_search(lst, target, 0, len(lst) - 1)
 
 
-def _binary_search(lst: Sequence[Any], target: Any, low: int, high: int) -> Tuple[bool, int]:
+def _binary_search(
+    lst: Sequence[Any], target: Any, low: int, high: int
+) -> Tuple[bool, int]:
     if high >= low:
         mid = (high + low) // 2
         if lst[mid] == target:
@@ -266,4 +283,5 @@ def _binary_search(lst: Sequence[Any], target: Any, low: int, high: int) -> Tupl
 
 if __name__ == '__main__':
     import doctest
+
     doctest.testmod()