WIP mktchart 3

volume_profile_test.go

@@ -0,0 +1,72 @@
+package mujica
+
+import (
+	"math"
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+)
+
+func TestVolumeProfileOfAndValueArea(t *testing.T) {
+	cs := candlesOf([5]float64{10, 14, 10, 12, 400}, [5]float64{12, 12, 12, 12, 50}, [5]float64{11, 12, 10, 11, 100})
+	bins := VolumeProfileOf(cs, 4) // bins of 1: 10-11, 11-12, 12-13, 13-14
+	want := []float64{100 + 50, 100 + 50, 100 + 50, 100}
+	for i, b := range bins {
+		if math.Abs(b.Volume-want[i]) > 1e-9 || b.Low != 10+float64(i) {
+			t.Fatalf("bin %d %+v, want volume %v", i, b, want[i])
+		}
+	}
+	vb := []ProfileBin{{0, 1, 5}, {1, 2, 10}, {2, 3, 40}, {3, 4, 30}, {4, 5, 15}}
+	poc, lo, hi := ValueArea(vb, 0.7)
+	// 40, then 30 above: 70 of 100 is enough.
+	if poc != 2 || lo != 2 || hi != 3 {
+		t.Fatalf("value area %d %d..%d", poc, lo, hi)
+	}
+	if _, _, hi := ValueArea(vb, 1); hi != 4 {
+		t.Fatal("full value area")
+	}
+	defer func() {
+		if recover() == nil {
+			t.Fatal("zero bins accepted")
+		}
+	}()
+	VolumeProfileOf(cs, 0)
+}
+
+func TestVolumeProfileKeyboard(t *testing.T) {
+	vb := []ProfileBin{{0, 1, 5}, {1, 2, 10}, {2, 3, 40}, {3, 4, 30}, {4, 5, 15}}
+	var res ChartResult
+	tt := ui.NewTester(func(c *ui.Context) {
+		Use(c, Settings{})
+		res = VolumeProfile(c, vb, VolumeProfileOptions{Frame: ChartFrame{Label: "Profile"}})
+	}, 500, 360)
+	tt.Key(0, ui.KeyTab)
+	tt.Key(0, ui.KeyUp)
+	tt.Key(0, ui.KeyUp)
+	tt.Key(0, ui.KeyUp)
+	if res.Point != 2 || !tt.HasText("2.00–3.00 · Vol 40 · POC · Value area") {
+		t.Fatalf("POC readout %d %q", res.Point, tt.Texts())
+	}
+	tt.Key(0, ui.KeyDown)
+	if !tt.HasText("1.00–2.00 · Vol 10") || tt.HasText("1.00–2.00 · Vol 10 · Value area") {
+		t.Fatalf("outside readout %q", tt.Texts())
+	}
+	defer func() {
+		if recover() == nil {
+			t.Fatal("overlapping bins accepted")
+		}
+	}()
+	ui.NewTester(func(c *ui.Context) {
+		Use(c, Settings{})
+		VolumeProfile(c, []ProfileBin{{0, 2, 1}, {1, 3, 1}}, VolumeProfileOptions{Frame: ChartFrame{Label: "x"}})
+	}, 200, 100)
+}
+
+func TestVolumeProfileSnapshot(t *testing.T) {
+	bins := VolumeProfileOf(sampleCandles(200), 18)
+	snapshot(t, "volume_profile", 480, 380, func(c *ui.Context) {
+		ui.Column(c).Padding(12).Children(func() {
+			VolumeProfile(c, bins, VolumeProfileOptions{Frame: ChartFrame{Label: "Profile", Height: 290}})
+		})
+	})
+}