#!/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)