Chore: improve tooling around benchmarks

Author: georgesittas

Makefile

@@ -4,7 +4,10 @@ install:
 	pip install -e .
 
 bench: install-dev-rs-release
-	python benchmarks/bench.py
+	python -m benchmarks.bench
+
+bench-optimize: install-dev-rs-release
+	python -m benchmarks.optimize
 
 install-dev-rs-release:
 	cd sqlglotrs/ && python -m maturin develop -r

README.md

@@ -533,6 +533,10 @@ make check  # Full test suite & linter checks
 |            long |   0.00889 (1.0) | 0.00572 (0.643) | 0.36982 (41.56) | 0.00614 (0.690) | 0.02530 (2.844) | 0.02931 (3.294) | 0.00059 (0.066) |
 |           crazy |   0.02918 (1.0) | 0.01991 (0.682) | 1.88695 (64.66) | 0.02003 (0.686) | 7.46894 (255.9) | 0.64994 (22.27) | 0.00327 (0.112) |
 
+```
+make bench            # Run parsing benchmark
+make bench-optimize   # Run optimization benchmark
+```
 
 ## Optional Dependencies
 

benchmarks/bench.py

@@ -1,6 +1,6 @@
 import collections.abc
 
-from helpers import ascii_table
+from benchmarks.helpers import ascii_table
 
 # moz_sql_parser 3.10 compatibility
 collections.Iterable = collections.abc.Iterable

benchmarks/optimize.py

@@ -1,12 +1,12 @@
+import sys
 import typing as t
 from argparse import ArgumentParser
 
-from helpers import ascii_table
+from benchmarks.helpers import ascii_table
 from sqlglot.optimizer import optimize
 from sqlglot import parse_one
 from tests.helpers import load_sql_fixture_pairs, TPCH_SCHEMA, TPCDS_SCHEMA
 from timeit import Timer
-import sys
 
 # Deeply nested conditions currently require a lot of recursion
 sys.setrecursionlimit(10000)