Files
sub2api/backend/internal/service/profit_control_threshold_helpers_test.go
T
李建琦 6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
2026-08-21 18:30:13 +08:00

39 lines
1.6 KiB
Go

package service
import (
"math"
"testing"
"github.com/stretchr/testify/require"
)
// TestClampProfitControlThreshold 钉死阈值清洗口径:非有限或为负一律按 0
// (等价于只放行免费上游),合法值原样保留。
func TestClampProfitControlThreshold(t *testing.T) {
require.Equal(t, 0.0, clampProfitControlThreshold(math.NaN()))
require.Equal(t, 0.0, clampProfitControlThreshold(math.Inf(1)))
require.Equal(t, 0.0, clampProfitControlThreshold(math.Inf(-1)))
require.Equal(t, 0.0, clampProfitControlThreshold(-0.5))
require.Equal(t, 0.0, clampProfitControlThreshold(0))
require.Equal(t, 0.7, clampProfitControlThreshold(0.7))
}
// TestProfitControlOverThreshold 钉死「越线」的唯一定义:相对 epsilon 吸收
// decimal(10,4) 落库与浮点乘法的边界误差,U == 阈值 判定为合格。
//
// 线上否决点(openAIProfitControlVetoReason)与 profit-preview 现在共用这个
// 函数与 clampProfitControlThreshold;此前各复制了一份,会随线上口径漂移,
// 导致预演结果与真实调度不一致。
func TestProfitControlOverThreshold(t *testing.T) {
require.False(t, profitControlOverThreshold(0.7, 0.7), "U == 阈值 必须判定为合格")
require.False(t, profitControlOverThreshold(0.6999, 0.7))
require.True(t, profitControlOverThreshold(0.7001, 0.7))
// 0.3*3 的浮点尾数不得被判成越线。
require.False(t, profitControlOverThreshold(0.1*3, 0.3))
// 阈值为 0 时只放行 0 成本上游。
require.False(t, profitControlOverThreshold(0, 0))
require.True(t, profitControlOverThreshold(0.0001, 0))
}