"""Test suite for the cProfile module.""" import sys import unittest # rip off all interesting stuff from test_profile import cProfile from test.test_profile import ProfileTest, regenerate_expected_output from test.support.script_helper import assert_python_failure from test import support class CProfileTest(ProfileTest): profilerclass = cProfile.Profile profilermodule = cProfile expected_max_output = "{built-in method builtins.max}" def get_expected_output(self): return _ProfileOutput def test_bad_counter_during_dealloc(self): # bpo-3895 import _lsprof with support.catch_unraisable_exception() as cm: obj = _lsprof.Profiler(lambda: int) obj.enable() obj.disable() obj.clear() self.assertEqual(cm.unraisable.exc_type, TypeError) def test_crash_with_not_enough_args(self): # gh-126220 import _lsprof for profile in [_lsprof.Profiler(), cProfile.Profile()]: for method in [ "_pystart_callback", "_pyreturn_callback", "_ccall_callback", "_creturn_callback", ]: with self.subTest(profile=profile, method=method): method_obj = getattr(profile, method) with self.assertRaises(TypeError): method_obj() # should not crash def test_evil_external_timer(self): # gh-120289 # Disabling profiler in external timer should not crash import _lsprof class EvilTimer(): def __init__(self, disable_count): self.count = 0 self.disable_count = disable_count def __call__(self): self.count += 1 if self.count == self.disable_count: profiler_with_evil_timer.disable() return self.count # this will trigger external timer to disable profiler at # call event - in initContext in _lsprof.c with support.catch_unraisable_exception() as cm: profiler_with_evil_timer = _lsprof.Profiler(EvilTimer(1)) profiler_with_evil_timer.enable() # Make a call to trigger timer (lambda: None)() profiler_with_evil_timer.disable() profiler_with_evil_timer.clear() self.assertEqual(cm.unraisable.exc_type, RuntimeError) # this will trigger external timer to disable profiler at # return event - in Stop in _lsprof.c with support.catch_unraisable_exception() as cm: profiler_with_evil_timer = _lsprof.Profiler(EvilTimer(2)) profiler_with_evil_timer.enable() # Make a call to trigger timer (lambda: None)() profiler_with_evil_timer.disable() profiler_with_evil_timer.clear() self.assertEqual(cm.unraisable.exc_type, RuntimeError) def test_profile_enable_disable(self): prof = self.profilerclass() # Make sure we clean ourselves up if the test fails for some reason. self.addCleanup(prof.disable) prof.enable() self.assertEqual( sys.monitoring.get_tool(sys.monitoring.PROFILER_ID), "cProfile") prof.disable() self.assertIs(sys.monitoring.get_tool(sys.monitoring.PROFILER_ID), None) def test_profile_as_context_manager(self): prof = self.profilerclass() # Make sure we clean ourselves up if the test fails for some reason. self.addCleanup(prof.disable) with prof as __enter__return_value: # profile.__enter__ should return itself. self.assertIs(prof, __enter__return_value) # profile should be set as the global profiler inside the # with-block self.assertEqual( sys.monitoring.get_tool(sys.monitoring.PROFILER_ID), "cProfile") # profile shouldn't be set once we leave the with-block. self.assertIs(sys.monitoring.get_tool(sys.monitoring.PROFILER_ID), None) def test_second_profiler(self): pr = self.profilerclass() pr2 = self.profilerclass() pr.enable() self.assertRaises(ValueError, pr2.enable) pr.disable() def test_throw(self): """ gh-106152 generator.throw() should trigger a call in cProfile In the any() call below, there should be two entries for the generator: * one for the call to __next__ which gets a True and terminates any * one when the generator is garbage collected which will effectively do a throw. """ pr = self.profilerclass() pr.enable() any(a == 1 for a in (1, 2)) pr.disable() pr.create_stats() for func, (cc, nc, _, _, _) in pr.stats.items(): if func[2] == "": self.assertEqual(cc, 1) self.assertEqual(nc, 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 main(): if '-r' not in sys.argv: unittest.main() else: regenerate_expected_output(__file__, CProfileTest) # Don't remove this comment. Everything below it is auto-generated. #--cut-------------------------------------------------------------------------- _ProfileOutput = {} _ProfileOutput['print_stats'] = """\ 28 0.028 0.001 0.028 0.001 profilee.py:110(__getattr__) 1 0.270 0.270 1.000 1.000 profilee.py:25(testfunc) 23/3 0.150 0.007 0.170 0.057 profilee.py:35(factorial) 20 0.020 0.001 0.020 0.001 profilee.py:48(mul) 2 0.040 0.020 0.600 0.300 profilee.py:55(helper) 4 0.116 0.029 0.120 0.030 profilee.py:73(helper1) 2 0.000 0.000 0.140 0.070 profilee.py:84(helper2_indirect) 8 0.312 0.039 0.400 0.050 profilee.py:88(helper2) 8 0.064 0.008 0.080 0.010 profilee.py:98(subhelper)""" _ProfileOutput['print_callers'] = """\ profilee.py:110(__getattr__) <- 16 0.016 0.016 profilee.py:98(subhelper) profilee.py:25(testfunc) <- 1 0.270 1.000 :1() profilee.py:35(factorial) <- 1 0.014 0.130 profilee.py:25(testfunc) 20/3 0.130 0.147 profilee.py:35(factorial) 2 0.006 0.040 profilee.py:84(helper2_indirect) profilee.py:48(mul) <- 20 0.020 0.020 profilee.py:35(factorial) profilee.py:55(helper) <- 2 0.040 0.600 profilee.py:25(testfunc) profilee.py:73(helper1) <- 4 0.116 0.120 profilee.py:55(helper) profilee.py:84(helper2_indirect) <- 2 0.000 0.140 profilee.py:55(helper) profilee.py:88(helper2) <- 6 0.234 0.300 profilee.py:55(helper) 2 0.078 0.100 profilee.py:84(helper2_indirect) profilee.py:98(subhelper) <- 8 0.064 0.080 profilee.py:88(helper2) {built-in method builtins.hasattr} <- 4 0.000 0.004 profilee.py:73(helper1) 8 0.000 0.008 profilee.py:88(helper2) {built-in method sys.exception} <- 4 0.000 0.000 profilee.py:73(helper1) {method 'append' of 'list' objects} <- 4 0.000 0.000 profilee.py:73(helper1)""" _ProfileOutput['print_callees'] = """\ :1() -> 1 0.270 1.000 profilee.py:25(testfunc) profilee.py:110(__getattr__) -> profilee.py:25(testfunc) -> 1 0.014 0.130 profilee.py:35(factorial) 2 0.040 0.600 profilee.py:55(helper) profilee.py:35(factorial) -> 20/3 0.130 0.147 profilee.py:35(factorial) 20 0.020 0.020 profilee.py:48(mul) profilee.py:48(mul) -> profilee.py:55(helper) -> 4 0.116 0.120 profilee.py:73(helper1) 2 0.000 0.140 profilee.py:84(helper2_indirect) 6 0.234 0.300 profilee.py:88(helper2) profilee.py:73(helper1) -> 4 0.000 0.004 {built-in method builtins.hasattr} profilee.py:84(helper2_indirect) -> 2 0.006 0.040 profilee.py:35(factorial) 2 0.078 0.100 profilee.py:88(helper2) profilee.py:88(helper2) -> 8 0.064 0.080 profilee.py:98(subhelper) profilee.py:98(subhelper) -> 16 0.016 0.016 profilee.py:110(__getattr__) {built-in method builtins.hasattr} -> 12 0.012 0.012 profilee.py:110(__getattr__)""" if __name__ == "__main__": main()