#!/usr/bin/python
# A simple wrapper for the timeit module.
import timeit
def Timeit(func,number=10000,module="__main__"):
""" A wrapper which can
be used to time any function """
name = func.__name__
t = timeit.Timer("%s()"%name, "from %s import %s" % (module, name))
return "%.2f usec/pass" % (1000000*t.timeit(number=number)/number)
if __name__=="__main__":
from mymodule import test
# Using wrapper
print Timeit(test)
# Directly using timeit
t = timeit.Timer("test()", "from __main__ import test")
print "%.2f usec/pass" % (1000000*t.timeit(number=10000)/10000)
