Benchmarking for python functions

benchmark.py

import time

def print_numbers(n):
  for i in xrange(n):
    pass

class BM(object):

    def __init__(self, name):
        self.name = name

    def __enter__(self):
        self.start = time.time()

    def __exit__(self, *args):
        print("{0} took {1}".format(self.name, time.clock() - self.start))

with BM('print_numbers'):
  for i in xrange(10000000):
    print_numbers(i)