mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-03-30 02:27:11 +00:00
29 lines
616 B
Go
29 lines
616 B
Go
package gemini
|
|
|
|
import "testing"
|
|
|
|
func TestDefaultModels_ContainsImageModels(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
models := DefaultModels()
|
|
byName := make(map[string]Model, len(models))
|
|
for _, model := range models {
|
|
byName[model.Name] = model
|
|
}
|
|
|
|
required := []string{
|
|
"models/gemini-2.5-flash-image",
|
|
"models/gemini-3.1-flash-image",
|
|
}
|
|
|
|
for _, name := range required {
|
|
model, ok := byName[name]
|
|
if !ok {
|
|
t.Fatalf("expected fallback model %q to exist", name)
|
|
}
|
|
if len(model.SupportedGenerationMethods) == 0 {
|
|
t.Fatalf("expected fallback model %q to advertise generation methods", name)
|
|
}
|
|
}
|
|
}
|