Add tests of IEA_EWEB

Author: khaeru

message_ix_models/tests/tools/iea/test_web.py

@@ -3,7 +3,9 @@
 
 import pandas as pd
 import pytest
+from genno import Computer
 
+from message_ix_models.tools.exo_data import prepare_computer
 from message_ix_models.tools.iea.web import (
     DIMS,
     FILES,
@@ -14,6 +16,53 @@
 from message_ix_models.util import package_data_path
 
 
+class TestIEA_EWEB:
+    @pytest.mark.parametrize("source", ("IEA_EWEB",))
+    @pytest.mark.parametrize(
+        "source_kw",
+        (
+            dict(
+                provider="OECD", edition="2021", product=["CHARCOAL"], flow=["RESIDENT"]
+            ),
+            # All flows related to transport
+            dict(
+                provider="OECD",
+                edition="2022",
+                flow=[
+                    "DOMESAIR",
+                    "DOMESNAV",
+                    "PIPELINE",
+                    "RAIL",
+                    "ROAD",
+                    "TOTTRANS",
+                    "TRNONSPE",
+                    "WORLDAV",
+                    "WORLDMAR",
+                ],
+            ),
+        ),
+    )
+    def test_prepare_computer(self, test_context, source, source_kw):
+        # FIXME The following should be redundant, but appears mutable on GHA linux and
+        #       Windows runners.
+        test_context.model.regions = "R14"
+
+        c = Computer()
+
+        keys = prepare_computer(test_context, c, source, source_kw)
+
+        # Preparation of data runs successfully
+        result = c.get(keys[0])
+        # print(result.to_string())
+
+        # Data has the expected dimensions
+        assert {"n", "y", "product", "flow"} == set(result.dims)
+
+        # Data is complete
+        assert 14 == len(result.coords["n"])
+        assert {1980, 2020} < set(result.coords["y"].data)
+
+
 @pytest.mark.parametrize("provider, edition", FILES.keys())
 def test_load_data(test_context, tmp_path, provider, edition):
     # # Store in the temporary directory for this test