Refactor test

20ee765d · Jason R. Coombs · 365ca3bd · 20ee765d
Commit 20ee765d authored May 17, 2014 by Jason R. Coombs
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

ez_setup.py ez_setup.py +2 -2

No files found.
--- a/ez_setup.py
+++ b/ez_setup.py
@@ -78,8 +78,8 @@ def get_zip_class():
            return self
        def __exit__(self, type, value, traceback):
            self.close()
-    return zipfile.ZipFile if hasattr(zipfile.ZipFile, '__exit__') else \
-        ContextualZipFile
+    zf_has_exit = hasattr(zipfile.ZipFile, '__exit__')
+    return zipfile.ZipFile if zf_has_exit else ContextualZipFile


 @contextlib.contextmanager