-

cool-RR · cool-RR · commit 0ad083a849c7 · 2014-02-09T01:43:07.000+02:00
diff --git a/source_py3/test_python_toolbox/test_cute_iter_tools/test_iter_with.py b/source_py3/test_python_toolbox/test_cute_iter_tools/test_iter_with.py
@@ -5,6 +5,7 @@
 
 import itertools
 
+from python_toolbox import nifty_collections
 from python_toolbox import context_management
 
 from python_toolbox.cute_iter_tools import iter_with
@@ -37,4 +38,20 @@ def test():
         assert inactive_context_manager.counter == -1
         assert inactive_context_manager.active is False
         
-    
+    
+def test_lazy_tuple():
+    
+    active_context_manager = MyContextManager()
+    inactive_context_manager = MyContextManager()
+    
+    lazy_tuple = iter_with(range(5), active_context_manager, lazy_tuple=True)
+    assert isinstance(lazy_tuple, nifty_collections.LazyTuple)
+    assert not lazy_tuple.collected_data
+    
+    for i, j in zip(lazy_tuple, range(5)):
+        assert i == j == active_context_manager.counter
+        assert active_context_manager.active is False
+        assert inactive_context_manager.counter == -1
+        assert inactive_context_manager.active is False
+        
+    assert lazy_tuple[2] == 2
diff --git a/source_py3/test_python_toolbox/test_cute_iter_tools/test_iterate_overlapping_subsequences.py b/source_py3/test_python_toolbox/test_cute_iter_tools/test_iterate_overlapping_subsequences.py
@@ -4,6 +4,7 @@
 '''Testing module for `cute_iter_tools.iterate_overlapping_subsequences`.'''
 
 from python_toolbox import gc_tools
+from python_toolbox import nifty_collections
 from python_toolbox import cute_testing
 from python_toolbox import sequence_tools
 
@@ -53,6 +54,17 @@ def test_various_lengths():
             (6, 0, 1, 2, 3))
                                           
            
+def test_lazy_tuple():
+    lazy_tuple = \
+          iterate_overlapping_subsequences(range(7), length=3, lazy_tuple=True)
+    assert isinstance(lazy_tuple, nifty_collections.LazyTuple)
+    assert not lazy_tuple.collected_data
+    
+    assert lazy_tuple == \
+                        ((0, 1, 2), (1, 2, 3), (2, 3, 4), (3, 4, 5), (4, 5, 6))
+    
+                                          
+           
 def test_garbage_collection():
     
     garbage_collected = set()
