1
0
mirror of https://github.com/wting/autojump synced 2024-10-27 20:34:07 +00:00

Add profiling code.

This commit is contained in:
jez 2011-05-26 18:22:00 +00:00
parent 34419c3eae
commit 7a3b8a8e7f

26
profile/profile.py Normal file
View File

@ -0,0 +1,26 @@
from __future__ import division, print_function
import cProfile, sys, imp, os, pstats
autojump = imp.load_source('autojump', 'autojump')
"""Profile the total time taken for autojump to generate completions as a
function of pattern length. This file must be run from the project root."""
if os.path.exists('./profile/autojump_py'):
autojump.CONFIG_DIR = './profile'
if len(sys.argv) > 1:
outfile = open(sys.argv[1], 'w')
else:
outfile = open('profile_results', 'w')
outfile.write('Pattern length\tTime taken/s\n')
# For maximum running time, we don't want to match any files.
test_search = '#' * 10
for i in range(0, 10):
autojump.argv = ['', '--completion', test_search[:i+1]]
cProfile.run('autojump.shell_utility()', 'shellprof')
p = pstats.Stats('shellprof')
outfile.write("%s\t%s\n"% (i + 1, p.total_tt))
p.sort_stats('time')
p.print_stats(10)