go-zero/core/mathx/entropy_test.go

18 lines
272 B
Go
Raw Normal View History

2020-07-26 17:09:05 +08:00
package mathx
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestCalcEntropy(t *testing.T) {
const total = 1000
const count = 100
m := make(map[any]int, total)
2020-07-26 17:09:05 +08:00
for i := 0; i < total; i++ {
m[i] = count
}
2020-08-06 23:06:26 +08:00
assert.True(t, CalcEntropy(m) > .99)
2020-07-26 17:09:05 +08:00
}