Add utils.consume from cytoolz

toolz/tests/test_utils.py

@@ -1,6 +1,15 @@
-from toolz.utils import raises
+from toolz.utils import raises, consume
 
 
 def test_raises():
     assert raises(ZeroDivisionError, lambda: 1 / 0)
     assert not raises(ZeroDivisionError, lambda: 1)
+
+
+def test_consume():
+    l = [1, 2, 3]
+    assert consume(l) is None
+    il = iter(l)
+    assert consume(il) is None
+    assert raises(StopIteration, lambda: next(il))
+    assert raises(TypeError, lambda: consume(1))

toolz/utils.py

@@ -1,3 +1,11 @@
+from collections import deque
+
+
+def consume(seq):
+    """Efficently consume an iterable"""
+    deque(seq, maxlen=0)
+
+
 def raises(err, lamda):
     try:
         lamda()