-

cool-RR · cool-RR · commit c2fb19f57b17 · 2014-02-09T01:40:38.000+02:00
diff --git a/source_py3/test_python_toolbox/test_cute_iter_tools/test_shorten.py b/source_py3/test_python_toolbox/test_cute_iter_tools/test_shorten.py
@@ -5,6 +5,7 @@
 
 import nose.tools
 
+from python_toolbox import nifty_collections
 from python_toolbox import cute_iter_tools
 from python_toolbox.cute_iter_tools import shorten
 
@@ -29,6 +30,16 @@ def test():
     assert list(shorten(iter(my_range), infinity)) == my_range
 
     
+def test_lazy_tuple():
+    my_range = [0, 1, 2, 3, 4]
+
+    lazy_tuple = shorten(my_range, 3, lazy_tuple=True)
+    assert isinstance(lazy_tuple, nifty_collections.LazyTuple)
+    assert not lazy_tuple.collected_data
+    
+    assert tuple(lazy_tuple) == (0, 1, 2)
+    
+    
 def test_dont_pull_extra_item():
     '''Test that `shorten` doesn't pull an extra member from the iterable.'''
     def generator():
