Skip to content

Commit 2e7c96a

Browse files
meshybluetech
authored andcommitted
Use correct return type for assertTemplateUsed
1 parent 924c132 commit 2e7c96a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytest_django/asserts.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ def assertion_func(*args, **kwargs):
4141

4242
if TYPE_CHECKING:
4343
from django.http import HttpResponse
44+
from django.test.testcases import _AssertTemplateUsedContext
4445

4546
def assertRedirects(
4647
response: HttpResponse,
@@ -102,7 +103,7 @@ def assertTemplateUsed(
102103
template_name: Optional[str] = ...,
103104
msg_prefix: str = ...,
104105
count: Optional[int] = ...,
105-
) -> None:
106+
) -> _AssertTemplateUsedContext:
106107
...
107108

108109
def assertTemplateNotUsed(

0 commit comments

Comments
 (0)