| import unittest |
| from test import test_support |
| import pstats |
| |
| |
| |
| class AddCallersTestCase(unittest.TestCase): |
| """Tests for pstats.add_callers helper.""" |
| |
| def test_combine_results(self): |
| """pstats.add_callers should combine the call results of both target |
| and source by adding the call time. See issue1269.""" |
| # new format: used by the cProfile module |
| target = {"a": (1, 2, 3, 4)} |
| source = {"a": (1, 2, 3, 4), "b": (5, 6, 7, 8)} |
| new_callers = pstats.add_callers(target, source) |
| self.assertEqual(new_callers, {'a': (2, 4, 6, 8), 'b': (5, 6, 7, 8)}) |
| # old format: used by the profile module |
| target = {"a": 1} |
| source = {"a": 1, "b": 5} |
| new_callers = pstats.add_callers(target, source) |
| self.assertEqual(new_callers, {'a': 2, 'b': 5}) |
| |
| |
| def test_main(): |
| test_support.run_unittest( |
| AddCallersTestCase |
| ) |
| |
| |
| if __name__ == "__main__": |
| test_main() |
