from datetime import datetime class Benchmark: def __init__(self): self.start() def _timestamp(self) -> float: """ Generate a timestamp. """ return float(datetime.utcnow().timestamp()) def start(self) -> int: """ Start a benchmark. """ self.current = self._timestamp() return self.current def end(self): """ Return the diff between now - start(). """ n = self._timestamp() - self.current self.current = float(0) return n