Adding more assertions

Author: malavhs

tests/integ/sagemaker/jumpstart/private_hub/test_hub_content.py

@@ -39,6 +39,8 @@ def test_hub_model_reference(setup):
     describe_model_response = hub_instance.describe_model(model_name=model_id)
     assert describe_model_response is not None
     assert type(describe_model_response) == DescribeHubContentResponse
+    assert describe_model_response.hub_content_name == model_id
+    assert describe_model_response.hub_content_type == "ModelReference"
 
     delete_model_response = hub_instance.delete_model_reference(model_name=model_id)
     assert delete_model_response is not None