mirrors
/
cpython
mirror of https://github.com/python/cpython.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 3.4 (test_warnings)

This commit is contained in:
Victor Stinner 2015-09-03 00:09:26 +02:00
parent 329ca7118f 1c40552087
commit 4912e7a3fd
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Lib/test/test_warnings.py
View File

@ -44,6 +44,7 @@ class BaseTest:
"""Basic bookkeeping required for testing.""" """Basic bookkeeping required for testing."""
def setUp(self): def setUp(self):
self.old_unittest_module = unittest.case.warnings
# The __warningregistry__ needs to be in a pristine state for tests # The __warningregistry__ needs to be in a pristine state for tests
# to work properly. # to work properly.
if '__warningregistry__' in globals(): if '__warningregistry__' in globals():
@ -55,10 +56,15 @@ def setUp(self):
# The 'warnings' module must be explicitly set so that the proper # The 'warnings' module must be explicitly set so that the proper
# interaction between _warnings and 'warnings' can be controlled. # interaction between _warnings and 'warnings' can be controlled.
sys.modules['warnings'] = self.module sys.modules['warnings'] = self.module
# Ensure that unittest.TestCase.assertWarns() uses the same warnings
# module than warnings.catch_warnings(). Otherwise,
# warnings.catch_warnings() will be unable to remove the added filter.
unittest.case.warnings = self.module
super(BaseTest, self).setUp() super(BaseTest, self).setUp()
def tearDown(self): def tearDown(self):
sys.modules['warnings'] = original_warnings sys.modules['warnings'] = original_warnings
unittest.case.warnings = self.old_unittest_module
super(BaseTest, self).tearDown() super(BaseTest, self).tearDown()
class PublicAPITests(BaseTest): class PublicAPITests(BaseTest):