path: root/tools/seqstat/sequence_test.go

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)
			}
		}
	}
}