Pull in another microbenchmark that can be improved later.

Julian · Julian · commit 7a2da6bd0d38 · 2023-07-25T12:37:53.000+02:00
diff --git a/jsonschema/benchmarks/subcomponents.py b/jsonschema/benchmarks/subcomponents.py
@@ -0,0 +1,42 @@
+"""
+A benchmark which tries to compare the possible slow subparts of validation.
+"""
+from referencing import Registry
+from referencing.jsonschema import DRAFT202012
+from rpds import HashTrieMap, HashTrieSet
+
+from jsonschema import Draft202012Validator
+
+schema = {
+    "type": "array",
+    "minLength": 1,
+    "maxLength": 1,
+    "items": {"type": "integer"}
+}
+
+hmap = HashTrieMap()
+hset = HashTrieSet()
+
+registry = Registry()
+
+v = Draft202012Validator(schema)
+
+
+def registry_data_structures():
+    return hmap.insert("foo", "bar"), hset.insert("foo")
+
+
+def registry_add():
+    resource = DRAFT202012.create_resource(schema)
+    return registry.with_resource(uri="urn:example", resource=resource)
+
+
+if __name__ == "__main__":
+    from pyperf import Runner
+    runner = Runner()
+
+    runner.bench_func("HashMap/HashSet insertion", registry_data_structures)
+    runner.bench_func("Registry insertion", registry_add)
+    runner.bench_func("Success", lambda: v.is_valid([1]))
+    runner.bench_func("Failure", lambda: v.is_valid(["foo"]))
+    runner.bench_func("Metaschema validation", lambda: v.check_schema(schema))
diff --git a/pyproject.toml b/pyproject.toml
@@ -136,6 +136,7 @@ multi_line_output = 3
 [tool.mypy]
 ignore_missing_imports = true
 show_error_codes = true
+exclude = ["jsonschema/benchmarks/*"]
 
 [tool.ruff]
 line-length = 79
