Add a benchmark to measure gc traversal (#244)

pablogsal · web-flow · commit bd386adea965 · 2022-11-07T17:08:21.000Z
diff --git a/pyperformance/data-files/benchmarks/bm_gc_traversal/pyproject.toml b/pyperformance/data-files/benchmarks/bm_gc_traversal/pyproject.toml
@@ -0,0 +1,9 @@
+[project]
+name = "pyperformance_bm_gc_traversal"
+requires-python = ">=3.8"
+dependencies = ["pyperf"]
+urls = {repository = "https://github.com/python/pyperformance"}
+dynamic = ["version"]
+
+[tool.pyperformance]
+name = "gc_traversal"
diff --git a/pyperformance/data-files/benchmarks/bm_gc_traversal/run_benchmark.py b/pyperformance/data-files/benchmarks/bm_gc_traversal/run_benchmark.py
@@ -0,0 +1,37 @@
+import pyperf
+import gc
+
+N_LEVELS = 1000
+
+
+def create_recursive_containers(n_levels):
+
+    current_list = []
+    for n in range(n_levels):
+        new_list = [None] * n
+        for index in range(n):
+            new_list[index] = current_list
+        current_list = new_list
+
+    return current_list
+
+
+def benchamark_collection(loops, n_levels):
+    total_time = 0
+    all_cycles = create_recursive_containers(n_levels)
+    for _ in range(loops):
+        gc.collect()
+        # Main loop to measure
+        t0 = pyperf.perf_counter()
+        collected = gc.collect()
+        total_time += pyperf.perf_counter() - t0
+
+        assert collected == 0
+
+    return total_time
+
+
+if __name__ == "__main__":
+    runner = pyperf.Runner()
+    runner.metadata["description"] = "GC traversal benchmark"
+    runner.bench_time_func("gc_traversal", benchamark_collection, N_LEVELS)
