mirror of
https://github.com/sharkdp/hyperfine.git
synced 2026-10-12 02:22:35 +00:00
60 lines
1.6 KiB
Python
Executable File
60 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# /// script
|
|
# requires-python = ">=3.10"
|
|
# dependencies = [
|
|
# "scipy",
|
|
# ]
|
|
# ///
|
|
|
|
"""This script performs Welch's t-test on a JSON export file with two
|
|
benchmark results to test for a difference in their population means."""
|
|
|
|
import argparse
|
|
import math
|
|
|
|
from scipy import stats
|
|
|
|
from plot_utils import METRICS, add_metric_argument, load_results
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("file", help="JSON file with two benchmark results")
|
|
add_metric_argument(parser)
|
|
args = parser.parse_args()
|
|
|
|
metric, [results] = load_results(parser, [args.file], args.metric)
|
|
|
|
if len(results) != 2:
|
|
parser.error("the input file must contain exactly two benchmarks")
|
|
|
|
a, b = (x.get("name", x["command"]) for x in results[:2])
|
|
X, Y = ([m[metric]["value"] for m in x["measurements"]] for x in results[:2])
|
|
if min(len(X), len(Y)) < 2:
|
|
parser.error("Welch's t-test requires at least two measurements per benchmark")
|
|
|
|
print(f"Metric: {METRICS[metric].label}")
|
|
print(f"Command 1: {a}")
|
|
print(f"Command 2: {b}\n")
|
|
|
|
t, p = stats.ttest_ind(X, Y, equal_var=False)
|
|
if not math.isfinite(p):
|
|
parser.error("Welch's t-test is undefined for these measurements")
|
|
th = 0.05
|
|
dispose = p < th
|
|
print(f"t = {t:.3}, p = {p:.3}")
|
|
print()
|
|
|
|
if dispose:
|
|
print(
|
|
f"A statistically significant difference in mean value was detected (p < {th})."
|
|
)
|
|
else:
|
|
print(
|
|
f"No statistically significant difference in mean value was detected (p >= {th})."
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|