[Python-checkins] r62887 - python/trunk/Lib/test/test_support.py

[brett.cannon]

Author: brett.cannon
Date: Thu May  8 21:50:51 2008
New Revision: 62887

Log:
Make test.test_support.catch_warning() take an argument specifying if any
triggered warnings should be captured. This allows the context manager to be
used to just prevent the internal state of the 'warnings' framework and thus
allow triggered warnings to be displayed.


Modified:
   python/trunk/Lib/test/test_support.py

Modified: python/trunk/Lib/test/test_support.py
==============================================================================
--- python/trunk/Lib/test/test_support.py	(original)
+++ python/trunk/Lib/test/test_support.py	Thu May  8 21:50:51 2008
@@ -40,7 +40,7 @@
 def import_module(name, deprecated=False):
     """Import the module to be tested, raising TestSkipped if it is not
     available."""
-    with catch_warning():
+    with catch_warning(record=False):
         if deprecated:
             warnings.filterwarnings("ignore", ".+ module", DeprecationWarning)
         try:
@@ -395,7 +395,7 @@
 
 
 @contextlib.contextmanager
-def catch_warning(module=warnings):
+def catch_warning(module=warnings, record=True):
     """
     Guard the warnings filter from being permanently changed and record the
     data of the last warning that has been issued.
@@ -406,12 +406,13 @@
             warnings.warn("foo")
             assert str(w.message) == "foo"
     """
-    warning_obj = WarningMessage()
+    if record:
+        warning_obj = WarningMessage()
+        module.showwarning = warning_obj._showwarning
     original_filters = module.filters[:]
     original_showwarning = module.showwarning
-    module.showwarning = warning_obj._showwarning
     try:
-        yield warning_obj
+        yield warning_obj if record else None
     finally:
         module.showwarning = original_showwarning
         module.filters = original_filters