Add some more boundary conditions

jaraco · Mar 28, 2019 · a24139c · a24139c
1 parent f107221
commit a24139c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/jaraco/itertools.py b/jaraco/itertools.py
@@ -695,6 +695,20 @@ def takewhile_peek(predicate, iterable):
 	>>> empty = takewhile_peek(is_small, Peekable([]))
 	>>> list(empty)
 	[]
+
+	>>> items = Peekable([3])
+	>>> small_items = takewhile_peek(is_small, items)
+	>>> list(small_items)
+	[3]
+	>>> list(items)
+	[]
+
+	>>> items = Peekable([4])
+	>>> small_items = takewhile_peek(is_small, items)
+	>>> list(small_items)
+	[]
+	>>> list(items)
+	[4]
 	"""
 	while True:
 		if not predicate(iterable.peek()):