Restore decimal context after decimal doctests (GH-120149)

serhiy-storchaka · web-flow · commit 2d7ff6e0e7d4 · 2024-06-06T20:12:32.000+03:00
The modified context caused tests failures in several other tests.
diff --git a/Lib/test/test_decimal.py b/Lib/test/test_decimal.py
@@ -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)

-Original file line number
+Diff line change
 ifTODO_TESTSisNone:
 fromdoctestimportDocTestSuite, IGNORE_EXCEPTION_DETAIL
 +orig_context=orig_sys_decimal.getcontext().copy()
 formodinC, P:
 ifnotmod:
 continue
 defsetUp(slf, mod=mod):
 sys.modules['decimal'] =mod
 -deftearDown(slf):
 +init(mod)
 +deftearDown(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_DETAILifmodisCelse0
 sys.modules['decimal'] =mod
 tests.addTest(DocTestSuite(mod, setUp=setUp, tearDown=tearDown,
 TEST_ALL=ARITHifARITHisnotNoneelseis_resource_enabled('decimal')
 deftearDownModule():
 -ifC: C.setcontext(ORIGINAL_CONTEXT[C])
 -P.setcontext(ORIGINAL_CONTEXT[P])
 +ifC: C.setcontext(ORIGINAL_CONTEXT[C].copy())
 +P.setcontext(ORIGINAL_CONTEXT[P].copy())
 ifnotC:
 warnings.warn('C tests skipped: no module named _decimal.',
 UserWarning)