bench

metrics/benchmarks/core/exceptions.py

@@ -12,3 +12,9 @@ class BenchmarksNotFound(Exception):
 
     def __str__(self):
         return f"Benchmarks with type '{self.type_}' not found"
+
+class BenchmarksWithSameNameAlreadyExists(Exception):
+    def __init__(self, benchmark_name: str):
+        self.benchmark_name = benchmark_name
+    def __str__(self):
+        return f"Benchmarks with name '{self.benchmark_name}' already exists"

metrics/benchmarks/core/models.py

@@ -13,8 +13,7 @@ import gc
 import statistics
 from typing import Callable, override
 
-from .exceptions import BenchmarkNotFound, BenchmarksNotFound
-
+from .exceptions import BenchmarkNotFound, BenchmarksNotFound, BenchmarksWithSameNameAlreadyExists
 
 FuncForBenchmark = Callable[[], None]
 MILLISECONDS_IN_SECONDS = 1000
@@ -35,6 +34,8 @@ class BenchmarkResult:
 @dataclass(frozen=True, slots=True)
 class BenchmarkGroupResult:
     type_: str
+    iterations: int
+    is_gc_disabled: bool
     benchmark_results: list[BenchmarkResult]
 
 
@@ -107,8 +108,11 @@ class Benchmarks:
                 name=func.__name__,
                 description=description or f'description for {func.__name__} with type {type_}',
             )
-            self._benchmarks.append(benchmark)
+            if self._benchmarks_paired_by_name.get(func.__name__):
+                raise BenchmarksWithSameNameAlreadyExists(func.__name__)
+
             self._benchmarks_paired_by_name[func.__name__] = benchmark
+            self._benchmarks.append(benchmark)
             self._benchmarks_grouped_by_type.setdefault(type_, []).append(benchmark)
             return func
         return decorator
@@ -145,6 +149,8 @@ class Benchmarks:
 
         return BenchmarkGroupResult(
             type_=type_,
+            iterations=iterations,
+            is_gc_disabled=is_gc_disabled,
             benchmark_results=benchmark_results
         )
 

metrics/benchmarks/most_similar_command.py

@@ -1,14 +1,11 @@
 __all__ = [
     "benchmark_few_commands",
-    "benchmark_many_commands",
+    "benchmark_many_commands_most_similar",
     "benchmark_many_aliases",
     "benchmark_partial_match",
     "benchmark_extreme_commands"
 ]
 
-import io
-from contextlib import redirect_stdout
-
 from argenta import App
 from argenta.command.models import Command
 from argenta.response import Response
@@ -39,7 +36,7 @@ def benchmark_few_commands() -> None:
 
 
 @benchmarks.register(type_="most_similar_command", description="Many commands (50 commands, no match)")
-def benchmark_many_commands() -> None:
+def benchmark_many_commands_most_similar() -> None:
     app = setup_app_with_commands(50)
     app._most_similar_command("unknown")