Python benchmark decorator

benchmark_decorator.py

class benchmark(object):
    """Run a function as benchmark for several times.

    Usage:
     >>> @benchmark(times=1)  # times is a optional parameter.
     >>> def awesome_benchmark(arg1, arg2):
         >>> # do awesome benchmarks.
    """
    def __init__(self, **kwargs):
        self.times = kwargs.get('times', 1)

    def __call__(self, func):
        def benchmark_func(*args):
            for i in range(self.times):
                func(*args)
        return benchmark_func