diff options
| -rw-r--r-- | setuptools/tests/py26compat.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/setuptools/tests/py26compat.py b/setuptools/tests/py26compat.py index 24e6dbe2..ead72fa6 100644 --- a/setuptools/tests/py26compat.py +++ b/setuptools/tests/py26compat.py @@ -17,10 +17,11 @@ except AttributeError: def _tarfile_open_ex(*args, **kwargs): """ - Extend result with an __exit__ to close the file. + Extend result as a context manager. """ res = tarfile.open(*args, **kwargs) res.__exit__ = lambda self: self.close() + res.__enter__ = lambda self: self return res tarfile_open = _tarfile_open_ex if sys.version_info < (2,7) else tarfile.open |