mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-09-07 10:07:07 +00:00
34 lines
1.3 KiB
Go
34 lines
1.3 KiB
Go
package task_pricing_setting
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/QuantumNous/new-api/setting/config"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTaskPricingDefaultsAndOptionUpdate(t *testing.T) {
|
|
original := GetCopy()
|
|
t.Cleanup(func() { taskPricingSetting = original })
|
|
assert.InDelta(t, 1.666667, SoraSizeRatio("1792x1024"), 0.000001)
|
|
assert.InDelta(t, 2.333333, VertexResolutionRatio("veo-3.1-fast-generate-preview", "4K"), 0.000001)
|
|
require.NoError(t, config.UpdateConfigFromMap(&taskPricingSetting, map[string]string{"sora_size_ratio": `{"1792x1024":2}`}))
|
|
assert.Equal(t, 2.0, SoraSizeRatio("1792x1024"))
|
|
assert.Equal(t, 1.0, SoraSizeRatio("720x1280"))
|
|
}
|
|
|
|
func TestVertexResolutionRatioPrefersMostSpecificModelPattern(t *testing.T) {
|
|
original := GetCopy()
|
|
t.Cleanup(func() { taskPricingSetting = original })
|
|
taskPricingSetting.VertexResolution4K = map[string]float64{
|
|
"veo-3.1": 1.5,
|
|
"veo-3.1-fast-generate": 2.333333,
|
|
"veo-3.1-fast-generate-preview": 3,
|
|
}
|
|
|
|
assert.Equal(t, 3.0, VertexResolutionRatio("veo-3.1-fast-generate-preview", "4K"))
|
|
assert.Equal(t, 2.333333, VertexResolutionRatio("veo-3.1-fast-generate", "4k"))
|
|
assert.Equal(t, 1.5, VertexResolutionRatio("veo-3.1-generate", "4K"))
|
|
}
|