Use fuzzed data in tests of generate_code_lists()

iiasa · Jan 24, 2024 · 2ac60df · 2ac60df
1 parent af871f1
commit 2ac60df
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/message_ix_models/tests/tools/iea/test_web.py b/message_ix_models/tests/tools/iea/test_web.py
@@ -93,7 +93,7 @@ def test_load_data(test_context, tmp_path, provider, edition):
 
 
 @pytest.mark.parametrize("provider, edition", list(FILES.keys()))
-def test_generate_code_lists(tmp_path, test_context, provider, edition):
+def test_generate_code_lists(tmp_path, provider, edition):
     # generate_code_lists() runs
     generate_code_lists(provider, edition, tmp_path)
 

diff --git a/message_ix_models/tools/iea/web.py b/message_ix_models/tools/iea/web.py
@@ -275,8 +275,13 @@ def generate_code_lists(
     register_agency(IEA)
 
     # Read the data
+    if HAS_MESSAGE_DATA:
+        path = private_data_path("iea")
+    else:
+        path = package_data_path("test", "iea")
+        log.warning(f"Reading random data from {path}")
     data = iea_web_data_for_query(
-        private_data_path("iea"), *FILES[(provider, edition)], query_expr="TIME > 0"
+        path, *FILES[(provider, edition)], query_expr="TIME > 0"
     )
 
     for concept_id in ("COUNTRY", "FLOW", "PRODUCT"):