• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1import unittest
2
3from test import support
4from io import StringIO
5from pstats import SortKey
6
7import pstats
8import cProfile
9
10class AddCallersTestCase(unittest.TestCase):
11    """Tests for pstats.add_callers helper."""
12
13    def test_combine_results(self):
14        # pstats.add_callers should combine the call results of both target
15        # and source by adding the call time. See issue1269.
16        # new format: used by the cProfile module
17        target = {"a": (1, 2, 3, 4)}
18        source = {"a": (1, 2, 3, 4), "b": (5, 6, 7, 8)}
19        new_callers = pstats.add_callers(target, source)
20        self.assertEqual(new_callers, {'a': (2, 4, 6, 8), 'b': (5, 6, 7, 8)})
21        # old format: used by the profile module
22        target = {"a": 1}
23        source = {"a": 1, "b": 5}
24        new_callers = pstats.add_callers(target, source)
25        self.assertEqual(new_callers, {'a': 2, 'b': 5})
26
27
28class StatsTestCase(unittest.TestCase):
29    def setUp(self):
30        stats_file = support.findfile('pstats.pck')
31        self.stats = pstats.Stats(stats_file)
32
33    def test_add(self):
34        stream = StringIO()
35        stats = pstats.Stats(stream=stream)
36        stats.add(self.stats, self.stats)
37
38    def test_sort_stats_int(self):
39        valid_args = {-1: 'stdname',
40                      0: 'calls',
41                      1: 'time',
42                      2: 'cumulative'}
43        for arg_int, arg_str in valid_args.items():
44            self.stats.sort_stats(arg_int)
45            self.assertEqual(self.stats.sort_type,
46                             self.stats.sort_arg_dict_default[arg_str][-1])
47
48    def test_sort_stats_string(self):
49        for sort_name in ['calls', 'ncalls', 'cumtime', 'cumulative',
50                    'filename', 'line', 'module', 'name', 'nfl', 'pcalls',
51                    'stdname', 'time', 'tottime']:
52            self.stats.sort_stats(sort_name)
53            self.assertEqual(self.stats.sort_type,
54                             self.stats.sort_arg_dict_default[sort_name][-1])
55
56    def test_sort_stats_partial(self):
57        sortkey = 'filename'
58        for sort_name in ['f', 'fi', 'fil', 'file', 'filen', 'filena',
59                           'filenam', 'filename']:
60            self.stats.sort_stats(sort_name)
61            self.assertEqual(self.stats.sort_type,
62                             self.stats.sort_arg_dict_default[sortkey][-1])
63
64    def test_sort_starts_mix(self):
65        self.assertRaises(TypeError, self.stats.sort_stats,
66                          'calls',
67                          SortKey.TIME)
68        self.assertRaises(TypeError, self.stats.sort_stats,
69                          SortKey.TIME,
70                          'calls')
71
72    def test_get_stats_profile(self):
73        def pass1(): pass
74        def pass2(): pass
75        def pass3(): pass
76
77        pr = cProfile.Profile()
78        pr.enable()
79        pass1()
80        pass2()
81        pass3()
82        pr.create_stats()
83        ps = pstats.Stats(pr)
84
85        stats_profile = ps.get_stats_profile()
86        funcs_called = set(stats_profile.func_profiles.keys())
87        self.assertIn('pass1', funcs_called)
88        self.assertIn('pass2', funcs_called)
89        self.assertIn('pass3', funcs_called)
90
91    def test_SortKey_enum(self):
92        self.assertEqual(SortKey.FILENAME, 'filename')
93        self.assertNotEqual(SortKey.FILENAME, SortKey.CALLS)
94
95if __name__ == "__main__":
96    unittest.main()
97