1
- from enum import Enum
2
- from typing import Tuple
1
+ from enum import Enum , IntEnum
3
2
4
3
from automapper import mapper
5
4
@@ -10,21 +9,21 @@ class StringEnum(str, Enum):
10
9
Value3 = "value3"
11
10
12
11
13
- class IntEnum ( int , Enum ):
12
+ class IntValueEnum ( IntEnum ):
14
13
Value1 = 1
15
14
Value2 = 2
16
15
Value3 = 3
17
16
18
17
19
- class TupleEnum (Tuple [ str , str ], Enum ):
18
+ class TupleEnum (Enum ):
20
19
Value1 = ("value" , "1" )
21
20
Value2 = ("value" , "2" )
22
21
Value3 = ("value" , "3" )
23
22
24
23
25
24
class SourceClass :
26
25
def __init__ (
27
- self , string_value : StringEnum , int_value : IntEnum , tuple_value : TupleEnum
26
+ self , string_value : StringEnum , int_value : IntValueEnum , tuple_value : TupleEnum
28
27
) -> None :
29
28
self .string_value = string_value
30
29
self .int_value = int_value
@@ -33,17 +32,17 @@ def __init__(
33
32
34
33
class TargetClass :
35
34
def __init__ (
36
- self , string_value : StringEnum , int_value : IntEnum , tuple_value : TupleEnum
35
+ self , string_value : StringEnum , int_value : IntValueEnum , tuple_value : TupleEnum
37
36
) -> None :
38
37
self .string_value = string_value
39
38
self .int_value = int_value
40
39
self .tuple_value = tuple_value
41
40
42
41
43
42
def test_map__enum ():
44
- src = SourceClass (StringEnum .Value1 , IntEnum .Value2 , TupleEnum .Value3 )
43
+ src = SourceClass (StringEnum .Value1 , IntValueEnum .Value2 , TupleEnum .Value3 )
45
44
dst = mapper .to (TargetClass ).map (src )
46
45
47
46
assert dst .string_value == StringEnum .Value1
48
- assert dst .int_value == IntEnum .Value2
47
+ assert dst .int_value == IntValueEnum .Value2
49
48
assert dst .tuple_value == TupleEnum .Value3
0 commit comments