Some type annotations

Author: bluetech

pytest_django/django_compat.py

@@ -1,9 +1,12 @@
 # Note that all functions here assume django is available.  So ensure
 # this is the case before you call them.
+from __future__ import annotations
 
+import pytest
 
-def is_django_unittest(request_or_item) -> bool:
-    """Returns True if the request_or_item is a Django test case, otherwise False"""
+
+def is_django_unittest(request_or_item: pytest.FixtureRequest | pytest.Item) -> bool:
+    """Returns whether the request or item is a Django test case."""
     from django.test import SimpleTestCase
 
     cls = getattr(request_or_item, "cls", None)

pytest_django/fixtures.py

@@ -598,11 +598,11 @@ def _live_server_helper(request: pytest.FixtureRequest) -> Generator[None, None,
 
 @contextmanager
 def _assert_num_queries(
-    config,
+    config: pytest.Config,
     num: int,
     exact: bool = True,
     connection=None,
-    info=None,
+    info: str | None = None,
 ) -> Generator[django.test.utils.CaptureQueriesContext, None, None]:
     from django.test.utils import CaptureQueriesContext
 
@@ -636,12 +636,12 @@ def _assert_num_queries(
 
 
 @pytest.fixture()
-def django_assert_num_queries(pytestconfig):
+def django_assert_num_queries(pytestconfig: pytest.Config):
     return partial(_assert_num_queries, pytestconfig)
 
 
 @pytest.fixture()
-def django_assert_max_num_queries(pytestconfig):
+def django_assert_max_num_queries(pytestconfig: pytest.Config):
     return partial(_assert_num_queries, pytestconfig, exact=False)