1
1
from collections import OrderedDict
2
2
from dataclasses import dataclass
3
+ from typing import Dict
3
4
4
5
from automapper import mapper
5
6
@@ -10,7 +11,7 @@ class Teacher:
10
11
11
12
12
13
class Student :
13
- def __init__ (self , name : str , classes : dict [str , Teacher ]):
14
+ def __init__ (self , name : str , classes : Dict [str , Teacher ]):
14
15
self .name = name
15
16
self .classes = classes
16
17
self .ordered_classes = OrderedDict (classes )
@@ -20,8 +21,8 @@ class PublicUserInfo:
20
21
def __init__ (
21
22
self ,
22
23
name : str ,
23
- classes : dict [str , Teacher ],
24
- ordered_classes : dict [str , Teacher ],
24
+ classes : Dict [str , Teacher ],
25
+ ordered_classes : Dict [str , Teacher ],
25
26
):
26
27
self .name = name
27
28
self .classes = classes
@@ -38,7 +39,7 @@ def test_map__dict_and_ordereddict_are_mapped_correctly_to_same_types():
38
39
39
40
assert public_info .classes == student .classes
40
41
assert public_info .classes is not student .classes
41
- assert isinstance (public_info .classes , dict )
42
+ assert isinstance (public_info .classes , Dict )
42
43
43
44
assert public_info .ordered_classes == student .ordered_classes
44
45
assert public_info .ordered_classes is not student .ordered_classes
0 commit comments