Fix type hints for assertFormError and assertForSetError. Refs #1137

pytest-dev · Sep 2, 2024 · b66d69a · b66d69a
1 parent b373db6
commit b66d69a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/AUTHORS b/AUTHORS
@@ -20,3 +20,4 @@ Nicolas Delaby <[email protected]>
 Hasan Ramezani <[email protected]>
 Michael Howitz
 Mark Gensler
+Pavel Taufer
diff --git a/pytest_django/asserts.py b/pytest_django/asserts.py
@@ -55,6 +55,7 @@ def assertion_func(*args, **kwargs):
 
 if TYPE_CHECKING:
     from django.http.response import HttpResponseBase
+    import django.forms as forms
 
     def assertRedirects(
         response: HttpResponseBase,
@@ -93,17 +94,15 @@ def assertNotContains(
         ...
 
     def assertFormError(
-        response: HttpResponseBase,
-        form: str,
+        form: forms.BaseForm,
         field: str | None,
         errors: str | Sequence[str],
         msg_prefix: str = ...,
     ) -> None:
         ...
 
     def assertFormsetError(
-        response: HttpResponseBase,
-        formset: str,
+        formset: forms.BaseFormSet,
         form_index: int | None,
         field: str | None,
         errors: str | Sequence[str],