python benchmark

benchmark.py

import sys, itertools
from multiprocessing import Pool


def do_permutations(string):
	for i in itertools.permutations(string):
		pass
	return 'Done count permutations of ' + string


if __name__ == '__main__':
	print 'Usage:{0} thread_size task_count'.format(sys.argv[0])
	try:
		thread_size = int(sys.argv[1])
		task_count = int(sys.argv[2])
	except:
		thread_size = 4
		task_count = 4

	print 'Using thread_size:{0} task_count:{1}'.format(thread_size, task_count)
	p = Pool(thread_size)
	s = 'ABCDEFGHIJ'
	tasks = [s] * task_count
	print('GO')
	print(p.map(do_permutations, tasks))