File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 3
3
from django .db import models
4
4
from django .test import TestCase
5
5
6
+ from model_utils .managers import InheritanceManager
6
7
from tests .models import (
7
8
InheritanceManagerTestChild1 ,
8
9
InheritanceManagerTestChild2 ,
@@ -25,7 +26,7 @@ def setUp(self) -> None:
25
26
self .grandchild1_2 = \
26
27
InheritanceManagerTestGrandChild1_2 .objects .create ()
27
28
28
- def get_manager (self ):
29
+ def get_manager (self ) -> InheritanceManager [ InheritanceManagerTestParent ] :
29
30
return InheritanceManagerTestParent .objects
30
31
31
32
def test_normal (self ) -> None :
@@ -495,7 +496,7 @@ def setUp(self) -> None:
495
496
self .grandchild1 = InheritanceManagerTestGrandChild1 .objects .create (related = self .related )
496
497
self .grandchild1_2 = InheritanceManagerTestGrandChild1_2 .objects .create (related = self .related )
497
498
498
- def get_manager (self ):
499
+ def get_manager (self ) -> models . Manager :
499
500
return self .related .imtests
500
501
501
502
def test_get_method_with_select_subclasses (self ) -> None :
You can’t perform that action at this time.
0 commit comments