Skip to content

Commit 99c938a

Browse files
committed
WIP: Annotate remainder of test_inheritance_manager module
1 parent bf03283 commit 99c938a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_managers/test_inheritance_manager.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from django.db import models
44
from django.test import TestCase
55

6+
from model_utils.managers import InheritanceManager
67
from tests.models import (
78
InheritanceManagerTestChild1,
89
InheritanceManagerTestChild2,
@@ -25,7 +26,7 @@ def setUp(self) -> None:
2526
self.grandchild1_2 = \
2627
InheritanceManagerTestGrandChild1_2.objects.create()
2728

28-
def get_manager(self):
29+
def get_manager(self) -> InheritanceManager[InheritanceManagerTestParent]:
2930
return InheritanceManagerTestParent.objects
3031

3132
def test_normal(self) -> None:
@@ -495,7 +496,7 @@ def setUp(self) -> None:
495496
self.grandchild1 = InheritanceManagerTestGrandChild1.objects.create(related=self.related)
496497
self.grandchild1_2 = InheritanceManagerTestGrandChild1_2.objects.create(related=self.related)
497498

498-
def get_manager(self):
499+
def get_manager(self) -> models.Manager:
499500
return self.related.imtests
500501

501502
def test_get_method_with_select_subclasses(self) -> None:

0 commit comments

Comments
 (0)