You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
563 B
Go
32 lines
563 B
Go
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)
|
|
for i := 0; i < total; i++ {
|
|
m[i] = count
|
|
}
|
|
assert.True(t, CalcEntropy(m) > .99)
|
|
}
|
|
|
|
func TestCalcEmptyEntropy(t *testing.T) {
|
|
m := make(map[any]int)
|
|
assert.Equal(t, float64(1), CalcEntropy(m))
|
|
}
|
|
|
|
func TestCalcDiffEntropy(t *testing.T) {
|
|
const total = 1000
|
|
m := make(map[any]int, total)
|
|
for i := 0; i < total; i++ {
|
|
m[i] = i
|
|
}
|
|
assert.True(t, CalcEntropy(m) < .99)
|
|
}
|