| """Test suite for the cProfile module.""" |
| |
| import sys |
| import unittest |
| |
| # rip off all interesting stuff from test_profile |
| import profiling.tracing as cProfile |
| import tempfile |
| import textwrap |
| from test.test_profile import ProfileTest, regenerate_expected_output |
| from test.support.script_helper import assert_python_failure, assert_python_ok |
| 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 |
| """ |
| |
| def gen(): |
| yield |
| |
| pr = self.profilerclass() |
| pr.enable() |
| g = gen() |
| try: |
| g.throw(SyntaxError) |
| except SyntaxError: |
| pass |
| pr.disable() |
| pr.create_stats() |
| |
| self.assertTrue(any("throw" in func[2] for func in pr.stats.keys())), |
| |
| def test_bad_descriptor(self): |
| # gh-132250 |
| # cProfile should not crash when the profiler callback fails to locate |
| # the actual function of a method. |
| with self.profilerclass() as prof: |
| with self.assertRaises(TypeError): |
| bytes.find(str()) |
| |
| |
| 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_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: |
| f.write(textwrap.dedent("""\ |
| class Foo: |
| pass |
| import __main__ |
| assert Foo == __main__.Foo |
| """)) |
| f.close() |
| assert_python_ok('-m', "cProfile", f.name) |
| |
| |
| 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 <string>:1(<module>) |
| 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'] = """\ |
| <string>:1(<module>) -> 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() |
