Get rid of files

Author: pgdr

pyperformance/data-files/benchmarks/bm_pickle_opt/run_benchmark.py

@@ -6,38 +6,23 @@
 
 """
 
-import tempfile
-from pathlib import Path
-import pyperf
 import pickle
 import pickletools
+import pyperf
 
 
-def setup(fname, N):
-    x = {}
-    for i in range(1, N):
-        x[i] = f"ii{i:>07}"
-
-    with open(fname, "wb") as fh:
-        pickle.dump(x, fh, protocol=4)
-
-
-def run(fname):
-    with open(fname, "rb") as fh:
-        p = fh.read()
+def setup(N: int) -> bytes:
+    x = {i: f"ii{i:>07}" for i in range(N)}
+    return pickle.dumps(x, protocol=4)
 
-    s = pickletools.optimize(p)
 
-    with open(fname.with_suffix(".out"), "wb") as fh:
-        fh.write(s)
+def run(p: bytes) -> None:
+    pickletools.optimize(p)
 
 
 if __name__ == "__main__":
     runner = pyperf.Runner()
-    N = 1_000_000
-    with tempfile.TemporaryDirectory() as tmp:
-        tmp_path = Path(tmp)
-        fname = tmp_path / "pickle"
-        setup(fname, N)
-        runner.metadata["description"] = "Pickletools optimize"
-        runner.bench_func("pickle_opt", run, fname)
+    runner.metadata["description"] = "Pickletools optimize"
+    N = 100_000
+    payload = setup(N)
+    runner.bench_func("pickle_opt", run, payload)