move version guard to compat.py

HypothesisWorks · Dec 15, 2023 · e9ba14c · e9ba14c
1 parent 720d918
commit e9ba14c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/hypothesis-python/src/hypothesis/internal/compat.py b/hypothesis-python/src/hypothesis/internal/compat.py
@@ -205,6 +205,11 @@ def dataclass_asdict(obj, *, dict_factory=dict):
     return _asdict_inner(obj, dict_factory)
 
 
+if sys.version_info[:2] >= (3, 12):
+    # see issue #3812
+    dataclass_asdict = dataclasses.asdict
+
+
 def _asdict_inner(obj, dict_factory):
     if dataclasses._is_dataclass_instance(obj):
         if dict_factory is dict:

diff --git a/hypothesis-python/src/hypothesis/strategies/_internal/utils.py b/hypothesis-python/src/hypothesis/strategies/_internal/utils.py
@@ -178,11 +178,7 @@ def to_jsonable(obj: object) -> object:
         and dcs.is_dataclass(obj)
         and not isinstance(obj, type)
     ):
-        if sys.version_info[:2] < (3, 12):
-            # see issue #3812
-            return to_jsonable(dataclass_asdict(obj))
-        else:  # pragma: no cover
-            return to_jsonable(dcs.asdict(obj))
+        return to_jsonable(dataclass_asdict(obj))
     if attr.has(type(obj)):
         return to_jsonable(attr.asdict(obj, recurse=False))  # type: ignore
     if (pyd := sys.modules.get("pydantic")) and isinstance(obj, pyd.BaseModel):