45 lines
846 B
Go
45 lines
846 B
Go
// stm: #unit
|
|
package messagepool
|
|
|
|
import (
|
|
"math"
|
|
"math/rand"
|
|
"testing"
|
|
)
|
|
|
|
func TestBlockProbability(t *testing.T) {
|
|
//stm: @OTHER_IMPLEMENTATION_BLOCK_PROB_001
|
|
mp := &MessagePool{}
|
|
bp := mp.blockProbabilities(1 - 0.15)
|
|
t.Logf("%+v\n", bp)
|
|
for i := 0; i < len(bp)-1; i++ {
|
|
if bp[i] < bp[i+1] {
|
|
t.Fatalf("expected decreasing block probabilities for this quality: %d %f %f",
|
|
i, bp[i], bp[i+1])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestWinnerProba(t *testing.T) {
|
|
//stm: @OTHER_IMPLEMENTATION_BLOCK_PROB_002
|
|
const N = 1000000
|
|
winnerProba := noWinnersProb()
|
|
sum := 0
|
|
for i := 0; i < N; i++ {
|
|
minersRand := rand.Float64()
|
|
j := 0
|
|
for ; j < MaxBlocks; j++ {
|
|
minersRand -= winnerProba[j]
|
|
if minersRand < 0 {
|
|
break
|
|
}
|
|
}
|
|
sum += j
|
|
}
|
|
|
|
if avg := float64(sum) / N; math.Abs(avg-5) > 0.01 {
|
|
t.Fatalf("avg too far off: %f", avg)
|
|
}
|
|
|
|
}
|