Codex
bb7968b612
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 9m20s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 55m7s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
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"))
|
|
}
|