| """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() |