FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

gh-131178: Add CLI tests for cProfile by ShauryaDusht · Pull Request #134269 · python/cpython · GitHub

/ cpython Public

gh-131178: Add CLI tests for cProfile #134269

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
ShauryaDusht wants to merge 3 commits into from
Conversation 16 Commits 3 Checks 0 Files changed
Closed

gh-131178: Add CLI tests for cProfile #134269

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
e1e4bc5
gh-131178: Add CLI tests for cProfile
ShauryaDusht May 19, 2025
acbf1f8
lint fixes
ShauryaDusht May 19, 2025
7f55b9f
Fix suggested changes
ShauryaDusht May 19, 2025
File filter

Filter by extension

Filter by extension .py  (1) All 1 file type selected
Viewed files
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Unified
Split
Hide whitespace
Diff view
Unified
Split
Hide whitespace
107 changes: 97 additions & 10 deletions Lib/test/test_cprofile.py
Show comments
View file
Open in desktop
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Show hidden characters
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import sys
import unittest

import os
# rip off all interesting stuff from test_profile
import cProfile
import tempfile
Expand Down Expand Up @@ -151,23 +151,110 @@ def test_bad_descriptor(self):


class TestCommandLine(unittest.TestCase):
def setUp(self):
self.test_script = tempfile.NamedTemporaryFile("w+", suffix=".py", delete=False)
self.test_script.write(textwrap.dedent("""
def simple_function():
sum(range(1000))

if __name__ == "__main__":
simple_function()
"""))
self.test_script.close()

def tearDown(self):
os.unlink(self.test_script.name)

def test_sort(self):
rc, out, err = assert_python_failure('-m', 'cProfile', '-s', 'demo')
rc, _, err = assert_python_failure('-m', 'cProfile', '-s', 'demo')
self.assertGreater(rc, 0)
self.assertIn(b"option -s: invalid choice: 'demo'", err)

def test_valid_sort_options(self):
for sort_opt in ('calls', 'cumulative', 'cumtime', 'filename',
'ncalls', 'pcalls', 'line', 'name', 'nfl',
'stdname', 'time', 'tottime'):
rc, out, _ = assert_python_ok('-m', 'cProfile', '-s', sort_opt, self.test_script.name)
self.assertEqual(rc, 0)
self.assertIn(b"function calls", out)

def test_outfile(self):
with tempfile.NamedTemporaryFile(suffix='.prof', delete=False) as outfile:
outfile_name = outfile.name
try:
rc, _, __ = assert_python_ok('-m', 'cProfile', '-o', outfile_name, self.test_script.name)
self.assertEqual(rc, 0)
self.assertTrue(os.path.exists(outfile_name))
self.assertGreater(os.path.getsize(outfile_name), 0)
finally:
if os.path.exists(outfile_name):
os.unlink(outfile_name)

def test_no_arguments(self):
rc, out, _ = assert_python_failure('-m', 'cProfile')
self.assertGreater(rc, 0)
self.assertIn(b'Usage:', out)

def test_help_option(self):
rc, out, _ = assert_python_ok('-m', 'cProfile', '--help')
self.assertEqual(rc, 0)
self.assertIn(b"Usage:", out)

def test_version_output(self):
rc, out, _ = assert_python_ok('-m', 'cProfile', self.test_script.name)
self.assertEqual(rc, 0)
self.assertIn(os.path.basename(self.test_script.name).encode(), out)

def test_run_command_line_module(self):
rc, out, _ = assert_python_ok('-m', 'cProfile', '-m', 'timeit', '-n', '1', 'pass')
self.assertEqual(rc, 0)
self.assertIn(b"function calls", out)

def test_profile_script_importing_main(self):
"""Check that scripts that reference __main__ see their own namespace
when being profiled."""
with tempfile.NamedTemporaryFile("w+", delete_on_close=False) as f:
with tempfile.NamedTemporaryFile("w+", suffix='.py', delete=False) as f:
f.write(textwrap.dedent("""\
class Foo:
pass
import __main__
assert Foo == __main__.Foo
def test_func():
x = 1 + 1
return x

if __name__ == "__main__":
test_func()
"""))
f.close()
assert_python_ok('-m', "cProfile", f.name)
try:
rc, out, _ = assert_python_ok('-m', "cProfile", f.name)
self.assertEqual(rc, 0)
self.assertIn(b"function calls", out)
self.assertIn(b"test_func", out)
finally:
os.unlink(f.name)

def test_output_format(self):
rc, out, _ = assert_python_ok('-m', 'cProfile', self.test_script.name)
self.assertEqual(rc, 0)

self.assertRegex(out, rb'\d+ function calls in \d+\.\d+ seconds')

self.assertIn(b'Ordered by:', out)
self.assertIn(b'ncalls', out)
self.assertIn(b'tottime', out)
self.assertIn(b'percall', out)
self.assertIn(b'cumtime', out)
self.assertIn(b'filename:lineno(function)', out)
self.assertIn(b'simple_function', out)

def test_different_sort_outputs(self):
rc1, out1, _ = assert_python_ok('-m', 'cProfile', '-s', 'time', self.test_script.name)
rc2, out2, _ = assert_python_ok('-m', 'cProfile', '-s', 'cumulative', self.test_script.name)

self.assertEqual(rc1, 0)
self.assertEqual(rc2, 0)

self.assertRegex(out1.decode(), r'Ordered by: (?:internal )?time')
self.assertRegex(out2.decode(), r'Ordered by: cumulative')

self.assertIn(b'simple_function', out1)
self.assertIn(b'simple_function', out2)


def main():
Expand Down
Back | FazBrowse Home | New Git URL