diff options
author | Serhiy Storchaka <storchaka@gmail.com> | 2021-09-19 12:27:33 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-09-19 12:27:33 (GMT) |
commit | 40348acc180580371d25f75f46b27048e35f2435 (patch) | |
tree | 4ddb641b0dfc1edda6d0a6c044c1380c4900d58a /Lib/test/support/__init__.py | |
parent | 664448d81f41c5fa971d8523a71b0f19e76cc136 (diff) | |
download | cpython-40348acc180580371d25f75f46b27048e35f2435.zip cpython-40348acc180580371d25f75f46b27048e35f2435.tar.gz cpython-40348acc180580371d25f75f46b27048e35f2435.tar.bz2 |
bpo-45229: Remove test_main in many tests (GH-28405)
Instead of explicitly enumerate test classes for run_unittest()
use the unittest ability to discover tests. This also makes these
tests discoverable and runnable with unittest.
load_tests() can be used for dynamic generating tests and adding
doctests. setUpModule(), tearDownModule() and addModuleCleanup()
can be used for running code before and after all module tests.
Diffstat (limited to 'Lib/test/support/__init__.py')
-rw-r--r-- | Lib/test/support/__init__.py | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/Lib/test/support/__init__.py b/Lib/test/support/__init__.py index 04516a5..bc40bba 100644 --- a/Lib/test/support/__init__.py +++ b/Lib/test/support/__init__.py @@ -695,9 +695,8 @@ def check_sizeof(test, o, size): # Decorator for running a function in a different locale, correctly resetting # it afterwards. +@contextlib.contextmanager def run_with_locale(catstr, *locales): - def decorator(func): - def inner(*args, **kwds): try: import locale category = getattr(locale, catstr) @@ -716,16 +715,11 @@ def run_with_locale(catstr, *locales): except: pass - # now run the function, resetting the locale on exceptions try: - return func(*args, **kwds) + yield finally: if locale and orig_locale: locale.setlocale(category, orig_locale) - inner.__name__ = func.__name__ - inner.__doc__ = func.__doc__ - return inner - return decorator #======================================================================= # Decorator for running a function in a specific timezone, correctly |