@@ -201,21 +201,29 @@ def test_get_type_string(self, mocker):
201201 inner_property = mocker .MagicMock ()
202202 inner_type_string = mocker .MagicMock ()
203203 inner_property .get_type_string .return_value = inner_type_string
204- p = ListProperty (name = "test" , required = True , default = None , inner_property = inner_property , nullable = False , description = None )
204+ p = ListProperty (
205+ name = "test" , required = True , default = None , inner_property = inner_property , nullable = False , description = None
206+ )
205207
206208 base_type_string = f"List[{ inner_type_string } ]"
207209
208210 assert p .get_type_string () == base_type_string
209211
210- p = ListProperty (name = "test" , required = True , default = None , inner_property = inner_property , nullable = True , description = None )
212+ p = ListProperty (
213+ name = "test" , required = True , default = None , inner_property = inner_property , nullable = True , description = None
214+ )
211215 assert p .get_type_string () == f"Optional[{ base_type_string } ]"
212216 assert p .get_type_string (no_optional = True ) == base_type_string
213217
214- p = ListProperty (name = "test" , required = False , default = None , inner_property = inner_property , nullable = True , description = None )
218+ p = ListProperty (
219+ name = "test" , required = False , default = None , inner_property = inner_property , nullable = True , description = None
220+ )
215221 assert p .get_type_string () == f"Union[Unset, None, { base_type_string } ]"
216222 assert p .get_type_string (no_optional = True ) == base_type_string
217223
218- p = ListProperty (name = "test" , required = False , default = None , inner_property = inner_property , nullable = False , description = None )
224+ p = ListProperty (
225+ name = "test" , required = False , default = None , inner_property = inner_property , nullable = False , description = None
226+ )
219227 assert p .get_type_string () == f"Union[Unset, { base_type_string } ]"
220228 assert p .get_type_string (no_optional = True ) == base_type_string
221229
@@ -226,22 +234,28 @@ def test_get_type_imports(self, mocker):
226234 inner_import = mocker .MagicMock ()
227235 inner_property .get_imports .return_value = {inner_import }
228236 prefix = "..."
229- p = ListProperty (name = "test" , required = True , default = None , inner_property = inner_property , nullable = False , description = None )
237+ p = ListProperty (
238+ name = "test" , required = True , default = None , inner_property = inner_property , nullable = False , description = None
239+ )
230240
231241 assert p .get_imports (prefix = prefix ) == {
232242 inner_import ,
233243 "from typing import cast, List" ,
234244 }
235245
236- p = ListProperty (name = "test" , required = False , default = None , inner_property = inner_property , nullable = False , description = None )
246+ p = ListProperty (
247+ name = "test" , required = False , default = None , inner_property = inner_property , nullable = False , description = None
248+ )
237249 assert p .get_imports (prefix = prefix ) == {
238250 inner_import ,
239251 "from typing import cast, List" ,
240252 "from typing import Union" ,
241253 "from ...types import UNSET, Unset" ,
242254 }
243255
244- p = ListProperty (name = "test" , required = False , default = None , inner_property = inner_property , nullable = True , description = None )
256+ p = ListProperty (
257+ name = "test" , required = False , default = None , inner_property = inner_property , nullable = True , description = None
258+ )
245259 assert p .get_imports (prefix = prefix ) == {
246260 inner_import ,
247261 "from typing import cast, List" ,
@@ -690,15 +704,19 @@ def test_property_from_data_simple_types(self, openapi_type, prop_type, python_t
690704 name = name , required = required , data = data , schemas = schemas , parent_name = "parent"
691705 )
692706
693- assert p == prop_type (name = name , required = required , default = python_type (data .default ), nullable = False , description = None )
707+ assert p == prop_type (
708+ name = name , required = required , default = python_type (data .default ), nullable = False , description = None
709+ )
694710 assert new_schemas == schemas
695711
696712 # Test nullable values
697713 data .default = 0
698714 data .nullable = True
699715
700716 p , _ = property_from_data (name = name , required = required , data = data , schemas = schemas , parent_name = "parent" )
701- assert p == prop_type (name = name , required = required , default = python_type (data .default ), nullable = True , description = None )
717+ assert p == prop_type (
718+ name = name , required = required , default = python_type (data .default ), nullable = True , description = None
719+ )
702720
703721 # Test bad default value
704722 data .default = "a"
@@ -900,7 +918,9 @@ def test_property_from_data_union(self, mocker):
900918
901919 p , s = property_from_data (name = name , required = required , data = data , schemas = Schemas (), parent_name = "parent" )
902920
903- FloatProperty .assert_called_once_with (name = name , required = required , default = 0.0 , nullable = False , description = None )
921+ FloatProperty .assert_called_once_with (
922+ name = name , required = required , default = 0.0 , nullable = False , description = None
923+ )
904924 IntProperty .assert_called_once_with (name = name , required = required , default = 0 , nullable = False , description = None )
905925 UnionProperty .assert_called_once_with (
906926 name = name ,
@@ -940,7 +960,9 @@ def test__string_based_property_no_format(self):
940960
941961 p = _string_based_property (name = name , required = required , data = data )
942962
943- assert p == StringProperty (name = name , required = required , nullable = True , default = "'\\ \\ \" hello world\\ \\ \" '" , description = None )
963+ assert p == StringProperty (
964+ name = name , required = required , nullable = True , default = "'\\ \\ \" hello world\\ \\ \" '" , description = None
965+ )
944966
945967 data .pattern = "abcdef"
946968 data .nullable = False
@@ -951,7 +973,12 @@ def test__string_based_property_no_format(self):
951973 data = data ,
952974 )
953975 assert p == StringProperty (
954- name = name , required = required , nullable = False , default = "'\\ \\ \" hello world\\ \\ \" '" , pattern = "abcdef" , description = None
976+ name = name ,
977+ required = required ,
978+ nullable = False ,
979+ default = "'\\ \\ \" hello world\\ \\ \" '" ,
980+ pattern = "abcdef" ,
981+ description = None ,
955982 )
956983
957984 def test__string_based_property_datetime_format (self ):
@@ -983,7 +1010,9 @@ def test__string_based_property_date_format(self):
9831010
9841011 p = _string_based_property (name = name , required = required , data = data )
9851012
986- assert p == DateProperty (name = name , required = required , nullable = True , default = "isoparse('2020-11-06').date()" , description = None )
1013+ assert p == DateProperty (
1014+ name = name , required = required , nullable = True , default = "isoparse('2020-11-06').date()" , description = None
1015+ )
9871016
9881017 # Test bad default
9891018 data .default = "a"
@@ -1088,7 +1117,7 @@ def test_build_enums(mocker):
10881117 (False , False ),
10891118 (
10901119 oai .Schema .construct (type = "string" ),
1091- StringProperty (name = "AdditionalProperty" , required = True , nullable = False , default = None ,description = None ),
1120+ StringProperty (name = "AdditionalProperty" , required = True , nullable = False , default = None , description = None ),
10921121 ),
10931122 ],
10941123)
@@ -1129,7 +1158,9 @@ def test_build_model_property(additional_properties_schema, expected_additional_
11291158 reference = Reference (class_name = "ParentMyModel" , module_name = "parent_my_model" ),
11301159 references = [],
11311160 required_properties = [StringProperty (name = "req" , required = True , nullable = False , default = None , description = None )],
1132- optional_properties = [DateTimeProperty (name = "opt" , required = False , nullable = False , default = None , description = None )],
1161+ optional_properties = [
1162+ DateTimeProperty (name = "opt" , required = False , nullable = False , default = None , description = None )
1163+ ],
11331164 description = data .description ,
11341165 relative_imports = {
11351166 "from dateutil.parser import isoparse" ,
0 commit comments