@@ -174,7 +174,9 @@ def test_get_type_string(self, mocker):
174
174
175
175
from openapi_python_client .openapi_parser .properties import EnumProperty
176
176
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
+ )
178
180
179
181
assert enum_property .get_type_string () == "MyTestEnum"
180
182
enum_property .required = False
@@ -199,7 +201,9 @@ def test_constructor_from_dict(self, mocker):
199
201
200
202
assert enum_property .constructor_from_dict ("my_dict" ) == 'MyTestEnum(my_dict["test_enum"])'
201
203
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
+ )
203
207
204
208
assert (
205
209
enum_property .constructor_from_dict ("my_dict" )
@@ -269,7 +273,11 @@ def test_property_from_dict_enum(self, mocker):
269
273
name = name , required = required , data = data ,
270
274
)
271
275
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 (),
273
281
)
274
282
275
283
def test_property_from_dict_ref (self , mocker ):
0 commit comments