https://github.com/python/cpython/commit/95f4db88d5ab7d900f05d0418b2a2e77bf9ff126
commit: 95f4db88d5ab7d900f05d0418b2a2e77bf9ff126
branch: main
author: Nikita Sobolev <[email protected]>
committer: sobolevn <[email protected]>
date: 2024-06-08T10:51:09+03:00
summary:
gh-120242: Fix handling of `[setUp,tearDown]Class` in `test_datetime` (#120243)
files:
M Lib/test/test_datetime.py
diff --git a/Lib/test/test_datetime.py b/Lib/test/test_datetime.py
index 3859733a4fe65b..005187f13e665f 100644
--- a/Lib/test/test_datetime.py
+++ b/Lib/test/test_datetime.py
@@ -1,5 +1,6 @@
import unittest
import sys
+import functools
from test.support.import_helper import import_fresh_module
@@ -39,21 +40,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
_______________________________________________
Python-checkins mailing list -- [email protected]
To unsubscribe send an email to [email protected]
https://mail.python.org/mailman3/lists/python-checkins.python.org/
Member address: [email protected]
