share belongs_to cache

hypothesis-python/RELEASE.rst

@@ -0,0 +1,3 @@
+RELEASE_TYPE: patch
+
+Improves sharing of some internal cache behavior.

hypothesis-python/src/hypothesis/internal/escalation.py

@@ -24,36 +24,34 @@
 from hypothesis.internal.compat import BaseExceptionGroup
 from hypothesis.utils.dynamicvariables import DynamicVariable
 
+FILE_CACHE: dict[ModuleType, dict[str, bool]] = {}
+
 
 def belongs_to(package: ModuleType) -> Callable[[str], bool]:
     if getattr(package, "__file__", None) is None:  # pragma: no cover
         return lambda filepath: False
 
     assert package.__file__ is not None
+    FILE_CACHE[package] = {}
     root = Path(package.__file__).resolve().parent
-    cache: dict[type, dict[str, bool]] = {str: {}, bytes: {}}
 
     def accept(filepath: str) -> bool:
-        ftype = type(filepath)
         try:
-            return cache[ftype][filepath]
+            return FILE_CACHE[package][filepath]
         except KeyError:
             pass
         try:
             Path(filepath).resolve().relative_to(root)
             result = True
         except Exception:
             result = False
-        cache[ftype][filepath] = result
+        FILE_CACHE[package][filepath] = result
         return result
 
     accept.__name__ = f"is_{package.__name__}_file"
     return accept
 
 
-FILE_CACHE: dict[bytes, bool] = {}
-
-
 is_hypothesis_file = belongs_to(hypothesis)