diff options
Diffstat (limited to 'lib/python2.7/test/test_pstats.py')
-rw-r--r-- | lib/python2.7/test/test_pstats.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/lib/python2.7/test/test_pstats.py b/lib/python2.7/test/test_pstats.py new file mode 100644 index 0000000..6ad9418 --- /dev/null +++ b/lib/python2.7/test/test_pstats.py @@ -0,0 +1,32 @@ +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() |