@@ -185,7 +185,7 @@ def test_add_class_with_extra_kwargs(
185185)
186186def test_add_enum_with_extra_permissible_values (
187187 enum_def : EnumDefinition ,
188- permissible_values : List [Union [str , PermissibleValue ]],
188+ permissible_values : list [Union [str , PermissibleValue ]],
189189 expected_added_enum : Optional [EnumDefinition ],
190190):
191191 """
@@ -231,7 +231,7 @@ def test_add_enum_with_extra_permissible_values(
231231)
232232def test_add_enum_with_extra_kwargs (
233233 enum_def : Union [EnumDefinition , dict , str ],
234- extra_kwargs : Dict [str , Any ],
234+ extra_kwargs : dict [str , Any ],
235235 expected_added_enum : Optional [EnumDefinition ],
236236):
237237 """
@@ -247,7 +247,7 @@ def test_add_enum_with_extra_kwargs(
247247 elif extra_kwargs .keys () - enum_meta_slots :
248248 # Handle the case of extra kwargs include a key that is not a meta slot of
249249 # `EnumDefinition`
250- with pytest .raises (ValueError ):
250+ with pytest .raises (TypeError ):
251251 builder .add_enum (enum_def , ** extra_kwargs )
252252 else :
253253 builder .add_enum (enum_def , ** extra_kwargs )
0 commit comments