from message_ix_models.project.advance.data import LOCATION, NAME
from message_ix_models.tools.iamc import describe
from message_ix_models.util import MESSAGE_MODELS_PATH, package_data_path
[docs]
def test_describe(test_context):
import zipfile
import pandas as pd
path = package_data_path("test", *LOCATION)
with zipfile.ZipFile(path) as zf:
data = pd.read_csv(zf.open(NAME), engine="pyarrow").rename(
columns=lambda c: c.upper()
)
sm = describe(
data, f"ADVANCE data in {path.relative_to(MESSAGE_MODELS_PATH.parent)}"
)
# Message contains the expected code lists.
# Code lists have the expected lengths.
for id, N in (
("MODEL", 12),
("SCENARIO", 51),
("REGION", 14),
("VARIABLE", 3080),
("UNIT", 29),
):
assert N == len(sm.codelist[id])
# from message_ix_models.util.sdmx import write
# write(sm, basename="ADVANCE")