import unittest import sys from test.test_support import check_py3k_warnings, CleanImport, run_unittest import warnings if not sys.py3kwarning: raise unittest.SkipTest('%s must be run with the -3 flag' % __name__) try: from test.test_support import __warningregistry__ as _registry except ImportError: def check_deprecated_module(module_name): return False else: past_warnings = _registry.keys() del _registry def check_deprecated_module(module_name): """Lookup the past warnings for module already loaded using test_support.import_module(..., deprecated=True) """ return any(module_name in msg and ' removed' in msg and issubclass(cls, DeprecationWarning) and ( ' module' in msg or ' package' in msg) for (msg, cls, line) in past_warnings) def reset_module_registry(module): try: registry = module.__warningregistry__
from test.test_support import (check_warnings, CleanImport, TestSkipped, run_unittest) import warnings from contextlib import nested if not sys.py3kwarning: raise TestSkipped('%s must be run with the -3 flag' % __name__) try: from test.test_support import __warningregistry__ as _registry except ImportError: def check_deprecated_module(module_name): return False else: past_warnings = _registry.keys() del _registry def check_deprecated_module(module_name): """Lookup the past warnings for module already loaded using test_support.import_module(..., deprecated=True) """ return any(module_name in msg and ' removed' in msg and issubclass(cls, DeprecationWarning) and (' module' in msg or ' package' in msg) for (msg, cls, line) in past_warnings) def reset_module_registry(module): try: registry = module.__warningregistry__ except AttributeError: pass