# Needed by test_datetime
import _strptime
+try:
+ import _pydatetime
+except ImportError:
+ pass
#
pickle_loads = {pickle.loads, pickle._loads}
if '_Fast' in self.__class__.__name__:
self.skipTest('Only run for Pure Python implementation')
- dar = datetime_module._divide_and_round
+ dar = _pydatetime._divide_and_round
self.assertEqual(dar(-10, -3), 3)
self.assertEqual(dar(5, -2), -2)
def load_tests(loader, tests, pattern):
try:
- pure_tests = import_fresh_module(TESTS, fresh=['datetime', '_strptime'],
- blocked=['_datetime'])
- fast_tests = import_fresh_module(TESTS, fresh=['datetime',
- '_datetime', '_strptime'])
+ pure_tests = import_fresh_module(TESTS,
+ fresh=['datetime', '_pydatetime', '_strptime'],
+ blocked=['_datetime'])
+ fast_tests = import_fresh_module(TESTS,
+ fresh=['datetime', '_strptime'],
+ blocked=['_pydatetime'])
finally:
# XXX: import_fresh_module() is supposed to leave sys.module cache untouched,
# XXX: but it does not, so we have to cleanup ourselves.
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_):