asserts: add assertQuerySetEqual typing

Fix #1056.
pytest-dev · Apr 5, 2023 · 5f5cc4c · 5f5cc4c
1 parent a499110
commit 5f5cc4c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/docs/changelog.rst b/docs/changelog.rst
@@ -13,6 +13,8 @@ Improvements
 
 * Drop support for Python version 3.6.
 
+* Add precise `pytest_django.asserts.assertQuerySetEqual` typing.
+
 v4.5.2 (2021-12-07)
 -------------------
 

diff --git a/pytest_django/asserts.py b/pytest_django/asserts.py
@@ -188,6 +188,7 @@ def assertXMLNotEqual(
     ) -> None:
         ...
 
+    # Removed in Django 5.1: use assertQuerySetEqual.
     def assertQuerysetEqual(
         qs,
         values,
@@ -197,6 +198,15 @@ def assertQuerysetEqual(
     ) -> None:
         ...
 
+    def assertQuerySetEqual(
+        qs,
+        values,
+        transform=...,
+        ordered: bool = ...,
+        msg: Optional[str] = ...,
+    ) -> None:
+        ...
+
     def assertNumQueries(
         num: int,
         func=...,