diff options
Diffstat (limited to 'Lib/test/test_cprofile.py')
| -rw-r--r-- | Lib/test/test_cprofile.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/Lib/test/test_cprofile.py b/Lib/test/test_cprofile.py index 2fd67ee..406d703 100644 --- a/Lib/test/test_cprofile.py +++ b/Lib/test/test_cprofile.py @@ -2,6 +2,7 @@ import sys from test.support import run_unittest, TESTFN, unlink +import unittest # rip off all interesting stuff from test_profile import cProfile @@ -76,9 +77,14 @@ class CProfileTest(ProfileTest): # profile shouldn't be set once we leave the with-block. self.assertIs(sys.getprofile(), None) +class TestCommandLine(unittest.TestCase): + def test_sort(self): + rc, out, err = assert_python_failure('-m', 'cProfile', '-s', 'demo') + self.assertGreater(rc, 0) + self.assertIn(b"option -s: invalid choice: 'demo'", err) def test_main(): - run_unittest(CProfileTest) + run_unittest(CProfileTest, TestCommandLine) def main(): if '-r' not in sys.argv: |