pretty gifff

metrics/utils.py

@@ -1,22 +1,23 @@
 __all__ = [
     "get_time_of_pre_cycle_setup",
+    "attempts_to_average"
 ]
 
 import io
 from contextlib import redirect_stdout
 import time
+from decimal import Decimal, ROUND_HALF_UP
 
 from argenta import App
 
 
-def get_time_of_pre_cycle_setup(app: App) -> float: 
-    """
-    Public. Return time of pre cycle setup
-    :param app: app instance for testing time of pre cycle setup
-    :return: time of pre cycle setup as float
-    """
+def get_time_of_pre_cycle_setup(app: App) -> float:
     start = time.monotonic()
     with redirect_stdout(io.StringIO()):
         app._pre_cycle_setup()  # pyright: ignore[reportPrivateUsage]
     end = time.monotonic()
     return end - start
+
+
+def attempts_to_average(bench_attempts: list[float], iterations: int) -> Decimal:
+    return Decimal(sum(bench_attempts) / iterations).quantize(Decimal("0.00001"), rounding=ROUND_HALF_UP)