54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
package gago
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestMigSizes(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
migrators = []Migrator{
|
||
|
|
MigRing{
|
||
|
|
NMigrants: 5,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
for _, migrator := range migrators {
|
||
|
|
for _, nbrPops := range []int{2, 3, 10} {
|
||
|
|
var fitnessMeans = make([]float64, nbrPops)
|
||
|
|
for _, nbrIndis := range []int{6, 10, 30} {
|
||
|
|
// Instantiate populations
|
||
|
|
var pops = make([]Population, nbrPops)
|
||
|
|
for i := range pops {
|
||
|
|
pops[i] = newPopulation(nbrIndis, NewVector, randString(3, rng))
|
||
|
|
pops[i].Individuals.Evaluate()
|
||
|
|
fitnessMeans[i] = pops[i].Individuals.FitAvg()
|
||
|
|
}
|
||
|
|
migrator.Apply(pops, rng)
|
||
|
|
// Check the Population sizes haven't changed
|
||
|
|
for _, pop := range pops {
|
||
|
|
if len(pop.Individuals) != nbrIndis {
|
||
|
|
t.Error("Migration changed the Population sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// Check the average fitnesses have changed
|
||
|
|
for i, pop := range pops {
|
||
|
|
if pop.Individuals.FitAvg() == fitnessMeans[i] {
|
||
|
|
t.Error("Average fitnesses didn't change")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMigRingValidate(t *testing.T) {
|
||
|
|
var mig = MigRing{1}
|
||
|
|
if err := mig.Validate(); err != nil {
|
||
|
|
t.Error("Validation should not have raised error")
|
||
|
|
}
|
||
|
|
// Set NMigrants lower than 1
|
||
|
|
mig.NMigrants = 0
|
||
|
|
if err := mig.Validate(); err == nil {
|
||
|
|
t.Error("Validation should raised error")
|
||
|
|
}
|
||
|
|
}
|