Charts I: components 101-120

sankey_chart_test.go

@@ -0,0 +1,81 @@
+package mujica
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+)
+
+var sankeyNodes = []SankeyNode{{"tithe", "Tithes"}, {"alms", "Alms"}, {"fund", "Fund"}, {"roof", "Roof"}, {"bread", "Bread"}, {"bells", "Bells"}}
+var sankeyLinks = []SankeyLink{
+	{"tithe", "fund", 60}, {"alms", "fund", 30}, {"alms", "bread", 10},
+	{"fund", "roof", 50}, {"fund", "bells", 40},
+}
+
+func TestSankeyPlaceStable(t *testing.T) {
+	l := sankeyPlace(sankeyNodes, sankeyLinks)
+	if fmt.Sprint(l.col) != "[0 0 1 2 1 2]" || l.cols != 3 || l.in[2] != 90 || l.out[1] != 40 {
+		t.Fatalf("layout %+v", l)
+	}
+	lit := sankeyLit(l, len(sankeyNodes), 0) // Tithes: its fund, roof and bells, not alms
+	if !lit[2] || !lit[3] || !lit[5] || lit[1] || lit[4] || !lit[6] || lit[7] {
+		t.Fatalf("lit from Tithes %v", lit)
+	}
+	lit = sankeyLit(l, len(sankeyNodes), 6+3) // the fund → roof link
+	if !lit[0] || !lit[1] || !lit[3] || lit[5] || lit[6+4] {
+		t.Fatalf("lit from fund → roof %v", lit)
+	}
+}
+
+func TestSankeyChartReadout(t *testing.T) {
+	h := &chartHarness{build: func(c *ui.Context) ChartResult {
+		return SankeyChart(c, sankeyNodes, sankeyLinks, SankeyChartOptions{Frame: ChartFrame{Label: "Money"}})
+	}}
+	tt := ui.NewTester(h.view, 560, 440)
+	tabTo(t, tt, "Money")
+	tt.Key(0, ui.KeyRight)
+	tt.Key(0, ui.KeyRight)
+	tt.Key(0, ui.KeyRight)
+	if got := chartReadout(tt); got != "Fund · in 90 · out 90" {
+		t.Fatalf("node %q", got)
+	}
+	tt.Key(0, ui.KeyEnd)
+	if got := chartReadout(tt); got != "Fund → Bells · 40" {
+		t.Fatalf("link %q", got)
+	}
+	tt.Key(0, ui.KeyEscape)
+	b := h.res.Element.Bounds()
+	tt.Move(b.X+8+4, b.Y+10) // the first node, at the top left
+	if got := chartReadout(tt); got != "Tithes · in 0 · out 60" {
+		t.Fatalf("hover %q", got)
+	}
+}
+
+func TestSankeyChartBad(t *testing.T) {
+	for name, links := range map[string][]SankeyLink{
+		"cycle":   {{"tithe", "fund", 1}, {"fund", "tithe", 1}},
+		"unknown": {{"tithe", "nowhere", 1}},
+		"zero":    {{"tithe", "fund", 0}},
+	} {
+		func() {
+			defer func() {
+				if recover() == nil {
+					t.Errorf("%s: no panic", name)
+				}
+			}()
+			ui.NewTester(func(c *ui.Context) {
+				Use(c, Settings{})
+				SankeyChart(c, sankeyNodes, links, SankeyChartOptions{Frame: ChartFrame{Label: "x"}})
+			}, 300, 300)
+		}()
+	}
+}
+
+func TestSankeyChartSnapshot(t *testing.T) {
+	snapshot(t, "sankey_chart", 560, 320, func(c *ui.Context) {
+		ui.Column(c).Padding(16).Children(func() {
+			SankeyChart(c, sankeyNodes, sankeyLinks, SankeyChartOptions{Frame: ChartFrame{Label: "Money"}})
+		})
+	})
+}