File tree Expand file tree Collapse file tree 2 files changed +7
-3
lines changed Expand file tree Collapse file tree 2 files changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -45,9 +45,9 @@ class SortKey(str, Enum):
4545 TIME = 'time' , 'tottime'
4646
4747 def __new__ (cls , * values ):
48- obj = str . __new__ ( cls )
49-
50- obj ._value_ = values [ 0 ]
48+ value = values [ 0 ]
49+ obj = str . __new__ ( cls , value )
50+ obj ._value_ = value
5151 for other_value in values [1 :]:
5252 cls ._value2member_map_ [other_value ] = obj
5353 obj ._all_values = values
Original file line number Diff line number Diff line change @@ -95,5 +95,9 @@ def pass3(): pass
9595 self .assertIn ('pass2' , funcs_called )
9696 self .assertIn ('pass3' , funcs_called )
9797
98+ def test_SortKey_enum (self ):
99+ self .assertEqual (SortKey .FILENAME , 'filename' )
100+ self .assertNotEqual (SortKey .FILENAME , SortKey .CALLS )
101+
98102if __name__ == "__main__" :
99103 unittest .main ()
You can’t perform that action at this time.
0 commit comments