fix: update benchmark test script

Author: shenxianpeng

.gitignore

@@ -13,6 +13,7 @@ result.txt
 testing/main.c
 */*compile_commands.json
 testing/benchmark_results.txt
+testing/test-examples/
 
 # Ignore Python wheel packages (clang-format, clang-tidy)
 clang-tidy-1*

.pre-commit-config.yaml

@@ -17,4 +17,5 @@ repos:
     rev: v0.12.11
     hooks:
       - id: ruff
+        args: [--fix]
       - id: ruff-format

testing/benchmark_hooks.py

@@ -30,17 +30,29 @@
 ]
 
 # Automatically find all C/C++ files in testing/ (and optionally src/, include/)
-TARGET_FILES = (
-    glob.glob("testing/**/*.c", recursive=True)
-    + glob.glob("testing/**/*.cpp", recursive=True)
-    + glob.glob("testing/**/*.h", recursive=True)
-    + glob.glob("testing/**/*.hpp", recursive=True)
-)
+TARGET_FILES = glob.glob("testing/test-examples/*.c", recursive=True)
 
 REPEATS = 5
 RESULTS_FILE = "testing/benchmark_results.txt"
 
 
+def git_clone():
+    try:
+        subprocess.run(
+            [
+                "git",
+                "clone",
+                "--depth",
+                "1",
+                "https://github.com/gouravthakur39/beginners-C-program-examples.git",
+                "testing/test-examples",
+            ],
+            check=True,
+        )
+    except subprocess.CalledProcessError:
+        pass
+
+
 def run_hook(config, files):
     cmd = ["pre-commit", "run", "--config", config, "--files"] + files
     start = time.perf_counter()
@@ -53,14 +65,28 @@ def run_hook(config, files):
     return end - start
 
 
+def safe_git_restore(files):
+    # Only restore files tracked by git
+    tracked = []
+    for f in files:
+        result = subprocess.run(
+            ["git", "ls-files", "--error-unmatch", f],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+        )
+        if result.returncode == 0:
+            tracked.append(f)
+    if tracked:
+        subprocess.run(["git", "restore"] + tracked)
+
+
 def benchmark():
     results = {}
     for hook in HOOKS:
         times = []
         print(f"Benchmarking {hook['name']}...")
         for i in range(REPEATS):
-            # Clean up any changes before each run
-            subprocess.run(["git", "restore"] + TARGET_FILES)
+            safe_git_restore(TARGET_FILES)
             subprocess.run(["pre-commit", "clean"])
             t = run_hook(hook["config"], TARGET_FILES)
             print(f"  Run {i + 1}: {t:.3f} seconds")
@@ -70,23 +96,44 @@ def benchmark():
 
 
 def report(results):
+    headers = ["Hook", "Avg (s)", "Std (s)", "Min (s)", "Max (s)", "Runs"]
+    col_widths = [max(len(h), 16) for h in headers]
+    # Calculate max width for each column
+    for name, times in results.items():
+        col_widths[0] = max(col_widths[0], len(name))
+    print("\nBenchmark Results:\n")
+    # Print header
+    header_row = " | ".join(h.ljust(w) for h, w in zip(headers, col_widths))
+    print(header_row)
+    print("-+-".join("-" * w for w in col_widths))
+    # Print rows
     lines = []
     for name, times in results.items():
         avg = statistics.mean(times)
         std = statistics.stdev(times) if len(times) > 1 else 0.0
         min_t = min(times)
         max_t = max(times)
-        lines.append(
-            f"{name}: avg={avg:.3f}s, std={std:.3f}s, min={min_t:.3f}s, max={max_t:.3f}s, runs={len(times)}"
-        )
-    print("\nBenchmark Results:")
-    print("\n".join(lines))
+        row = [
+            name.ljust(col_widths[0]),
+            f"{avg:.3f}".ljust(col_widths[1]),
+            f"{std:.3f}".ljust(col_widths[2]),
+            f"{min_t:.3f}".ljust(col_widths[3]),
+            f"{max_t:.3f}".ljust(col_widths[4]),
+            str(len(times)).ljust(col_widths[5]),
+        ]
+        print(" | ".join(row))
+        lines.append(" | ".join(row))
+    # Save to file
     with open(RESULTS_FILE, "w") as f:
-        f.write("\n".join(lines) + "\n")
-    print(f"Results saved to {RESULTS_FILE}")
+        f.write(header_row + "\n")
+        f.write("-+-".join("-" * w for w in col_widths) + "\n")
+        for line in lines:
+            f.write(line + "\n")
+    print(f"\nResults saved to {RESULTS_FILE}")
 
 
 def main():
+    git_clone()
     results = benchmark()
     report(results)