Skip to content

Commit 85a2adb

Browse files
committed
Fix tests until proposed PR is merged
1 parent de2b735 commit 85a2adb

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

tests/unit/schema/v1/test_template_builder.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,8 @@ def test_coordiante_with_units() -> None:
201201
cdp_var = builder._coordinates[0]
202202
assert cdp_var.name == "cdp"
203203
assert cdp_var.data_type == ScalarType.FLOAT32
204-
assert cdp_var.metadata.units_v1.length == "m"
204+
# assert cdp_var.metadata.units_v1.length == "m"
205+
assert cdp_var.metadata[0].units_v1.length == "m"
205206

206207

207208
def test_coordinate_with_attributes() -> None:
@@ -220,8 +221,8 @@ def test_coordinate_with_attributes() -> None:
220221
cdp_var = builder._coordinates[0]
221222
assert cdp_var.name == "cdp"
222223
assert cdp_var.data_type == ScalarType.FLOAT32
223-
assert cdp_var.metadata.attributes["MGA"] == 51 # noqa: PLR2004
224-
224+
# assert cdp_var.metadata.attributes["MGA"] == 51 # noqa: PLR2004
225+
assert cdp_var.metadata[0].attributes["MGA"] == 51 # noqa: PLR2004
225226

226227
def test_coordinate_with_full_metadata() -> None:
227228
"""Test adding coordinates with all metadata."""
@@ -241,8 +242,13 @@ def test_coordinate_with_full_metadata() -> None:
241242
cdp_var = builder._coordinates[0]
242243
assert cdp_var.name == "cdp"
243244
assert cdp_var.data_type == ScalarType.FLOAT32
244-
assert cdp_var.metadata.units_v1.length == "m"
245-
assert cdp_var.metadata.attributes["MGA"] == 51 # noqa: PLR2004
245+
246+
# TODO(BrianMichell): #553 - If this PR is merged, we can remove the subscripting
247+
248+
# assert cdp_var.metadata.units_v1.length == "m"
249+
assert cdp_var.metadata[0].units_v1.length == "m"
250+
# assert cdp_var.metadata.attributes["MGA"] == 51 # noqa: PLR2004
251+
assert cdp_var.metadata[1].attributes["MGA"] == 51 # noqa: PLR2004
246252

247253

248254
def test_coordinate_builder_state() -> None:

0 commit comments

Comments
 (0)