@@ -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