Update test_datetime from 3.13.5 (#5911)

This commit is contained in:
Shahar Naveh
2025-07-08 13:46:26 +03:00
committed by GitHub
parent ec577e556d
commit 5c8b027af4

View File

@@ -1,5 +1,6 @@
import unittest
import sys
import functools
from test.support.import_helper import import_fresh_module
@@ -45,21 +46,26 @@ def load_tests(loader, tests, pattern):
for cls in test_classes:
cls.__name__ += suffix
cls.__qualname__ += suffix
@classmethod
def setUpClass(cls_, module=module):
cls_._save_sys_modules = sys.modules.copy()
sys.modules[TESTS] = module
sys.modules['datetime'] = module.datetime_module
if hasattr(module, '_pydatetime'):
sys.modules['_pydatetime'] = module._pydatetime
sys.modules['_strptime'] = module._strptime
@classmethod
def tearDownClass(cls_):
sys.modules.clear()
sys.modules.update(cls_._save_sys_modules)
cls.setUpClass = setUpClass
cls.tearDownClass = tearDownClass
tests.addTests(loader.loadTestsFromTestCase(cls))
@functools.wraps(cls, updated=())
class Wrapper(cls):
@classmethod
def setUpClass(cls_, module=module):
cls_._save_sys_modules = sys.modules.copy()
sys.modules[TESTS] = module
sys.modules['datetime'] = module.datetime_module
if hasattr(module, '_pydatetime'):
sys.modules['_pydatetime'] = module._pydatetime
sys.modules['_strptime'] = module._strptime
super().setUpClass()
@classmethod
def tearDownClass(cls_):
super().tearDownClass()
sys.modules.clear()
sys.modules.update(cls_._save_sys_modules)
tests.addTests(loader.loadTestsFromTestCase(Wrapper))
return tests