-

cool-RR · cool-RR · commit 4c0be037b421 · 2014-09-14T03:31:10.000+03:00
diff --git a/source_py3/python_toolbox/cute_iter_tools.py b/source_py3/python_toolbox/cute_iter_tools.py
@@ -346,7 +346,7 @@ def call_until_exception(function, exception, lazy_tuple=False):
     '''
     iterator = _call_until_exception(function, exception)
     if lazy_tuple:
-        from python_toolbox import nifty_collections # Avoiding circular import.
+        from python_toolbox import nifty_collections # Avoiding circular import
         return nifty_collections.LazyTuple(iterator)
     else:
         return iterator
@@ -469,3 +469,18 @@ def push_back(self):
         self.just_pushed_back = True
         
  
+ 
+def iterate_pop(poppable, lazy_tuple=False):
+    return call_until_exception(poppable.pop, IndexError,
+                                lazy_tuple=lazy_tuple)
+ 
+def iterate_popleft(left_poppable, lazy_tuple=False):
+    return call_until_exception(left_poppable.popleft, IndexError,
+                                lazy_tuple=lazy_tuple)
+
+def iterate_popitem(item_poppable, lazy_tuple=False):
+    return call_until_exception(item_poppable.popitem, KeyError,
+                                lazy_tuple=lazy_tuple)
+    
+
+
diff --git a/source_py3/test_python_toolbox/test_cute_iter_tools/test_pop_iterators.py b/source_py3/test_python_toolbox/test_cute_iter_tools/test_pop_iterators.py
@@ -0,0 +1,27 @@
+# Copyright 2009-2014 Ram Rachum.
+# This program is distributed under the MIT license.
+
+import collections
+from python_toolbox import nifty_collections
+
+from python_toolbox.cute_iter_tools import (iterate_pop, iterate_popitem,
+                                            iterate_popleft)
+
+
+def test():
+    
+    deque = collections.deque(range(10))
+    assert tuple(iterate_pop(deque)) == tuple(range(9, -1, -1))
+    assert not deque
+    
+    deque = collections.deque(range(10))
+    assert tuple(iterate_popleft(deque)) == tuple(range(10))
+    assert not deque
+    
+    dict_ = {1: 2, 3: 4, 5: 6,}
+    assert dict(iterate_popitem(dict_)) == {1: 2, 3: 4, 5: 6,}
+    assert not dict_
+    
+    lazy_tuple = iterate_pop(list(range(5)))
+    assert isinstance(lazy_tuple, nifty_collections.LazyTuple)
+    
