WIP: Annotate remainder of test_inheritance_manager module

jazzband · Apr 10, 2024 · d88b22f · d88b22f
1 parent e52f22a
commit d88b22f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_managers/test_inheritance_manager.py b/tests/test_managers/test_inheritance_manager.py
@@ -3,6 +3,7 @@
 from django.db import models
 from django.test import TestCase
 
+from model_utils.managers import InheritanceManager
 from tests.models import (
     InheritanceManagerTestChild1,
     InheritanceManagerTestChild2,
@@ -25,7 +26,7 @@ def setUp(self) -> None:
         self.grandchild1_2 = \
             InheritanceManagerTestGrandChild1_2.objects.create()
 
-    def get_manager(self):
+    def get_manager(self) -> InheritanceManager[InheritanceManagerTestParent]:
         return InheritanceManagerTestParent.objects
 
     def test_normal(self) -> None:
@@ -495,7 +496,7 @@ def setUp(self) -> None:
         self.grandchild1 = InheritanceManagerTestGrandChild1.objects.create(related=self.related)
         self.grandchild1_2 = InheritanceManagerTestGrandChild1_2.objects.create(related=self.related)
 
-    def get_manager(self):
+    def get_manager(self) -> models.Manager:
         return self.related.imtests
 
     def test_get_method_with_select_subclasses(self) -> None: