| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124 |
- """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 = _lsprof.Profiler(1)
- obj.disable()
- obj.clear()
- self.assertEqual(cm.unraisable.exc_type, TypeError)
- 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.assertIs(sys.getprofile(), prof)
- prof.disable()
- self.assertIs(sys.getprofile(), 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.assertIs(sys.getprofile(), prof)
- # 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 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.exc_info} <- 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()
|
