pytest-codspeed 3.0.0b1__tar.gz → 3.0.0b2__tar.gz

{pytest_codspeed-3.0.0b1 → pytest_codspeed-3.0.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pytest-codspeed
-Version: 3.0.0b1
+Version: 3.0.0b2
 Summary: Pytest plugin to create CodSpeed benchmarks
 Project-URL: Homepage, https://codspeed.io/
 Project-URL: Documentation, https://docs.codspeed.io/

{pytest_codspeed-3.0.0b1 → pytest_codspeed-3.0.0b2}/src/pytest_codspeed/__init__.py

@@ -1,6 +1,6 @@
-__version__ = "3.0.0b1"
+__version__ = "3.0.0b2"
 # We also have the semver version since __version__ is not semver compliant
-__semver_version__ = "3.0.0-b1"
+__semver_version__ = "3.0.0-b2"
 
 from .plugin import BenchmarkFixture
 

{pytest_codspeed-3.0.0b1 → pytest_codspeed-3.0.0b2}/src/pytest_codspeed/instruments/walltime.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from dataclasses import asdict, dataclass
 from math import ceil
-from statistics import mean, median, stdev
+from statistics import mean, quantiles, stdev
 from time import get_clock_info, perf_counter_ns
 from typing import TYPE_CHECKING
 
@@ -40,8 +40,10 @@ class BenchmarkConfig:
             if config.warmup_time_ns is not None
             else DEFAULT_WARMUP_TIME_NS,
             min_round_time_ns=DEFAULT_MIN_ROUND_TIME_NS,
-            max_time_ns=DEFAULT_MAX_TIME_NS,
-            max_rounds=config.max_rounds or None,
+            max_time_ns=config.max_time_ns
+            if config.max_time_ns is not None
+            else DEFAULT_MAX_TIME_NS,
+            max_rounds=config.max_rounds,
         )
 
 
@@ -51,7 +53,9 @@ class BenchmarkStats:
     max_ns: float
     stdev_ns: float
     mean_ns: float
+    q1_ns: float
     median_ns: float
+    q3_ns: float
 
     rounds: int
     total_time: float
@@ -71,12 +75,15 @@ class BenchmarkStats:
     ) -> BenchmarkStats:
         stdev_ns = stdev(times_ns) if len(times_ns) > 1 else 0
         mean_ns = mean(times_ns)
+        q1_ns, median_ns, q3_ns = quantiles(times_ns, n=4)
         return cls(
             min_ns=min(times_ns),
             max_ns=max(times_ns),
             stdev_ns=stdev_ns,
             mean_ns=mean_ns,
-            median_ns=median(times_ns),
+            median_ns=median_ns,
+            q1_ns=q1_ns,
+            q3_ns=q3_ns,
             rounds=rounds,
             total_time=total_time,
             outlier_rounds=sum(1 for t in times_ns if abs(t - mean_ns) > stdev_ns),
@@ -125,6 +132,7 @@ def run_benchmark(
         rounds = int(config.max_time_ns / round_time_ns)
     else:
         rounds = config.max_rounds
+    rounds = max(1, rounds)
 
     # Benchmark
     iter_range = range(iter_per_round)

{pytest_codspeed-3.0.0b1 → pytest_codspeed-3.0.0b2}/src/pytest_codspeed/plugin.py

@@ -61,6 +61,15 @@ def pytest_addoption(parser: pytest.Parser):
             "only for walltime mode"
         ),
     )
+    group.addoption(
+        "--codspeed-max-time",
+        action="store",
+        type=float,
+        help=(
+            "The maximum time to run a benchmark for (in seconds), "
+            "only for walltime mode"
+        ),
+    )
     group.addoption(
         "--codspeed-max-rounds",
         action="store",
@@ -75,6 +84,7 @@ def pytest_addoption(parser: pytest.Parser):
 @dataclass(frozen=True)
 class CodSpeedConfig:
     warmup_time_ns: int | None = None
+    max_time_ns: int | None = None
     max_rounds: int | None = None
 
     @classmethod
@@ -83,9 +93,12 @@ class CodSpeedConfig:
         warmup_time_ns = (
             int(warmup_time * 1_000_000_000) if warmup_time is not None else None
         )
+        max_time = config.getoption("--codspeed-max-time", None)
+        max_time_ns = int(max_time * 1_000_000_000) if max_time is not None else None
         return cls(
             warmup_time_ns=warmup_time_ns,
             max_rounds=config.getoption("--codspeed-max-rounds", None),
+            max_time_ns=max_time_ns,
         )
 
 
@@ -248,15 +261,17 @@ def _measure(
 ) -> T:
     is_gc_enabled = gc.isenabled()
     if is_gc_enabled:
-        gc.collect()
-        gc.disable()
+        pass
+        # gc.collect()
+        # gc.disable()
     try:
         uri, name = get_git_relative_uri_and_name(nodeid, config.rootpath)
         return plugin.instrument.measure(name, uri, fn, *args, **kwargs)
     finally:
         # Ensure GC is re-enabled even if the test failed
         if is_gc_enabled:
-            gc.enable()
+            pass
+            # gc.enable()
 
 
 def wrap_runtest(