Skip to content

Commit 0ac51f7

Browse files
committed
Updates tests of getters in test_schema.py
1 parent 81a81ac commit 0ac51f7

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/unit/test_schema.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1197,7 +1197,7 @@ def _make_one(self, *args, **kw):
11971197
def test_ctor_valid_input(self, type_system, expected):
11981198
result = self._make_one(type_system=type_system)
11991199

1200-
assert result._properties["typeSystem"] == expected
1200+
assert result.type_system == expected
12011201

12021202
def test_ctor_invalid_input(self):
12031203
with pytest.raises(TypeError) as e:
@@ -1293,15 +1293,15 @@ def test_ctor_valid_input(
12931293
output_format=output_format,
12941294
serde_info=serde_info,
12951295
)
1296-
assert storage_descriptor._properties["inputFormat"] == input_format
1297-
assert storage_descriptor._properties["locationUri"] == location_uri
1298-
assert storage_descriptor._properties["outputFormat"] == output_format
1296+
assert storage_descriptor.input_format == input_format
1297+
assert storage_descriptor.location_uri == location_uri
1298+
assert storage_descriptor.output_format == output_format
12991299
if serde_info is not None:
13001300
assert (
1301-
storage_descriptor._properties["serDeInfo"] == serde_info.to_api_repr()
1301+
storage_descriptor.serde_info == serde_info.to_api_repr()
13021302
)
13031303
else:
1304-
assert storage_descriptor._properties["serDeInfo"] == serde_info
1304+
assert storage_descriptor.serde_info == serde_info
13051305

13061306
@pytest.mark.parametrize(
13071307
"input_format,location_uri,output_format,serde_info",
@@ -1397,9 +1397,9 @@ def test_ctor_valid_input(self, serialization_library, name, parameters):
13971397
name=name,
13981398
parameters=parameters,
13991399
)
1400-
assert serde_info._properties["serializationLibrary"] == serialization_library
1401-
assert serde_info._properties["name"] == name
1402-
assert serde_info._properties["parameters"] == parameters
1400+
assert serde_info.serialization_library == serialization_library
1401+
assert serde_info.name == name
1402+
assert serde_info.parameters == parameters
14031403

14041404
@pytest.mark.parametrize(
14051405
"serialization_library,name,parameters",

0 commit comments

Comments
 (0)