go-zero/core/mathx/entropy_test.go

package mathx

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestCalcEntropy(t *testing.T) {
	const total = 1000
	const count = 100
	m := make(map[interface{}]int, total)
	for i := 0; i < total; i++ {
		m[i] = count
	}
	assert.True(t, CalcEntropy(m, total*count) > .99)
}