From ed4ba121c910324a1a08a659ee08dd01bdcc6371 Mon Sep 17 00:00:00 2001
From: Eivind Jahren <ejah@equinor.com>
Date: Fri, 11 Oct 2024 08:57:55 +0200
Subject: [PATCH] Test empty gen_kw in plotapi

---
 .../gui/tools/plot/test_plot_api.py           | 49 ++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

diff --git a/tests/ert/unit_tests/gui/tools/plot/test_plot_api.py b/tests/ert/unit_tests/gui/tools/plot/test_plot_api.py
index 42ce5802286..1e9a7acb8f2 100644
--- a/tests/ert/unit_tests/gui/tools/plot/test_plot_api.py
+++ b/tests/ert/unit_tests/gui/tools/plot/test_plot_api.py
@@ -7,10 +7,11 @@
 import pandas as pd
 import polars
 import pytest
+import xarray as xr
 from pandas.testing import assert_frame_equal
 from starlette.testclient import TestClient
 
-from ert.config import SummaryConfig
+from ert.config import GenKwConfig, SummaryConfig
 from ert.dark_storage import enkf
 from ert.dark_storage.app import app
 from ert.gui.tools.plot.plot_api import PlotApi, PlotApiKeyDefinition
@@ -236,3 +237,49 @@ def test_plot_api_handles_urlescape(tmp_path, monkeypatch):
             enkf._storage.close()
         enkf._storage = None
         gc.collect()
+
+
+def test_plot_api_handles_empty_gen_kw(tmp_path, monkeypatch):
+    with open_storage(tmp_path / "storage", mode="w") as storage:
+        monkeypatch.setenv("ERT_STORAGE_NO_TOKEN", "yup")
+        monkeypatch.setenv("ERT_STORAGE_ENS_PATH", storage.path)
+        api = PlotApi()
+        key = "gen_kw"
+        name = "<poro>"
+        experiment = storage.create_experiment(
+            parameters=[
+                GenKwConfig(
+                    name=key,
+                    forward_init=False,
+                    update=False,
+                    template_file=None,
+                    output_file=None,
+                    transform_function_definitions=[],
+                ),
+            ],
+            responses=[],
+            observations={},
+        )
+        ensemble = storage.create_ensemble(experiment.id, ensemble_size=10)
+        assert api.data_for_key(str(ensemble.id), key).empty
+        ensemble.save_parameters(
+            key,
+            1,
+            xr.Dataset(
+                {
+                    "values": ("names", [1.0]),
+                    "transformed_values": ("names", [1.0]),
+                    "names": [name],
+                }
+            ),
+        )
+        assert api.data_for_key(str(ensemble.id), key + ":" + name).to_csv() == dedent(
+            """\
+            Realization,0
+            1,1.0
+            """
+        )
+        if enkf._storage is not None:
+            enkf._storage.close()
+        enkf._storage = None
+        gc.collect()