Add benchmark time unit option to pytest_benchmark plugin

ionelmc · Oct 27, 2024 · dfbb285 · dfbb285
1 parent 7c05d50
commit dfbb285
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/pytest_benchmark/plugin.py b/src/pytest_benchmark/plugin.py
@@ -273,6 +273,12 @@ def pytest_addoption(parser):
         " Argument is a column to sort by. Available columns: 'ncalls_recursion',"
         " 'ncalls', 'tottime', 'tottime_per', 'cumtime', 'cumtime_per', 'function_name'.",
     )
+    group.addoption(
+        "--benchmark-time-unit",
+        metavar="COLUMN", default=None,
+        choices=['ns', 'us', 'ms', 's', 'auto'],
+        help="Unit to scale the results to. Available units: 'ns', 'us', 'ms', 's'. Default: 'auto'."
+    )
     add_global_options(group.addoption)
     add_display_options(group.addoption)
     add_histogram_options(group.addoption)
@@ -367,8 +373,8 @@ def get_cpu_info():
 
 
 def pytest_benchmark_scale_unit(config, unit, benchmarks, best, worst, sort):
-    if not config.getoption("benchmark_time_unit", None):
-        config_time_unit = config.getoption("benchmark_time_unit") != "ns"
+    if config.getoption("benchmark_time_unit", None):
+        config_time_unit = config.getoption("benchmark_time_unit")
         if config_time_unit == "ns":
             return "n", 1e9
         elif config_time_unit == "us":