Merge pull request #3772 from Viicos/patch-1

Zac-HD · web-flow · commit ca63e9f452ce · 2023-10-16T11:53:51.000+01:00
Make return type of `from_model` parameterized with respect to `model`
diff --git a/hypothesis-python/src/hypothesis/extra/django/_impl.py b/hypothesis-python/src/hypothesis/extra/django/_impl.py
@@ -11,7 +11,7 @@
 import sys
 import unittest
 from functools import partial
-from typing import TYPE_CHECKING, Optional, Type, Union
+from typing import TYPE_CHECKING, Optional, Type, TypeVar, Union
 
 from django import forms as df, test as dt
 from django.contrib.staticfiles import testing as dst
@@ -30,6 +30,8 @@
 else:
     EllipsisType = type(Ellipsis)
 
+ModelT = TypeVar("ModelT", bound=dm.Model)
+
 
 class HypothesisTestCase:
     def setup_example(self):
@@ -64,8 +66,8 @@ class StaticLiveServerTestCase(HypothesisTestCase, dst.StaticLiveServerTestCase)
 
 @defines_strategy()
 def from_model(
-    model: Type[dm.Model], /, **field_strategies: Union[st.SearchStrategy, EllipsisType]
-) -> st.SearchStrategy:
+    model: Type[ModelT], /, **field_strategies: Union[st.SearchStrategy, EllipsisType]
+) -> st.SearchStrategy[ModelT]:
     """Return a strategy for examples of ``model``.
 
     .. warning::
