add benchmarks

Author: P403n1x87

pyproject.toml

@@ -38,11 +38,13 @@
   dev = [
     "mypy>=1.16.1",
     "pytest>=8",
+    "pytest-benchmark>=5",
     "pytest-cov>=6",
     "ruff>=0.12.0",
   ]
   test = [
     "pytest>=8",
+    "pytest-benchmark>=5",
     "pytest-cov",
   ]
 

tests/test_bench_roundtrip.py

@@ -0,0 +1,44 @@
+"""Round-trip decompile/recompile benchmarks.
+
+Run with:  pytest tests/test_bench_roundtrip.py --benchmark-compare
+"""
+
+import types
+
+import pytest
+
+from bytecode import Bytecode
+
+
+def _collect_code_objects(root: types.CodeType, depth: int = 1) -> list[types.CodeType]:
+    result = [root]
+    if depth > 0:
+        for const in root.co_consts:
+            if isinstance(const, types.CodeType):
+                result.extend(_collect_code_objects(const, depth - 1))
+    return result
+
+
+def _dis_corpus() -> list[types.CodeType]:
+    import importlib.util
+
+    spec = importlib.util.find_spec("dis")
+    assert spec and spec.origin
+    src = open(spec.origin).read()
+    top = compile(src, spec.origin, "exec")
+    return _collect_code_objects(top)
+
+
+_CORPUS = _dis_corpus()
+
+
+@pytest.fixture(params=_CORPUS, ids=[c.co_name for c in _CORPUS])
+def code_object(request):
+    return request.param
+
+
+def test_roundtrip(benchmark, code_object):
+    def roundtrip():
+        Bytecode.from_code(code_object).to_code()
+
+    benchmark(roundtrip)

tox.ini

@@ -6,6 +6,7 @@ isolated_build = true
 passenv = BYTECODE_PURE_PYTHON
 deps=
     pytest
+    pytest-benchmark
     pytest-cov
     pytest-subtests
 commands = pytest --cov bytecode --cov-report=xml -v tests