Close #19396: make test_contextlib tolerate -S

0ddaed34 · Nick Coghlan · 7bad39f1 · 0ddaed34
Commit 0ddaed34 authored Oct 26, 2013 by Nick Coghlan
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

Lib/test/test_contextlib.py Lib/test/test_contextlib.py +4 -3

No files found.
--- a/Lib/test/test_contextlib.py
+++ b/Lib/test/test_contextlib.py
@@ -636,10 +636,11 @@ class TestRedirectStdout(unittest.TestCase):
    def test_redirect_to_string_io(self):
        f = io.StringIO()
+        msg = "Consider an API like help(), which prints directly to stdout"
        with redirect_stdout(f):
-            help(pow)
+            print(msg)
-        s = f.getvalue()
+        s = f.getvalue().strip()
-        self.assertIn('pow', s)
+        self.assertEqual(s, msg)
    def test_enter_result_is_target(self):
        f = io.StringIO()