Add test client code to current prototype IT

opendatahub-io · Sep 25, 2023 · ae1e617 · ae1e617
1 parent 2083ca0
commit ae1e617
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/python/test_mlmetadata.py b/test/python/test_mlmetadata.py
@@ -45,6 +45,12 @@ def main():
     response = store.PutArtifactType(request)
     model_type_id = response.type_id
 
+    request = metadata_store_service_pb2.GetArtifactTypeRequest()
+    request.type_name = "SavedModel"
+    response = store.GetArtifactType(request)
+    assert response.artifact_type.id == 2
+    assert response.artifact_type.name == "SavedModel"
+
     # Query all registered Artifact types.
     # artifact_types = store.GetArtifactTypes()