Pre-read text, output to string

Author: AA-Turner

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

@@ -2,8 +2,9 @@
 Convert Docutils' documentation from reStructuredText to <format>.
 """
 
+import contextlib
 from pathlib import Path
-import shutil
+import time
 
 import docutils
 from docutils import core
@@ -16,36 +17,35 @@
 else:
     Trace.show = lambda message, channel: ...  # don't print to console
 
+DOC_ROOT = Path("data/docs").resolve()
 
-def build_html(doc_root, out_root):
+
+def build_html(doc_root):
+    elapsed = 0
     for file in doc_root.rglob("*.txt"):
-        try:
-            dest = out_root / file.relative_to(doc_root).with_suffix(".html")
-            dest.parent.mkdir(parents=True, exist_ok=True)
-            core.publish_file(source_path=str(file),
-                              destination_path=str(dest),
-                              reader_name="standalone",
-                              parser_name="restructuredtext",
-                              writer_name="html5",
-                              settings_overrides={
-                                  "input_encoding": "utf-8",
-                                  "report_level": 5,
-                              })
-        except docutils.ApplicationError:
-            ...
-
-
-def bench_docutils(loops, doc_root, out_root):
-    runs = []
+        file_contents = file.read_text(encoding="utf-8")
+        t0 = time.perf_counter_ns()
+        with contextlib.suppress(docutils.ApplicationError):
+            core.publish_string(source=file_contents,
+                                reader_name="standalone",
+                                parser_name="restructuredtext",
+                                writer_name="html5",
+                                settings_overrides={
+                                    "input_encoding": "unicode",
+                                    "output_encoding": "unicode",
+                                    "report_level": 5,
+                                })
+        elapsed += time.perf_counter_ns() - t0
+    return elapsed
+
+
+def bench_docutils(loops, doc_root):
+    runs_total_ns = 0
 
     for _ in range(loops):
-        t0 = pyperf.perf_counter()
-        build_html(doc_root, out_root)
-        runs.append(pyperf.perf_counter() - t0)
-
-        shutil.rmtree(out_root, ignore_errors=True)
+        runs_total_ns += build_html(doc_root)
 
-    return sum(runs)
+    return runs_total_ns / 10**9
 
 
 if __name__ == "__main__":
@@ -54,7 +54,4 @@ def bench_docutils(loops, doc_root, out_root):
     runner.metadata['description'] = "Render documentation with Docutils"
     args = runner.parse_args()
 
-    DOC_ROOT = Path("data/docs").resolve()
-    OUT_ROOT = Path("data/out").resolve()
-
-    runner.bench_time_func("docutils", bench_docutils, DOC_ROOT, OUT_ROOT)
+    runner.bench_time_func("docutils", bench_docutils, DOC_ROOT)