diff options
Diffstat (limited to 'setuptools/tests/contexts.py')
-rw-r--r-- | setuptools/tests/contexts.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/setuptools/tests/contexts.py b/setuptools/tests/contexts.py index a9626ae6..1620bdb3 100644 --- a/setuptools/tests/contexts.py +++ b/setuptools/tests/contexts.py @@ -3,6 +3,7 @@ import os import shutil import sys import contextlib +import site from ..compat import StringIO @@ -57,3 +58,12 @@ def quiet(): new_stderr.seek(0) sys.stdout = old_stdout sys.stderr = old_stderr + + +@contextlib.contextmanager +def save_user_site_setting(): + saved = site.ENABLE_USER_SITE + try: + yield saved + finally: + site.ENABLE_USER_SITE = saved |