mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-03-30 10:16:52 +00:00
24 lines
445 B
Go
24 lines
445 B
Go
package geminicli
|
|
|
|
import "testing"
|
|
|
|
func TestDefaultModels_ContainsImageModels(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
byID := make(map[string]Model, len(DefaultModels))
|
|
for _, model := range DefaultModels {
|
|
byID[model.ID] = model
|
|
}
|
|
|
|
required := []string{
|
|
"gemini-2.5-flash-image",
|
|
"gemini-3.1-flash-image",
|
|
}
|
|
|
|
for _, id := range required {
|
|
if _, ok := byID[id]; !ok {
|
|
t.Fatalf("expected curated Gemini model %q to exist", id)
|
|
}
|
|
}
|
|
}
|