Tests: add world load benchmark (#2768)

This commit is contained in:
Fabian Dill
2024-02-04 09:09:07 +01:00
committed by GitHub
parent 33237bd5c0
commit 6c19bc42bb
6 changed files with 181 additions and 128 deletions

23
test/benchmark/time_it.py Normal file
View File

@@ -0,0 +1,23 @@
import time
class TimeIt:
def __init__(self, name: str, time_logger=None):
self.name = name
self.logger = time_logger
self.timer = None
self.end_timer = None
def __enter__(self):
self.timer = time.perf_counter()
return self
@property
def dif(self):
return self.end_timer - self.timer
def __exit__(self, exc_type, exc_val, exc_tb):
if not self.end_timer:
self.end_timer = time.perf_counter()
if self.logger:
self.logger.info(f"{self.dif:.4f} seconds in {self.name}.")