Skip to content

Commit

Permalink
Suppress expected DeprecationWarning output in test.
Browse files Browse the repository at this point in the history
  • Loading branch information
tseaver committed Nov 22, 2005
1 parent ad391a5 commit 4cc2a70
Showing 1 changed file with 27 additions and 1 deletion.
28 changes: 27 additions & 1 deletion tests/testPythonScript.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,30 @@
if not here:
here = os.getcwd()

class WarningInterceptor:

_old_stderr = None
_our_stderr_stream = None

def _trap_warning_output( self ):

if self._old_stderr is not None:
return

import sys
from StringIO import StringIO

self._old_stderr = sys.stderr
self._our_stderr_stream = sys.stderr = StringIO()

def _free_warning_output( self ):

if self._old_stderr is None:
return

import sys
sys.stderr = self._old_stderr

# Test Classes

def readf(name):
Expand Down Expand Up @@ -219,7 +243,7 @@ def testAttributeAssignment(self):
f = self._newPS(defn + "\n" + asn % name)
self.assertRaises(TypeError, f)

class TestPythonScriptGlobals(PythonScriptTestBase):
class TestPythonScriptGlobals(PythonScriptTestBase, WarningInterceptor):
def _exec(self, script, bound_names=None, args=None, kws=None):
if args is None:
args = ()
Expand All @@ -241,8 +265,10 @@ def test__name__(self):
def test_filepath(self):
# This test is meant to raise a deprecation warning.
# It used to fail mysteriously instead.
self._trap_warning_output()
f = self._filePS('filepath')
self.assertEqual(f(), [0])
self._free_warning_output()

def test_suite():
suite = unittest.TestSuite()
Expand Down

0 comments on commit 4cc2a70

Please sign in to comment.