update

Signed-off-by: edwardfeng-db <jinghao.feng@databricks.com>
mlflow · Feb 20, 2024 · 1e8a1b2 · 1e8a1b2
1 parent 8c676d8
commit 1e8a1b2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/tracking/test_rest_tracking.py b/tests/tracking/test_rest_tracking.py
@@ -1764,7 +1764,8 @@ def test_get_experiment_graphql(mlflow_client):
         headers={"content-type": "application/json; charset=utf-8"},
     )
     assert response.status_code == 200
-    assert "GraphqlTest" in response.text
+    json = response.json()
+    assert json["data"]["mlflowGetExperiment"]["experiment"]["name"] == "GraphqlTest"
 
 
 def test_get_run_and_experiment_graphql(mlflow_client):
@@ -1790,4 +1791,5 @@ def test_get_run_and_experiment_graphql(mlflow_client):
         headers={"content-type": "application/json; charset=utf-8"},
     )
     assert response.status_code == 200
-    assert "GraphqlTest" in response.text
+    json = response.json()
+    assert json["data"]["mlflowGetRun"]["run"]["experiment"]["name"] == "GraphqlTest"