116 lines
3.0 KiB
Go
116 lines
3.0 KiB
Go
package chat
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/core"
|
|
)
|
|
|
|
func paramSample(temp, topP *float64) []ModelParam {
|
|
return []ModelParam{
|
|
{Label: "Temperature", Value: temp, Min: 0, Max: 2, Step: 0.1, Default: 1},
|
|
{Label: "Top-p", Value: topP, Min: 0, Max: 1, Step: 0.05, Default: 1},
|
|
}
|
|
}
|
|
|
|
func TestParameterPanelAdjustAndReset(t *testing.T) {
|
|
temp, topP, n, off := 1.0, 0.9, 0, false
|
|
tt := ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
ui.Column(c).Padding(16).Disabled(off).Children(func() {
|
|
if ParameterPanel(c, paramSample(&temp, &topP)).Changed() {
|
|
n++
|
|
}
|
|
})
|
|
}, 500, 400)
|
|
for _, s := range []string{"1.0", "0.90", "Default 1.0", "2.0"} {
|
|
if !tt.HasText(s) {
|
|
t.Errorf("missing %q in %q", s, tt.Texts())
|
|
}
|
|
}
|
|
tt.Key(0, ui.KeyTab)
|
|
tt.Key(0, ui.KeyRight)
|
|
if temp != 1.1 || n != 1 {
|
|
t.Fatalf("Right: temp %v changes %d", temp, n)
|
|
}
|
|
_ = tt.Click("Reset Temperature")
|
|
if temp != 1 {
|
|
t.Fatalf("reset temp %v", temp)
|
|
}
|
|
_ = tt.Click("Reset all")
|
|
if topP != 1 || n != 3 {
|
|
t.Fatalf("reset all: topP %v changes %d", topP, n)
|
|
}
|
|
off = true
|
|
tt.Key(0, ui.KeyRight)
|
|
if temp != 1 {
|
|
t.Fatal("disabled panel changed")
|
|
}
|
|
}
|
|
|
|
func TestParameterPanelBadParam(t *testing.T) {
|
|
v := 0.0
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("no panic for a default outside the range")
|
|
}
|
|
}()
|
|
ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
ParameterPanel(c, []ModelParam{{Label: "x", Value: &v, Min: 0, Max: 1, Step: 0.1, Default: 3}})
|
|
}, 100, 100)
|
|
}
|
|
|
|
func TestParameterPanelSnapshot(t *testing.T) {
|
|
temp, topP := 0.7, 1.0
|
|
snapshot(t, "parameter_panel", 420, 260, func(c *ui.Context) {
|
|
ui.Column(c).Padding(16).Children(func() { ParameterPanel(c, paramSample(&temp, &topP)) })
|
|
})
|
|
}
|
|
|
|
func TestParameterPanelRejectsNonFiniteAndOutOfRange(t *testing.T) {
|
|
for _, v := range []float64{math.NaN(), math.Inf(1), -1, 3} {
|
|
t.Run(fmt.Sprint(v), func(t *testing.T) {
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("invalid parameter value accepted")
|
|
}
|
|
}()
|
|
ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
ParameterPanel(c, []ModelParam{{Label: "Temperature", Value: &v, Min: 0, Max: 2, Step: 0.1, Default: 1}})
|
|
}, 400, 300)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParameterPanelRequiresDefaultOnStepGrid(t *testing.T) {
|
|
v := 0.9
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("off-grid default accepted")
|
|
}
|
|
}()
|
|
ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
ParameterPanel(c, []ModelParam{{Label: "Temperature", Value: &v, Min: 0, Max: 2, Step: 0.1, Default: 0.95}})
|
|
}, 400, 300)
|
|
}
|
|
|
|
func TestParameterPanelRequiresValueOnStepGrid(t *testing.T) {
|
|
v := 0.5
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("off-grid value accepted")
|
|
}
|
|
}()
|
|
ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
ParameterPanel(c, []ModelParam{{Label: "Temperature", Value: &v, Min: 0, Max: 1, Step: 0.2, Default: 0.4}})
|
|
}, 400, 300)
|
|
}
|