11from collections import OrderedDict
22from dataclasses import dataclass
3+ from typing import Dict
34
45from automapper import mapper
56
@@ -10,7 +11,7 @@ class Teacher:
1011
1112
1213class Student :
13- def __init__ (self , name : str , classes : dict [str , Teacher ]):
14+ def __init__ (self , name : str , classes : Dict [str , Teacher ]):
1415 self .name = name
1516 self .classes = classes
1617 self .ordered_classes = OrderedDict (classes )
@@ -20,8 +21,8 @@ class PublicUserInfo:
2021 def __init__ (
2122 self ,
2223 name : str ,
23- classes : dict [str , Teacher ],
24- ordered_classes : dict [str , Teacher ],
24+ classes : Dict [str , Teacher ],
25+ ordered_classes : Dict [str , Teacher ],
2526 ):
2627 self .name = name
2728 self .classes = classes
@@ -38,7 +39,7 @@ def test_map__dict_and_ordereddict_are_mapped_correctly_to_same_types():
3839
3940 assert public_info .classes == student .classes
4041 assert public_info .classes is not student .classes
41- assert isinstance (public_info .classes , dict )
42+ assert isinstance (public_info .classes , Dict )
4243
4344 assert public_info .ordered_classes == student .ordered_classes
4445 assert public_info .ordered_classes is not student .ordered_classes
0 commit comments