mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-11-03 16:41:20 +00:00
45 lines
1.3 KiB
Go
45 lines
1.3 KiB
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")
|
|
}
|
|
|
|
func TestGPT6SolLunaModelIdentity(t *testing.T) {
|
|
for _, model := range []string{"gpt-6-sol", "gpt-6-luna"} {
|
|
require.Contains(t, DefaultModelIDs(), model)
|
|
require.True(t, IsGPT6SolOrLunaModelSpelling(model))
|
|
}
|
|
require.False(t, IsGPT6SolOrLunaModelSpelling("gpt-6-astra"))
|
|
require.False(t, IsGPT6SolOrLunaModelSpelling("gpt-6-solitude"))
|
|
require.False(t, IsGPT6SolOrLunaModelSpelling("gpt-6-luna-preview"))
|
|
}
|