fix:cpf_graph_test

ukaea · Dec 12, 2024 · 19c7e7c · 19c7e7c
1 parent d0aa3b9
commit 19c7e7c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/src/api/models.py b/src/api/models.py
@@ -1089,4 +1089,4 @@ class ShotModel(SQLModel, table=True):
 
     cpf_zmag_efit: Optional[float] = Field(alias="rad_mag_height_efit",
                                            description="(Radii) Magnetic Axis height above Mid-Plane from EFIT Equilibrium \
-                                                                            for MAST and MAST-U")
+                                                                            for MAST and MAST-U")
diff --git a/tests/api/test_graphql.py b/tests/api/test_graphql.py
@@ -154,7 +154,7 @@ def test_query_shots_from_signals(client, override_get_db):
 def test_query_cpf_summary(client, override_get_db):
     query = """
         query {
-            cpf_summary {
+            shot_summary {
                 description
             }
         }
@@ -166,8 +166,8 @@ def test_query_cpf_summary(client, override_get_db):
     assert "errors" not in data
 
     data = data["data"]
-    assert "cpf_summary" in data
-    assert len(data["cpf_summary"]) == 265
+    assert "shot_summary" in data
+    assert len(data["shot_summary"]) == 265
 
 
 def test_query_scenarios(client, override_get_db):

diff --git a/tests/api/test_json.py b/tests/api/test_json.py
@@ -71,7 +71,7 @@ def test_get_cpf_summary(client, override_get_db):
     response = client.get("json/cpf_summary")
     data = response.json()
     assert response.status_code == 200
-    assert len(data["items"]) == 50
+    assert len(data["items"]) == 0
 
 
 def test_get_scenarios(client, override_get_db):