diff options
| author | Hai Shi <shihai1992@gmail.com> | 2020-07-06 09:12:49 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-07-06 09:12:49 (GMT) |
| commit | 883bc638335a57a6e6a6344c2fc132c4f9a0ec42 (patch) | |
| tree | 1dd74a4aeeb80d8b85ff18de700763199a7bc1bc /Lib/test/test_py_compile.py | |
| parent | 9ce8132e1f2339cfe116dfd4795574182c2245b4 (diff) | |
| download | cpython-883bc638335a57a6e6a6344c2fc132c4f9a0ec42.zip cpython-883bc638335a57a6e6a6344c2fc132c4f9a0ec42.tar.gz cpython-883bc638335a57a6e6a6344c2fc132c4f9a0ec42.tar.bz2 | |
bpo-40275: Use new test.support helper submodules in tests (GH-21314)
Diffstat (limited to 'Lib/test/test_py_compile.py')
| -rw-r--r-- | Lib/test/test_py_compile.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/Lib/test/test_py_compile.py b/Lib/test/test_py_compile.py index d4a68c9..11b579c 100644 --- a/Lib/test/test_py_compile.py +++ b/Lib/test/test_py_compile.py @@ -9,13 +9,14 @@ import tempfile import unittest from test import support +from test.support import os_helper def without_source_date_epoch(fxn): """Runs function with SOURCE_DATE_EPOCH unset.""" @functools.wraps(fxn) def wrapper(*args, **kwargs): - with support.EnvironmentVarGuard() as env: + with os_helper.EnvironmentVarGuard() as env: env.unset('SOURCE_DATE_EPOCH') return fxn(*args, **kwargs) return wrapper @@ -25,7 +26,7 @@ def with_source_date_epoch(fxn): """Runs function with SOURCE_DATE_EPOCH set.""" @functools.wraps(fxn) def wrapper(*args, **kwargs): - with support.EnvironmentVarGuard() as env: + with os_helper.EnvironmentVarGuard() as env: env['SOURCE_DATE_EPOCH'] = '123456789' return fxn(*args, **kwargs) return wrapper |