Added benchmarks

1 files changed, 32 insertions, 0 deletions

diff --git a/benchmarks/by-n/plot.py b/benchmarks/by-n/plot.py
new file mode 100755
index 0000000..ba96a0c
--- /dev/null
+++ b/benchmarks/by-n/plot.py
@@ -0,0 +1,32 @@
+#!/bin/env python3
+
+import matplotlib.pyplot as plt
+import json
+import os, glob
+
+
+def render(file):
+    data = json.load(open(file, "r"))["results"]
+    basename = os.path.basename(file).split(".")[0]
+    reducers = list({p["parameters"]["REDUCER"] for p in data})
+    for reducer in reducers:
+        labels = [
+            p["parameters"]["N"]
+            for p in data
+            if p["parameters"]["REDUCER"] == reducer
+        ]
+        times = [
+            p["median"] for p in data if p["parameters"]["REDUCER"] == reducer
+        ]
+        plt.plot(times)
+        plt.xticks(range(len(labels)), labels)
+    plt.legend(reducers)
+    plt.title(f"reducer comparison for {basename}")
+    plt.xlabel("N")
+    plt.ylabel("Time (s)")
+    plt.savefig(f"{basename}.png")
+    plt.close()
+
+
+for file in glob.glob("*.json"):
+    render(file)