Skip to content

Commit 6856918

Browse files
committed
Add suggestions
1 parent 864f656 commit 6856918

1 file changed

Lines changed: 5 additions & 8 deletions

File tree

Lib/test/test_pickle.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,10 @@
1010
import tempfile
1111
import warnings
1212
import weakref
13+
from textwrap import dedent
1314

1415
import doctest
1516
import unittest
16-
from textwrap import dedent
1717
from test import support
1818
from test.support import import_helper, os_helper
1919

@@ -746,15 +746,12 @@ def test_invocation(self):
746746
self.assertListEqual(res.splitlines(), expect.splitlines())
747747

748748
def test_unknown_flag(self):
749-
output = io.StringIO()
749+
stderr = io.StringIO()
750750
with self.assertRaises(SystemExit):
751-
# suppress argparse error message
752-
with contextlib.redirect_stderr(output):
751+
# check that the parser help is shown
752+
with contextlib.redirect_stderr(stderr):
753753
_ = self.invoke_pickle('--unknown')
754-
msg = output.getvalue()
755-
self.assertTrue(msg.startswith('usage: '),
756-
"Output does not start with 'usage: '. "
757-
f"Output was: {msg}")
754+
self.assertStartsWith(stderr.getvalue(), 'usage: ')
758755

759756

760757
def load_tests(loader, tests, pattern):

0 commit comments

Comments
 (0)