make linter happy

Makefile

@@ -63,6 +63,10 @@ serve-docs: $(VENV)
 manpage: $(VENV)
 	help2man $(BIN)/blag --no-info -n "blog-aware, static site generator" -o debian/blag.1
 
+.PHONY: benchmark
+benchmark: $(VENV)
+	$(BIN)/pytest --no-cov -s -rP tests/benchmark.py
+
 .PHONY: clean
 clean:
 	rm -rf build dist *.egg-info

tests/benchmark.py

@@ -0,0 +1,40 @@
+"""Benchmark for Blag."""
+
+import os
+from argparse import Namespace
+
+import blag
+from blag.blag import build
+
+
+def test_performance(args: Namespace) -> None:
+    """Test performance.
+
+    This test checks how quickly blag can generate `FILES` amount of pages from
+    markdown.
+
+    """
+    FILES = 1000
+    print(f"Generating {FILES} files")
+    # create random markdown files in the content directory
+    with open(os.path.join(blag.__path__[0], "content", "testpage.md")) as fh:
+              markdown = fh.read()
+    for i in range(FILES):
+        with open(f"content/{i}.md", "w") as f:
+            f.write(markdown)
+            f.write(str(i))
+
+    from time import time
+
+    t = time()
+    build(args)
+    t_first = time() - t
+    print(t_first)
+
+    t = time()
+    build(args)
+    t_second = time() - t
+    print(t_second)
+    print(f"First run: {t_first:.2f}s, second run: {t_second:.2f}s")
+    print(f"Speedup: {t_first/t_second:.2f}")
+