wormhole/adaptive_config_test.go at main · garyblankenship/wormhole · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package wormhole

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestNormalizeEnhancedAdaptiveConfigFillsPartialPIDDefaults(t *testing.T) {
	t.Parallel()

	normalized := normalizeEnhancedAdaptiveConfig(EnhancedAdaptiveConfig{
		AdaptiveConfig: AdaptiveConfig{
			TargetLatency:      250 * time.Millisecond,
			MinCapacity:        2,
			MaxCapacity:        20,
			InitialCapacity:    4,
			AdjustmentInterval: time.Second,
			LatencyWindowSize:  8,
		},
		PIDConfig: PIDConfig{
			Kp: 1.25,
		},
	})

	defaults := DefaultPIDConfig()
	assert.Equal(t, 1.25, normalized.PIDConfig.Kp)
	assert.Equal(t, defaults.Ki, normalized.PIDConfig.Ki)
	assert.Equal(t, defaults.Kd, normalized.PIDConfig.Kd)
	assert.Equal(t, defaults.MaxIntegral, normalized.PIDConfig.MaxIntegral)
	assert.Equal(t, defaults.MinIntegral, normalized.PIDConfig.MinIntegral)
	assert.Equal(t, defaults.MaxOutput, normalized.PIDConfig.MaxOutput)
	assert.Equal(t, defaults.MinOutput, normalized.PIDConfig.MinOutput)
}

func TestNormalizeEnhancedAdaptiveConfigFillsProviderPartialPIDDefaults(t *testing.T) {
	t.Parallel()

	normalized := normalizeEnhancedAdaptiveConfig(EnhancedAdaptiveConfig{
		AdaptiveConfig: AdaptiveConfig{
			TargetLatency:      250 * time.Millisecond,
			MinCapacity:        2,
			MaxCapacity:        20,
			InitialCapacity:    4,
			AdjustmentInterval: time.Second,
			LatencyWindowSize:  8,
		},
		PIDConfig: PIDConfig{
			Ki: 0.2,
		},
		ProviderSettings: map[string]ProviderSetting{
			"openai": {
				PIDConfig: &PIDConfig{
					Kp: 3.0,
				},
			},
		},
	})

	providerConfig := normalized.ProviderSettings["openai"].PIDConfig
	require.NotNil(t, providerConfig)
	defaults := DefaultPIDConfig()
	assert.Equal(t, 3.0, providerConfig.Kp)
	assert.Equal(t, 0.2, providerConfig.Ki)
	assert.Equal(t, defaults.Kd, providerConfig.Kd)
	assert.Equal(t, defaults.MaxIntegral, providerConfig.MaxIntegral)
	assert.Equal(t, defaults.MinIntegral, providerConfig.MinIntegral)
	assert.Equal(t, defaults.MaxOutput, providerConfig.MaxOutput)
	assert.Equal(t, defaults.MinOutput, providerConfig.MinOutput)
}