Accumulate times

This commit is contained in:
Michel Breyer
2021-09-12 17:55:42 +02:00
parent 732effa39f
commit f2a12efb0f
3 changed files with 8 additions and 2 deletions

View File

@@ -6,6 +6,7 @@ class Timer:
def __init__(self, name):
self.name = name
self.timers.setdefault(name, 0)
def __enter__(self):
self.start()
@@ -14,9 +15,13 @@ class Timer:
def __exit__(self, *exc_info):
self.stop()
@classmethod
def reset(cls):
cls.timers = dict()
def start(self):
self.tic = time.perf_counter()
def stop(self):
elapsed_time = time.perf_counter() - self.tic
self.timers[self.name] = elapsed_time
self.timers[self.name] += elapsed_time