-

cool-RR · cool-RR · commit cf0604769b57 · 2014-09-14T03:21:33.000+03:00
diff --git a/source_py3/python_toolbox/context_management/functions.py b/source_py3/python_toolbox/context_management/functions.py
@@ -7,6 +7,8 @@
 See their documentation for more information.
 '''
 
+import sys
+
 from .context_manager_type import ContextManagerType
 
 
diff --git a/source_py3/test_python_toolbox/test_context_management/test_context_manager.py b/source_py3/test_python_toolbox/test_context_management/test_context_manager.py
@@ -1,8 +1,6 @@
 # Copyright 2009-2014 Ram Rachum.
 # This program is distributed under the MIT license.
 
-'''Test the `python_toolbox.context_management` module.'''
-
 from python_toolbox import cute_testing
 
 from python_toolbox.context_management import (ContextManager,
diff --git a/source_py3/test_python_toolbox/test_context_management/test_functions.py b/source_py3/test_python_toolbox/test_context_management/test_functions.py
@@ -0,0 +1,26 @@
+# Copyright 2009-2014 Ram Rachum.
+# This program is distributed under the MIT license.
+
+from python_toolbox import context_management
+
+from python_toolbox import freezing
+
+
+def test_nested():
+    freezer_a = freezing.Freezer()
+    freezer_b = freezing.Freezer()
+    freezer_c = freezing.Freezer()
+    freezer_d = freezing.Freezer()
+    
+    freezers = (freezer_a, freezer_b, freezer_c)
+    
+    assert freezer_a.frozen == freezer_b.frozen == freezer_c.frozen == \
+                                                          freezer_d.frozen == 0
+    
+    with context_management.nested(*freezers):
+        assert freezer_a.frozen == freezer_b.frozen == freezer_c.frozen == 1
+        assert freezer_d.frozen == 0
+        
+    assert freezer_a.frozen == freezer_b.frozen == freezer_c.frozen == \
+               freezer_d.frozen == 0
+    
