bundle/v0/tests/test_task_dispatch.py

from __future__ import annotations

import pytest

from loss_aware_dro_repro.core import ContractError
from loss_aware_dro_repro.task_dispatch import task_suite


@pytest.mark.parametrize(
    "task_id,expected",
    [
        ("portfolio_gaussian_main/d001/r00/n010", "portfolio_gaussian_main"),
        ("portfolio_gaussian_coverage_ablation/d001/r00/n010", "portfolio_gaussian_coverage_ablation"),
        ("portfolio_gaussian_highdim/d001/r00/n010", "portfolio_gaussian_highdim"),
        ("portfolio_discrete/d001/r00/n010", "portfolio_discrete"),
        ("portfolio_gmm/d001/r00/n010", "portfolio_gmm"),
        ("regression_absolute_main/d001/r00/n010", "regression_absolute_main"),
        ("regression_squared/d001/r00/n010", "regression_squared"),
    ],
)
def test_dispatch_covers_every_paper_scale_suite(task_id, expected):
    assert task_suite(task_id) == expected


def test_dispatch_rejects_unknown_task():
    with pytest.raises(ContractError, match="does not resolve exactly once"):
        task_suite("missing/d001/r00/n010")