51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
package domain
|
|
|
|
import "testing"
|
|
|
|
func TestDefaultAntigravityModelMapping_ImageCompatibilityAliases(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cases := map[string]string{
|
|
"gemini-2.5-flash-image": "gemini-2.5-flash-image",
|
|
"gemini-2.5-flash-image-preview": "gemini-2.5-flash-image",
|
|
"gemini-3.1-flash-image": "gemini-3.1-flash-image",
|
|
"gemini-3.1-flash-image-preview": "gemini-3.1-flash-image",
|
|
"gemini-3-pro-image": "gemini-3.1-flash-image",
|
|
"gemini-3-pro-image-preview": "gemini-3.1-flash-image",
|
|
}
|
|
|
|
for from, want := range cases {
|
|
got, ok := DefaultAntigravityModelMapping[from]
|
|
if !ok {
|
|
t.Fatalf("expected mapping for %q to exist", from)
|
|
}
|
|
if got != want {
|
|
t.Fatalf("unexpected mapping for %q: got %q want %q", from, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDefaultAntigravityModelMapping_ContainsOpus48(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got, ok := DefaultAntigravityModelMapping["claude-opus-4-8"]
|
|
if !ok {
|
|
t.Fatal("expected mapping for claude-opus-4-8 to exist")
|
|
}
|
|
if got != "claude-opus-4-8" {
|
|
t.Fatalf("unexpected claude-opus-4-8 mapping: got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestDefaultBedrockModelMapping_ContainsOpus48(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got, ok := DefaultBedrockModelMapping["claude-opus-4-8"]
|
|
if !ok {
|
|
t.Fatal("expected Bedrock mapping for claude-opus-4-8 to exist")
|
|
}
|
|
if got != "us.anthropic.claude-opus-4-8-v1" {
|
|
t.Fatalf("unexpected Bedrock claude-opus-4-8 mapping: got %q", got)
|
|
}
|
|
}
|