import pytest

from scripts.assess_belgium_building_training_iteration import (
    INFERENCE_CONFIG_FIELDS,
    assert_same_inference_config,
)


def test_regional_model_routes_are_frozen_in_inference_config() -> None:
    assert "regional_models" in INFERENCE_CONFIG_FIELDS
    calibration = {field: None for field in INFERENCE_CONFIG_FIELDS}
    calibration["regional_models"] = {"flanders": "/models/flanders.pt"}
    test = dict(calibration)
    test["regional_models"] = {"flanders": "/models/other.pt"}

    with pytest.raises(ValueError, match="regional_models"):
        assert_same_inference_config(calibration, test, "test")


def test_regional_tile_counts_may_differ_between_splits() -> None:
    calibration = {field: None for field in INFERENCE_CONFIG_FIELDS}
    test = dict(calibration)
    calibration["regional_model_tile_counts"] = {"flanders": 48}
    test["regional_model_tile_counts"] = {"flanders": 64}

    assert_same_inference_config(calibration, test, "test")