commit f2b55fb5ee0374babd770dd5aabad5045a1f78dc
author Brett Cannon <bcannon@gmail.com> Mon Mar 09 00:02:01 2009 +0000
committer Brett Cannon <bcannon@gmail.com> Mon Mar 09 00:02:01 2009 +0000
tree 00253bb29a16dc763e44544578f4e7053effee56
parent e8662063028a77ae89fc73c0d7721f9818a3d004

Make importlib.test.source.util.write_bytecode reset sys.dont_write_bytecode.

Lib/importlib/test/source/util.py

diff --git a/Lib/importlib/test/source/util.py b/Lib/importlib/test/source/util.py
index 333647d..f02d491 100644
--- a/Lib/importlib/test/source/util.py
+++ b/Lib/importlib/test/source/util.py
@@ -14,7 +14,11 @@
     if sys.dont_write_bytecode:
         return lambda *args, **kwargs: None
     else:
-        return fxn
+        def wrapper(*args, **kwargs):
+            to_return = fxn(*args, **kwargs)
+            sys.dont_write_bytecode = False
+            return to_return
+        return wrapper
 
 
 def bytecode_path(source_path):