-

cool-RR · cool-RR · commit ba8129c05416 · 2015-12-12T00:44:50.000+02:00
diff --git a/source_py3/python_toolbox/context_management/functions.py b/source_py3/python_toolbox/context_management/functions.py
@@ -78,7 +78,8 @@ def __enter__(self):
             return self._enter_value
                 
             
-        def __exit__(self, exc_type, exc_value, exc_traceback):
+        def __exit__(self, exc_type=None, exc_value=None,
+                     exc_traceback=None):
             if self._entered:
                 exit_value = self.__wrapped__.__exit__(exc_type, exc_value,
                                                        exc_traceback)
diff --git a/source_py3/test_python_toolbox/test_context_management/test_idempotentify.py b/source_py3/test_python_toolbox/test_context_management/test_idempotentify.py
@@ -54,7 +54,7 @@ def test_idempotentify():
     assert idempotent_context_manager.__wrapped__.x == 1
     idempotent_context_manager.__exit__(None, None, None)
     assert idempotent_context_manager.__wrapped__.x == 0
-    idempotent_context_manager.__exit__(None, None, None)
+    idempotent_context_manager.__exit__()
     assert idempotent_context_manager.__wrapped__.x == 0
     idempotent_context_manager.__exit__(None, None, None)
     assert idempotent_context_manager.__wrapped__.x == 0
