Split into category packages; module github.com/ZacharyZhang-NY/MujicaUI

chat/parameter_panel_test.go

@@ -0,0 +1,115 @@
+package chat
+
+import (
+	"fmt"
+	"math"
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+
+	"github.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)
+}