Benchmarking helpers

benchmark.py

import time

class Measurement(object):
    def __init__(self, name):
        self.name = name
        return

    def __enter__(self):
        self.t_start = time.time()
        return

    def __exit__(self, exc_type, exc_val, exc_tb):
        t_end = time.time()
        print("{} took {:.3f} seconds".format(self.name, t_end - self.t_start))
        return