@@ -174,7 +174,9 @@ def test_get_type_string(self, mocker):
174174
175175 from openapi_python_client .openapi_parser .properties import EnumProperty
176176
177- enum_property = EnumProperty (name = "test" , required = True , default = None , values = {}, reference = mocker .MagicMock (class_name = "MyTestEnum" ))
177+ enum_property = EnumProperty (
178+ name = "test" , required = True , default = None , values = {}, reference = mocker .MagicMock (class_name = "MyTestEnum" )
179+ )
178180
179181 assert enum_property .get_type_string () == "MyTestEnum"
180182 enum_property .required = False
@@ -199,7 +201,9 @@ def test_constructor_from_dict(self, mocker):
199201
200202 assert enum_property .constructor_from_dict ("my_dict" ) == 'MyTestEnum(my_dict["test_enum"])'
201203
202- enum_property = EnumProperty (name = "test_enum" , required = False , default = None , values = {})
204+ enum_property = EnumProperty (
205+ name = "test_enum" , required = False , default = None , values = {}, reference = fake_reference
206+ )
203207
204208 assert (
205209 enum_property .constructor_from_dict ("my_dict" )
@@ -269,7 +273,11 @@ def test_property_from_dict_enum(self, mocker):
269273 name = name , required = required , data = data ,
270274 )
271275 EnumProperty .assert_called_once_with (
272- name = name , required = required , values = EnumProperty .values_from_list (), default = data ["default" ], reference = from_ref ()
276+ name = name ,
277+ required = required ,
278+ values = EnumProperty .values_from_list (),
279+ default = data ["default" ],
280+ reference = from_ref (),
273281 )
274282
275283 def test_property_from_dict_ref (self , mocker ):
0 commit comments