Correctly retrieve forward referenced annotations from typed dictionaries

Author: silasary

AUTHORS.rst

@@ -78,6 +78,7 @@ their individual contributions.
 * `jwg4 <https://www.github.com/jwg4>`_
 * `Kai Chen <https://www.github.com/kx-chen>`_ ([email protected])
 * `Karthikeyan Singaravelan <https://www.github.com/tirkarthi>`_ ([email protected])
+* `Katelyn Gigante <https://github.com/silasary>`_
 * `Katrina Durance <https://github.com/kdurance>`_
 * `kbara <https://www.github.com/kbara>`_
 * `Kristian Glass <https://www.github.com/doismellburning>`_

hypothesis-python/RELEASE.rst

@@ -0,0 +1,5 @@
+RELEASE_TYPE: patch
+
+This patch fixes :func:`~hypothesis.strategies.from_type` on a :class:`~python:typing.TypedDict`
+with complex annotations, defined in a file using ``from __future__ import annotations``.
+Thanks to Katelyn Gigante for identifying and fixing this bug!

hypothesis-python/src/hypothesis/strategies/_internal/core.py

@@ -1076,7 +1076,7 @@ def as_strategy(strat_or_callable, thing, final=True):
         # way to tell and we just have to assume that everything is required.
         # See https://github.com/python/cpython/pull/17214 for details.
         optional = getattr(thing, "__optional_keys__", ())
-        anns = {k: from_type(v) for k, v in thing.__annotations__.items()}
+        anns = {k: from_type(v) for k, v in get_type_hints(thing).items()}
         return fixed_dictionaries(  # type: ignore
             mapping={k: v for k, v in anns.items() if k not in optional},
             optional={k: v for k, v in anns.items() if k in optional},

hypothesis-python/tests/nocover/test_type_lookup_future_annotations.py

@@ -0,0 +1,33 @@
+# This file is part of Hypothesis, which may be found at
+# https://github.com/HypothesisWorks/hypothesis/
+#
+# Copyright the Hypothesis Authors.
+# Individual contributors are listed in AUTHORS.rst and the git log.
+#
+# This Source Code Form is subject to the terms of the Mozilla Public License,
+# v. 2.0. If a copy of the MPL was not distributed with this file, You can
+# obtain one at https://mozilla.org/MPL/2.0/.
+
+from __future__ import annotations
+
+from typing import TypedDict, Union
+
+from hypothesis import given, strategies as st
+
+
+@given(st.data())
+def test_complex_forward_ref_in_typed_dict(data):
+    alias = Union[int, bool]
+
+    class A(TypedDict):
+        a: int
+
+    class B(TypedDict):
+        a: A
+        b: alias
+
+    b_strategy = st.from_type(B)
+    d = data.draw(b_strategy)
+    assert isinstance(d["a"], dict)
+    assert isinstance(d["a"]["a"], int)
+    assert isinstance(d["b"], (int, bool))