@@ -201,21 +201,29 @@ def test_get_type_string(self, mocker):
201
201
inner_property = mocker .MagicMock ()
202
202
inner_type_string = mocker .MagicMock ()
203
203
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
+ )
205
207
206
208
base_type_string = f"List[{ inner_type_string } ]"
207
209
208
210
assert p .get_type_string () == base_type_string
209
211
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
+ )
211
215
assert p .get_type_string () == f"Optional[{ base_type_string } ]"
212
216
assert p .get_type_string (no_optional = True ) == base_type_string
213
217
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
+ )
215
221
assert p .get_type_string () == f"Union[Unset, None, { base_type_string } ]"
216
222
assert p .get_type_string (no_optional = True ) == base_type_string
217
223
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
+ )
219
227
assert p .get_type_string () == f"Union[Unset, { base_type_string } ]"
220
228
assert p .get_type_string (no_optional = True ) == base_type_string
221
229
@@ -226,22 +234,28 @@ def test_get_type_imports(self, mocker):
226
234
inner_import = mocker .MagicMock ()
227
235
inner_property .get_imports .return_value = {inner_import }
228
236
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
+ )
230
240
231
241
assert p .get_imports (prefix = prefix ) == {
232
242
inner_import ,
233
243
"from typing import cast, List" ,
234
244
}
235
245
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
+ )
237
249
assert p .get_imports (prefix = prefix ) == {
238
250
inner_import ,
239
251
"from typing import cast, List" ,
240
252
"from typing import Union" ,
241
253
"from ...types import UNSET, Unset" ,
242
254
}
243
255
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
+ )
245
259
assert p .get_imports (prefix = prefix ) == {
246
260
inner_import ,
247
261
"from typing import cast, List" ,
@@ -690,15 +704,19 @@ def test_property_from_data_simple_types(self, openapi_type, prop_type, python_t
690
704
name = name , required = required , data = data , schemas = schemas , parent_name = "parent"
691
705
)
692
706
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
+ )
694
710
assert new_schemas == schemas
695
711
696
712
# Test nullable values
697
713
data .default = 0
698
714
data .nullable = True
699
715
700
716
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
+ )
702
720
703
721
# Test bad default value
704
722
data .default = "a"
@@ -900,7 +918,9 @@ def test_property_from_data_union(self, mocker):
900
918
901
919
p , s = property_from_data (name = name , required = required , data = data , schemas = Schemas (), parent_name = "parent" )
902
920
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
+ )
904
924
IntProperty .assert_called_once_with (name = name , required = required , default = 0 , nullable = False , description = None )
905
925
UnionProperty .assert_called_once_with (
906
926
name = name ,
@@ -940,7 +960,9 @@ def test__string_based_property_no_format(self):
940
960
941
961
p = _string_based_property (name = name , required = required , data = data )
942
962
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
+ )
944
966
945
967
data .pattern = "abcdef"
946
968
data .nullable = False
@@ -951,7 +973,12 @@ def test__string_based_property_no_format(self):
951
973
data = data ,
952
974
)
953
975
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 ,
955
982
)
956
983
957
984
def test__string_based_property_datetime_format (self ):
@@ -983,7 +1010,9 @@ def test__string_based_property_date_format(self):
983
1010
984
1011
p = _string_based_property (name = name , required = required , data = data )
985
1012
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
+ )
987
1016
988
1017
# Test bad default
989
1018
data .default = "a"
@@ -1088,7 +1117,7 @@ def test_build_enums(mocker):
1088
1117
(False , False ),
1089
1118
(
1090
1119
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 ),
1092
1121
),
1093
1122
],
1094
1123
)
@@ -1129,7 +1158,9 @@ def test_build_model_property(additional_properties_schema, expected_additional_
1129
1158
reference = Reference (class_name = "ParentMyModel" , module_name = "parent_my_model" ),
1130
1159
references = [],
1131
1160
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
+ ],
1133
1164
description = data .description ,
1134
1165
relative_imports = {
1135
1166
"from dateutil.parser import isoparse" ,
0 commit comments