Skip to content

Commit 3bc6745

Browse files
authored
updates to test_registering_model() based on previous review comments (#227)
1 parent a34d419 commit 3bc6745

File tree

1 file changed

+14
-13
lines changed

1 file changed

+14
-13
lines changed

tests/model_registry/test_model_registry_creation.py

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -67,19 +67,20 @@ def test_registering_model(
6767
registered_model: RegisteredModel,
6868
):
6969
model = model_registry_client.get_registered_model(MODEL_NAME)
70-
errors = []
71-
if not registered_model.id == model.id:
72-
errors.append(f"Unexpected id, received {model.id}")
73-
if not registered_model.name == model.name:
74-
errors.append(f"Unexpected name, received {model.name}")
75-
if not registered_model.description == model.description:
76-
errors.append(f"Unexpected description, received {model.description}")
77-
if not registered_model.owner == model.owner:
78-
errors.append(f"Unexpected owner, received {model.owner}")
79-
if not registered_model.state == model.state:
80-
errors.append(f"Unexpected state, received {model.state}")
81-
82-
assert not errors, "errors found in model registry response validation:\n{}".format("\n".join(errors))
70+
expected_attrs = {
71+
"id": registered_model.id,
72+
"name": registered_model.name,
73+
"description": registered_model.description,
74+
"owner": registered_model.owner,
75+
"state": registered_model.state,
76+
}
77+
errors = [
78+
f"Unexpected {attr} expected: {expected}, received {getattr(model, attr)}"
79+
for attr, expected in expected_attrs.items()
80+
if getattr(model, attr) != expected
81+
]
82+
if errors:
83+
pytest.fail("errors found in model registry response validation:\n{}".format("\n".join(errors)))
8384

8485
def test_model_registry_operator_env(
8586
self,

0 commit comments

Comments
 (0)