35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
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())
|