Update test to expect empty df instead of KeyError

equinor · May 24, 2024 · 01b4537 · 01b4537
1 parent d5039b2
commit 01b4537
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/unit_tests/test_libres_facade.py b/tests/unit_tests/test_libres_facade.py
@@ -187,11 +187,12 @@ def test_summary_collector(
     snapshot.assert_match(data.iloc[:4].to_csv(), "summary_collector_3.csv")
     assert data.shape == (200, 2)
     non_existing_realization_index = 150
-    with pytest.raises(KeyError):
-        _ = snake_oil_default_storage.load_all_summary_data(
-            ["WWCT:OP1", "WWCT:OP2"],
-            realization_index=non_existing_realization_index,
-        )
+    summary_data = snake_oil_default_storage.load_all_summary_data(
+        ["WWCT:OP1", "WWCT:OP2"],
+        realization_index=non_existing_realization_index,
+    )
+
+    assert summary_data.empty
 
 
 def test_misfit_collector(snake_oil_case_storage, snake_oil_default_storage, snapshot):