python timer in context manager

timer.py

from timeit import default_timer


class Timer(object):
    def __init__(self, name: str, verbose: bool = True):
        self.name = name
        self.verbose = verbose
        self.timer = default_timer

    def __enter__(self):
        self.start = self.timer()
        return self

    def __exit__(self, *args):
        end = self.timer()
        elapsed = (end - self.start) * 1000  # ms
        if self.verbose:
            print(f"{self.name} elapsed time: {elapsed} ms")