Codebase list golang-github-beorn7-perks / ba304a31-db3f-4ab8-bbba-f7317978bd64/main histogram / histogram_test.go
ba304a31-db3f-4ab8-bbba-f7317978bd64/main

Tree @ba304a31-db3f-4ab8-bbba-f7317978bd64/main (Download .tar.gz)

histogram_test.go @ba304a31-db3f-4ab8-bbba-f7317978bd64/mainraw · history · blame

package histogram

import (
	"math/rand"
	"testing"
)

func TestHistogram(t *testing.T) {
	const numPoints = 1000000
	const maxBins = 3

	h := New(maxBins)
	for i := 0; i < numPoints; i++ {
		f := rand.ExpFloat64()
		h.Insert(f)
	}

	bins := h.Bins()
	if g := len(bins); g > maxBins {
		t.Fatalf("got %d bins, wanted <= %d", g, maxBins)
	}

	for _, b := range bins {
		t.Logf("%+v", b)
	}

	if g := count(h.Bins()); g != numPoints {
		t.Fatalf("binned %d points, wanted %d", g, numPoints)
	}
}

func count(bins Bins) int {
	binCounts := 0
	for _, b := range bins {
		binCounts += b.Count
	}
	return binCounts
}