73 lines
2.2 KiB
Go
73 lines
2.2 KiB
Go
package mujica
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
)
|
|
|
|
var modelsSample = []ModelOption{
|
|
{ID: "herald", Name: "Herald 2", Capabilities: []string{"Vision", "Tools"}, ContextTokens: 200000},
|
|
{ID: "page", Name: "Page Mini", Capabilities: []string{"Fast"}, ContextTokens: 32000},
|
|
{ID: "oracle", Name: "Oracle", ContextTokens: 1000000, Unavailable: "Needs the Court plan"},
|
|
}
|
|
|
|
func modelTester(v *string, changes *int, off *bool) *ui.Tester {
|
|
return ui.NewTester(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
ui.Column(c).Padding(16).Children(func() {
|
|
if ModelSelector(c, v, modelsSample, ModelSelectorOptions{Disabled: *off}).Changed() {
|
|
*changes++
|
|
}
|
|
})
|
|
}, 500, 500)
|
|
}
|
|
|
|
func TestModelSelectorShowsDetailsAndChooses(t *testing.T) {
|
|
v, n, off := "herald", 0, false
|
|
tt := modelTester(&v, &n, &off)
|
|
_ = tt.Click("Model")
|
|
for _, s := range []string{"200K context", "Vision", "1M context", "Needs the Court plan", "32K context"} {
|
|
if !tt.HasText(s) {
|
|
t.Errorf("missing %q in %q", s, tt.Texts())
|
|
}
|
|
}
|
|
_ = tt.Click("Oracle")
|
|
if v != "herald" || n != 0 {
|
|
t.Fatalf("unavailable model chosen: %q", v)
|
|
}
|
|
tt.Key(0, ui.KeyEscape)
|
|
tt.Key(0, ui.KeyTab)
|
|
tt.Key(0, ui.KeyDown)
|
|
tt.Key(0, ui.KeyDown)
|
|
tt.Key(0, ui.KeyDown)
|
|
tt.Key(0, ui.KeyEnter)
|
|
if v != "page" || n != 1 {
|
|
t.Fatalf("keyboard choice %q changes %d (Down stops before the unavailable model)", v, n)
|
|
}
|
|
}
|
|
|
|
func TestModelSelectorDisabled(t *testing.T) {
|
|
v, n, off := "herald", 0, true
|
|
tt := modelTester(&v, &n, &off)
|
|
_ = tt.Click("Model")
|
|
if tt.HasText("Page Mini") {
|
|
t.Fatal("disabled selector opened")
|
|
}
|
|
}
|
|
|
|
func TestModelSelectorTokens(t *testing.T) {
|
|
for n, want := range map[int]string{512: "512", 8000: "8K", 200000: "200K", 1000000: "1M", 1500000: "1.5M"} {
|
|
if got := modelSelectorTokens(n); got != want {
|
|
t.Errorf("%d: %q, want %q", n, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestModelSelectorSnapshot(t *testing.T) {
|
|
snapshotAfter(t, "model_selector", 440, 330, func(c *ui.Context) {
|
|
v := ui.Local(c.Root(), "v", func() string { return "herald" })
|
|
ui.Column(c).Padding(16).Children(func() { ModelSelector(c, v, modelsSample, ModelSelectorOptions{}) })
|
|
}, func(tt *ui.Tester) { _ = tt.Click("Model") })
|
|
}
|