diff options
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/pstats.py | 3 | ||||
-rw-r--r-- | Lib/test/test_pstats.py | 26 |
2 files changed, 28 insertions, 1 deletions
diff --git a/Lib/pstats.py b/Lib/pstats.py index 5a2f2d0..39f1b64 100644 --- a/Lib/pstats.py +++ b/Lib/pstats.py @@ -511,7 +511,8 @@ def add_callers(target, source): new_callers[func] = caller for func, caller in source.items(): if func in new_callers: - new_callers[func] = caller + new_callers[func] + new_callers[func] = tuple([i[0] + i[1] for i in + zip(caller, new_callers[func])]) else: new_callers[func] = caller return new_callers diff --git a/Lib/test/test_pstats.py b/Lib/test/test_pstats.py new file mode 100644 index 0000000..660316b --- /dev/null +++ b/Lib/test/test_pstats.py @@ -0,0 +1,26 @@ +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.""" + 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)}) + + +def test_main(): + test_support.run_unittest( + AddCallersTestCase + ) + + +if __name__ == "__main__": + test_main() |