diff options
Diffstat (limited to 'Lib/test/test_cprofile.py')
| -rw-r--r-- | Lib/test/test_cprofile.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/Lib/test/test_cprofile.py b/Lib/test/test_cprofile.py index 1430d22..5c4ec5b 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 @@ -50,8 +51,14 @@ class CProfileTest(ProfileTest): assert_python_ok('-m', 'cProfile', '-m', 'timeit', '-n', '1') +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: |