from scripts.provision_belgium_building_training_portfolio import AOIS


V63_CANDIDATES = {
    "aarschot-mixed-cal-candidate-v63",
    "beveren-ribbon-cal-candidate-v63",
    "oostkamp-suburban-cal-candidate-v63",
}
V66_CANDIDATES = {
    "zutendaal-lowrise-cal-candidate-v66",
    "zoersel-lowrise-cal-candidate-v66",
    "landen-rural-cal-candidate-v66",
}


def test_portfolio_slugs_are_unique() -> None:
    slugs = [aoi.slug for aoi in AOIS]
    assert len(slugs) == len(set(slugs))


def test_v63_calibration_candidates_are_flemish_train_role_inputs() -> None:
    candidates = {aoi.slug: aoi for aoi in AOIS if aoi.slug in V63_CANDIDATES}
    assert set(candidates) == V63_CANDIDATES
    assert all(aoi.region == "flanders" for aoi in candidates.values())
    assert all(aoi.split == "train" for aoi in candidates.values())
    assert all(aoi.sample_role == "positive" for aoi in candidates.values())


def test_v66_lowrise_candidates_are_flemish_train_role_inputs() -> None:
    candidates = {aoi.slug: aoi for aoi in AOIS if aoi.slug in V66_CANDIDATES}
    assert set(candidates) == V66_CANDIDATES
    assert all(aoi.region == "flanders" for aoi in candidates.values())
    assert all(aoi.split == "train" for aoi in candidates.values())
    assert all("lowrise" in aoi.context for aoi in candidates.values())