Commit d6f6e50c authored by Raymond Hettinger's avatar Raymond Hettinger

Reworked test_warnings.py:

* It ran fine under "python regrtest.py test_warnings" but failed under
  "python regrtest.py" presumably because other tests would add to
  filtered warnings and not reset them at the end of the test.

* Converted to a unittest format for better control.  Renamed
  monkey() and unmonkey() to setUp() and tearDown().

* Increased coverage by testing all warnings in __builtin__.

* Increased coverage by testing regex matching of specific messages.
parent dc9dcf13
test_warnings
('ignore', False, 'FutureWarning', False, 0)
('ignore', True, 'OverflowWarning', True, 0)
('ignore', True, 'PendingDeprecationWarning', True, 0)
test_warnings.py:31: UserWarning: hello world
test_warnings.py:32: UserWarning: hello world
test_warnings.py:33: DeprecationWarning: hello world
test_warnings.py:35: UserWarning: hello world
Caught UserWarning: hello world
Caught AssertionError: invalid action: 'booh'
import warnings import warnings
import os import os
import unittest
from test import test_support
# The warnings module isn't easily tested, because it relies on module # The warnings module isn't easily tested, because it relies on module
# globals to store configuration information. We need to extract the # globals to store configuration information. setUp() and tearDown()
# current settings to avoid bashing them while running tests. # preserve the current settings to avoid bashing them while running tests.
_filters = [] # To capture the warning messages, a replacement for showwarning() is
_showwarning = None # used to save warning information in a global variable.
class WarningMessage:
"Holds results of latest showwarning() call"
pass
def showwarning(message, category, filename, lineno, file=None): def showwarning(message, category, filename, lineno, file=None):
filename = os.path.basename(filename) msg.message = str(message)
print "%s:%s: %s: %s" % (filename, lineno, category.__name__, message) msg.category = category.__name__
msg.filename = os.path.basename(filename)
msg.lineno = lineno
class TestModule(unittest.TestCase):
def monkey(): def setUp(self):
global _filters, _showwarning global msg
_filters = warnings.filters[:] msg = WarningMessage()
_showwarning = warnings.showwarning self._filters = warnings.filters[:]
self._showwarning = warnings.showwarning
warnings.showwarning = showwarning warnings.showwarning = showwarning
self.ignored = [w[2].__name__ for w in self._filters
if w[0]=='ignore' and w[1] is None and w[3] is None]
def unmonkey(): def tearDown(self):
warnings.filters = _filters[:] warnings.filters = self._filters[:]
warnings.showwarning = _showwarning warnings.showwarning = self._showwarning
def test(): def test_warn_default_category(self):
for item in warnings.filters:
print (item[0], item[1] is None, item[2].__name__, item[3] is None,
item[4])
hello = "hello world"
for i in range(4): for i in range(4):
warnings.warn(hello) text = 'multi %d' %i # Different text on each call
warnings.warn(hello, UserWarning) warnings.warn(text)
warnings.warn(hello, DeprecationWarning) self.assertEqual(msg.message, text)
for i in range(3): self.assertEqual(msg.category, 'UserWarning')
warnings.warn(hello)
warnings.filterwarnings("error", "", Warning, "", 0) def test_warn_specific_category(self):
try: text = 'None'
warnings.warn(hello) for category in [DeprecationWarning, FutureWarning, OverflowWarning,
except Exception, msg: PendingDeprecationWarning, RuntimeWarning,
print "Caught", msg.__class__.__name__ + ":", msg SyntaxWarning, UserWarning, Warning]:
if category.__name__ in self.ignored:
text = 'filtered out' + category.__name__
warnings.warn(text, category)
self.assertNotEqual(msg.message, text)
else: else:
print "No exception" text = 'unfiltered %s' % category.__name__
warnings.warn(text, category)
self.assertEqual(msg.message, text)
self.assertEqual(msg.category, category.__name__)
def test_filtering(self):
warnings.filterwarnings("error", "", Warning, "", 0)
self.assertRaises(UserWarning, warnings.warn, 'convert to error')
warnings.resetwarnings() warnings.resetwarnings()
try: text = 'handle normally'
warnings.filterwarnings("booh", "", Warning, "", 0) warnings.warn(text)
except Exception, msg: self.assertEqual(msg.message, text)
print "Caught", msg.__class__.__name__ + ":", msg self.assertEqual(msg.category, 'UserWarning')
else:
print "No exception" warnings.filterwarnings("ignore", "", Warning, "", 0)
text = 'filtered out'
warnings.warn(text)
self.assertNotEqual(msg.message, text)
warnings.resetwarnings()
warnings.filterwarnings("error", "hex*", Warning, "", 0)
self.assertRaises(UserWarning, warnings.warn, 'hex/oct')
text = 'nonmatching text'
warnings.warn(text)
self.assertEqual(msg.message, text)
self.assertEqual(msg.category, 'UserWarning')
def test_main(verbose=None):
test_support.run_unittest(TestModule)
monkey() if __name__ == "__main__":
test() test_main(verbose=True)
unmonkey()
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment