1 files changed, 39 insertions, 0 deletions
diff --git a/bench.py b/bench.py
new file mode 100644
index 0000000..1a1f9bc
--- /dev/null
+++ b/bench.py
@@ -0,0 +1,39 @@
+import pyperf
+
+runner = pyperf.Runner()
+
+for name in ("native", "rust_speedups"):  # ("native", "speedups", "rust_speedups"):
+    runner.timeit(
+        f"short escape {name}",
+        setup=f"from markupsafe._{name} import escape",
+        stmt='escape("<strong>Hello, World!</strong>")',
+    )
+    # runner.timeit(
+    #     f"long escape {name}",
+    #     setup=(
+    #         f"from markupsafe._{name} import escape;\n"
+    #         "s = \"Hello, World!\" * 1000"
+    #     ),
+    #     stmt="escape(s)",
+    # )
+    # runner.timeit(
+    #     f"short empty {name}",
+    #     setup=f"from markupsafe._{name} import escape",
+    #     stmt="escape(\"Hello, World!\")",
+    # )
+    # runner.timeit(
+    #     f"long empty {name}",
+    #     setup=(
+    #         f"from markupsafe._{name} import escape;\n"
+    #         "s = \"<strong>Hello, World!</strong>\" * 1000"
+    #     ),
+    #     stmt="escape(s)",
+    # )
+    # runner.timeit(
+    #     f"long suffix {name}",
+    #     setup=(
+    #         f"from markupsafe._{name} import escape;\n"
+    #         "s = \"<strong>Hello, World!</strong>\" + \"x\" * 100000"
+    #     ),
+    #     stmt="escape(s)",
+    # )