Restore decimal context after decimal doctests (GH-120149)

The modified context caused tests failures in several other tests.
This commit is contained in:
Serhiy Storchaka 2024-06-06 20:12:32 +03:00 committed by GitHub
parent d50a7c478f
commit 2d7ff6e0e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions

View File

@ -5892,13 +5892,17 @@ def load_tests(loader, tests, pattern):
if TODO_TESTS is None:
from doctest import DocTestSuite, IGNORE_EXCEPTION_DETAIL
orig_context = orig_sys_decimal.getcontext().copy()
for mod in C, P:
if not mod:
continue
def setUp(slf, mod=mod):
sys.modules['decimal'] = mod
def tearDown(slf):
init(mod)
def tearDown(slf, mod=mod):
sys.modules['decimal'] = orig_sys_decimal
mod.setcontext(ORIGINAL_CONTEXT[mod].copy())
orig_sys_decimal.setcontext(orig_context.copy())
optionflags = IGNORE_EXCEPTION_DETAIL if mod is C else 0
sys.modules['decimal'] = mod
tests.addTest(DocTestSuite(mod, setUp=setUp, tearDown=tearDown,
@ -5913,8 +5917,8 @@ def setUpModule():
TEST_ALL = ARITH if ARITH is not None else is_resource_enabled('decimal')
def tearDownModule():
if C: C.setcontext(ORIGINAL_CONTEXT[C])
P.setcontext(ORIGINAL_CONTEXT[P])
if C: C.setcontext(ORIGINAL_CONTEXT[C].copy())
P.setcontext(ORIGINAL_CONTEXT[P].copy())
if not C:
warnings.warn('C tests skipped: no module named _decimal.',
UserWarning)