mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-30 16:23:31 +00:00
35 lines
904 B
Go
35 lines
904 B
Go
package openai
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDefaultModelsIncludeBareGPT56Alias(t *testing.T) {
|
|
require.Contains(t, DefaultModelIDs(), "gpt-5.6")
|
|
}
|
|
|
|
func TestDefaultModelsIncludeGPT6Astra(t *testing.T) {
|
|
require.Contains(t, DefaultModelIDs(), "gpt-6-astra")
|
|
require.Contains(t, DefaultModelIDs(), "gpt-6")
|
|
var displayName string
|
|
for _, model := range DefaultModels {
|
|
if model.ID == "gpt-6-astra" {
|
|
displayName = model.DisplayName
|
|
break
|
|
}
|
|
}
|
|
require.Equal(t, "GPT-6 Astra", displayName)
|
|
}
|
|
|
|
func TestDefaultModelsPreferConcreteGPT56SolForAccountTests(t *testing.T) {
|
|
require.NotEmpty(t, DefaultModels)
|
|
require.Equal(t, "gpt-5.6-sol", DefaultModels[0].ID)
|
|
}
|
|
|
|
func TestDefaultModelsIncludeGPTImage25(t *testing.T) {
|
|
require.Contains(t, DefaultModelIDs(), "gpt-image-2.5-flare")
|
|
require.Contains(t, DefaultModelIDs(), "gpt-image-2.5-sunburst")
|
|
}
|