-

cool-RR · cool-RR · commit 94ad385b0561 · 2013-01-26T02:21:59.000+02:00
diff --git a/python_toolbox/cute_iter_tools.py b/python_toolbox/cute_iter_tools.py
@@ -190,3 +190,13 @@ def sentinel(counter=([fill_value] * (len(iterables) - 1)).pop):
             yield tuple_
     except IndexError:
         raise StopIteration
+
+
+def get_items(iterable, n):
+    '''
+    Get the next `n` items from the iterable as a `tuple`.
+    
+    If there are less than `n` items, no exception will be raised. Whatever
+    items are there will be returned.
+    '''
+    return tuple(shorten(iterable, n))
diff --git a/test_python_toolbox/test_cute_iter_tools/test_get_items.py b/test_python_toolbox/test_cute_iter_tools/test_get_items.py
@@ -0,0 +1,23 @@
+# Copyright 2009-2012 Ram Rachum.
+# This program is distributed under the MIT license.
+
+'''Testing module for `cute_iter_tools.get_items`.'''
+
+import itertools
+
+
+from python_toolbox.cute_iter_tools import get_items
+
+
+
+def test():
+    '''Test the basic workings of `get_items`.'''
+    
+    iterable = iter(xrange(10))
+    assert get_items(iterable, 3) == (0, 1, 2)
+    assert get_items(iterable, 0) == ()
+    assert get_items(iterable, 2) == (3, 4)
+    assert get_items(iterable, 4) == (5, 6, 7, 8)
+    assert get_items(iterable, 3) == (9,)
+    assert get_items(iterable, 3) == ()
+    assert get_items(iterable, 4) == ()
