about summary refs log tree commit diff
path: root/tools/seqstat/sequence_test.go
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--tools/seqstat/sequence_test.go49
1 files changed, 49 insertions, 0 deletions
diff --git a/tools/seqstat/sequence_test.go b/tools/seqstat/sequence_test.go
new file mode 100644
index 0000000..1299086
--- /dev/null
+++ b/tools/seqstat/sequence_test.go
@@ -0,0 +1,49 @@
+package main
+
+import (
+	"testing"
+)
+
+func TestHistogram(t *testing.T) {
+	cases := []struct {
+		in        []float64
+		histogram []rune
+	}{
+		{[]float64{1, 2}, []rune{'▁', '█'}},
+		{[]float64{1, 10, 4}, []rune{'▁', '█', '▃'}},
+		{[]float64{1, 5, 22, 13, 53}, []rune{'▁', '▁', '▃', '▂', '█'}},
+	}
+
+	for _, c := range cases {
+		seq := newSequence(c.in)
+		if string(seq.histogram()) != string(c.histogram) {
+			t.Errorf("Not matching: got %q, want %q", string(seq.histogram()), string(c.histogram))
+		}
+	}
+}
+
+func TestStats(t *testing.T) {
+	cases := []struct {
+		in   []float64
+		min  float64
+		max  float64
+		p999 float64
+	}{
+		{[]float64{1, 10, 52, 12}, 1, 52, 52},
+	}
+
+	for _, c := range cases {
+		seq := newSequence(c.in)
+		if seq.min != c.min {
+			t.Errorf("Not matching: got min %f want %f", seq.min, c.min)
+		}
+		if seq.max != c.max {
+			if seq.max != c.max {
+				t.Errorf("Not matching: got max %f want %f", seq.max, c.max)
+			}
+			if seq.p999() != c.p999 {
+				t.Errorf("Not matching: got p999 %f want %f", seq.p999(), c.p999)
+			}
+		}
+	}
+}