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

553bdaf5 · Brett Cannon · c66c354a · 553bdaf5
Commit 553bdaf5 authored Mar 09, 2009 by Brett Cannon
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

Lib/importlib/test/source/util.py Lib/importlib/test/source/util.py +5 -1

No files found.
--- a/Lib/importlib/test/source/util.py
+++ b/Lib/importlib/test/source/util.py
@@ -14,7 +14,11 @@ def writes_bytecode(fxn):
    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):