65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
package mujica
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
)
|
|
|
|
func tokenUsageSample() []TokenUsage {
|
|
return []TokenUsage{{Label: "Plan", Input: 1200, Output: 300}, {Label: "Search", Input: 4800, Output: 650}, {Label: "Write", Input: 2100, Output: 2900}}
|
|
}
|
|
|
|
func TestTokenUsageChartBarsKeyboard(t *testing.T) {
|
|
usage := tokenUsageSample()
|
|
h := &chartHarness{build: func(c *ui.Context) ChartResult { return TokenUsageChart(c, usage, TokenUsageChartOptions{}) }}
|
|
tt := ui.NewTester(h.view, 520, 420)
|
|
tabTo(t, tt, "Token usage")
|
|
want := []string{"Plan · Input 1200", "Plan · Output 300", "Search · Input 4800"}
|
|
for i, w := range want {
|
|
tt.Key(0, ui.KeyRight)
|
|
if got := chartReadout(tt); got != w {
|
|
t.Fatalf("mark %d readout %q, want %q", i, got, w)
|
|
}
|
|
}
|
|
if !tt.HasText("Input") || !tt.HasText("Output") {
|
|
t.Fatal("legend missing")
|
|
}
|
|
}
|
|
|
|
func TestTokenUsageChartLineEmptyInvalid(t *testing.T) {
|
|
usage := tokenUsageSample()
|
|
h := &chartHarness{build: func(c *ui.Context) ChartResult {
|
|
return TokenUsageChart(c, usage, TokenUsageChartOptions{Line: true, Frame: ChartFrame{Label: "Tokens per step"}})
|
|
}}
|
|
tt := ui.NewTester(h.view, 520, 420)
|
|
tabTo(t, tt, "Tokens per step")
|
|
tt.Key(0, ui.KeyRight)
|
|
if h.res.Point < 0 {
|
|
t.Fatal("line chart has no keyboard path")
|
|
}
|
|
usage = nil
|
|
tt.Frame()
|
|
if !tt.HasText("No data") {
|
|
t.Fatalf("empty: %q", tt.Texts())
|
|
}
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("negative tokens did not panic")
|
|
}
|
|
}()
|
|
ui.Render(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
TokenUsageChart(c, []TokenUsage{{Label: "x", Input: -1}}, TokenUsageChartOptions{})
|
|
}, 300, 200, 1)
|
|
}
|
|
|
|
func TestTokenUsageChartSnapshot(t *testing.T) {
|
|
usage := tokenUsageSample()
|
|
snapshot(t, "token_usage_chart", 520, 360, func(c *ui.Context) {
|
|
ui.Column(c).Padding(16).Children(func() {
|
|
TokenUsageChart(c, usage, TokenUsageChartOptions{})
|
|
})
|
|
})
|
|
}
|