Codebase list golang-github-beorn7-perks / ab83c5a6-100c-4685-8404-8febe6ad6863/main histogram / histogram_test.go
ab83c5a6-100c-4685-8404-8febe6ad6863/main

Tree @ab83c5a6-100c-4685-8404-8febe6ad6863/main (Download .tar.gz)

histogram_test.go @ab83c5a6-100c-4685-8404-8febe6ad6863/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
}