53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
"""Regression tests for #28156 — Bedrock picker must be region-scoped.
|
|
|
|
Geo-prefixed cross-region inference profiles (us.*, eu.*, apac.*, ...) only
|
|
route from endpoints in their own geography. Offering us.* profiles to an
|
|
eu-central-2 picker produces configs AWS rejects regardless of credentials.
|
|
"""
|
|
|
|
from hermes_cli.model_setup_flows import (
|
|
BEDROCK_GEO_PREFIXES,
|
|
bedrock_model_routable_from_region,
|
|
bedrock_region_geo_prefix,
|
|
)
|
|
|
|
|
|
class TestRegionGeoPrefix:
|
|
def test_known_geographies(self):
|
|
assert bedrock_region_geo_prefix("us-east-1") == "us."
|
|
assert bedrock_region_geo_prefix("eu-central-2") == "eu."
|
|
assert bedrock_region_geo_prefix("ap-southeast-1") == "ap."
|
|
assert bedrock_region_geo_prefix("ca-central-1") == "ca."
|
|
assert bedrock_region_geo_prefix("sa-east-1") == "sa."
|
|
assert bedrock_region_geo_prefix("me-south-1") == "me."
|
|
assert bedrock_region_geo_prefix("af-south-1") == "af."
|
|
|
|
def test_unknown_region_is_empty(self):
|
|
assert bedrock_region_geo_prefix("") == ""
|
|
assert bedrock_region_geo_prefix("moon-base-1") == ""
|
|
|
|
|
|
class TestRoutableFromRegion:
|
|
def test_us_profile_not_offered_in_eu(self):
|
|
assert not bedrock_model_routable_from_region(
|
|
"us.anthropic.claude-sonnet-4-6", "eu-central-2"
|
|
)
|
|
|
|
|
|
|
|
|
|
class TestGeoPrefixContract:
|
|
def test_every_geo_prefix_maps_to_a_routable_region_or_is_alias(self):
|
|
"""Invariant: each geo prefix is either reachable from some region's
|
|
geo mapping or an ap-family alias — no dead entries."""
|
|
mapped = {
|
|
bedrock_region_geo_prefix(r)
|
|
for r in (
|
|
"us-east-1", "eu-central-2", "ap-southeast-1",
|
|
"ca-central-1", "sa-east-1", "me-south-1", "af-south-1",
|
|
)
|
|
}
|
|
ap_aliases = {"apac.", "jp."}
|
|
for prefix in BEDROCK_GEO_PREFIXES:
|
|
assert prefix in mapped or prefix in ap_aliases
|