429 lines
10 KiB
Go
429 lines
10 KiB
Go
|
package gago
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCrossUniformFloat64(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = NewVector(rng).(Vector)
|
||
|
|
p2 = NewVector(rng).(Vector)
|
||
|
|
o1, o2 = CrossUniformFloat64(p1, p2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossUniform should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
// Check new values are contained in hyper-rectangle defined by parents
|
||
|
|
var (
|
||
|
|
bounded = func(x, lower, upper float64) bool { return x > lower && x < upper }
|
||
|
|
lower float64
|
||
|
|
upper float64
|
||
|
|
)
|
||
|
|
for i := 0; i < len(p1); i++ {
|
||
|
|
lower = math.Min(p1[i], p2[i])
|
||
|
|
upper = math.Max(p1[i], p2[i])
|
||
|
|
if !bounded(o1[i], lower, upper) || !bounded(o2[i], lower, upper) {
|
||
|
|
t.Error("New values are not contained in hyper-rectangle")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGNX(t *testing.T) {
|
||
|
|
var testCases = []struct {
|
||
|
|
p1 []int
|
||
|
|
p2 []int
|
||
|
|
indexes []int
|
||
|
|
o1 []int
|
||
|
|
o2 []int
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
indexes: []int{3, 7},
|
||
|
|
o1: []int{1, 2, 3, 8, 2, 6, 5, 8, 9},
|
||
|
|
o2: []int{9, 3, 7, 4, 5, 6, 7, 1, 4},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3},
|
||
|
|
p2: []int{3, 2, 1},
|
||
|
|
indexes: []int{0},
|
||
|
|
o1: []int{3, 2, 1},
|
||
|
|
o2: []int{1, 2, 3},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3},
|
||
|
|
p2: []int{3, 2, 1},
|
||
|
|
indexes: []int{1},
|
||
|
|
o1: []int{1, 2, 1},
|
||
|
|
o2: []int{3, 2, 3},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3},
|
||
|
|
p2: []int{3, 2, 1},
|
||
|
|
indexes: []int{2},
|
||
|
|
o1: []int{1, 2, 1},
|
||
|
|
o2: []int{3, 2, 3},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3},
|
||
|
|
p2: []int{3, 2, 1},
|
||
|
|
indexes: []int{3},
|
||
|
|
o1: []int{1, 2, 3},
|
||
|
|
o2: []int{3, 2, 1},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, test := range testCases {
|
||
|
|
var (
|
||
|
|
n = len(test.p1)
|
||
|
|
o1, o2 = gnx(IntSlice(test.p1), IntSlice(test.p2), test.indexes)
|
||
|
|
)
|
||
|
|
for i := 0; i < n; i++ {
|
||
|
|
if o1.At(i).(int) != test.o1[i] || o2.At(i).(int) != test.o2[i] {
|
||
|
|
t.Error("Something went wrong during GNX crossover")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossGNXFloat64(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []float64{1, 2, 3}
|
||
|
|
p2 = []float64{3, 2, 1}
|
||
|
|
o1, o2 = CrossGNXFloat64(p1, p2, 2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossGNXFloat64 should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossGNXInt(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []int{1, 2, 3}
|
||
|
|
p2 = []int{3, 2, 1}
|
||
|
|
o1, o2 = CrossGNXInt(p1, p2, 2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossGNXInt should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossGNXString(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []string{"a", "b", "c"}
|
||
|
|
p2 = []string{"c", "b", "a"}
|
||
|
|
o1, o2 = CrossGNXString(p1, p2, 2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossGNXString should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPMX(t *testing.T) {
|
||
|
|
var testCases = []struct {
|
||
|
|
p1 []int
|
||
|
|
p2 []int
|
||
|
|
a int
|
||
|
|
b int
|
||
|
|
o1 []int
|
||
|
|
o2 []int
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
a: 3,
|
||
|
|
b: 7,
|
||
|
|
o1: []int{9, 3, 2, 4, 5, 6, 7, 1, 8},
|
||
|
|
o2: []int{1, 7, 3, 8, 2, 6, 5, 4, 9},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
a: 0,
|
||
|
|
b: 9,
|
||
|
|
o1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
o2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
a: 0,
|
||
|
|
b: 0,
|
||
|
|
o1: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
o2: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, test := range testCases {
|
||
|
|
var (
|
||
|
|
n = len(test.p1)
|
||
|
|
o1, o2 = pmx(IntSlice(test.p1), IntSlice(test.p2), test.a, test.b)
|
||
|
|
)
|
||
|
|
for i := 0; i < n; i++ {
|
||
|
|
if o1.At(i).(int) != test.o1[i] || o2.At(i).(int) != test.o2[i] {
|
||
|
|
t.Error("Something went wrong during PMX crossover")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossPMXFloat64(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []float64{1, 2, 3}
|
||
|
|
p2 = []float64{3, 2, 1}
|
||
|
|
o1, o2 = CrossPMXFloat64(p1, p2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossPMXFloat64 should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossPMXInt(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []int{1, 2, 3}
|
||
|
|
p2 = []int{3, 2, 1}
|
||
|
|
o1, o2 = CrossPMXInt(p1, p2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossPMXInt should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossPMXString(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []string{"a", "b", "c"}
|
||
|
|
p2 = []string{"c", "b", "a"}
|
||
|
|
o1, o2 = CrossPMXString(p1, p2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossPMXString should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOX(t *testing.T) {
|
||
|
|
var testCases = []struct {
|
||
|
|
p1 []int
|
||
|
|
p2 []int
|
||
|
|
a int
|
||
|
|
b int
|
||
|
|
o1 []int
|
||
|
|
o2 []int
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
a: 3,
|
||
|
|
b: 7,
|
||
|
|
o1: []int{3, 8, 2, 4, 5, 6, 7, 1, 9},
|
||
|
|
o2: []int{3, 4, 7, 8, 2, 6, 5, 9, 1},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
a: 0,
|
||
|
|
b: 9,
|
||
|
|
o1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
o2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
a: 0,
|
||
|
|
b: 0,
|
||
|
|
o1: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
o2: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, test := range testCases {
|
||
|
|
var (
|
||
|
|
n = len(test.p1)
|
||
|
|
o1, o2 = ox(IntSlice(test.p1), IntSlice(test.p2), test.a, test.b)
|
||
|
|
)
|
||
|
|
for i := 0; i < n; i++ {
|
||
|
|
if o1.At(i).(int) != test.o1[i] || o2.At(i).(int) != test.o2[i] {
|
||
|
|
t.Error("Something went wrong during OX crossover")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossOXFloat64(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []float64{1, 2, 3}
|
||
|
|
p2 = []float64{3, 2, 1}
|
||
|
|
o1, o2 = CrossOXFloat64(p1, p2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossOXFloat64 should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossOXInt(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []int{1, 2, 3}
|
||
|
|
p2 = []int{3, 2, 1}
|
||
|
|
o1, o2 = CrossOXInt(p1, p2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossOXInt should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossOXString(t *testing.T) {
|
||
|
|
var (
|
||
|
|
rng = newRandomNumberGenerator()
|
||
|
|
p1 = []string{"a", "b", "c"}
|
||
|
|
p2 = []string{"c", "b", "a"}
|
||
|
|
o1, o2 = CrossOXString(p1, p2, rng)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossOXString should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossCX(t *testing.T) {
|
||
|
|
var testCases = []struct {
|
||
|
|
p1 []int
|
||
|
|
p2 []int
|
||
|
|
o1 []int
|
||
|
|
o2 []int
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{9, 3, 7, 8, 2, 6, 5, 1, 4},
|
||
|
|
o1: []int{1, 3, 7, 4, 2, 6, 5, 8, 9},
|
||
|
|
o2: []int{9, 2, 3, 8, 5, 6, 7, 1, 4},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
p1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
p2: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
o1: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
o2: []int{1, 2, 3, 4, 5, 6, 7, 8, 9},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, test := range testCases {
|
||
|
|
var (
|
||
|
|
n = len(test.p1)
|
||
|
|
o1, o2 = CrossCX(IntSlice(test.p1), IntSlice(test.p2))
|
||
|
|
)
|
||
|
|
for i := 0; i < n; i++ {
|
||
|
|
if o1.At(i).(int) != test.o1[i] || o2.At(i).(int) != test.o2[i] {
|
||
|
|
t.Error("Something went wrong during CX crossover")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossCXFloat64(t *testing.T) {
|
||
|
|
var (
|
||
|
|
p1 = []float64{1, 2, 3}
|
||
|
|
p2 = []float64{3, 2, 1}
|
||
|
|
o1, o2 = CrossCXFloat64(p1, p2)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossCXFloat64 should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossCXInt(t *testing.T) {
|
||
|
|
var (
|
||
|
|
p1 = []int{1, 2, 3}
|
||
|
|
p2 = []int{3, 2, 1}
|
||
|
|
o1, o2 = CrossCXInt(p1, p2)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossCXInt should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossCXString(t *testing.T) {
|
||
|
|
var (
|
||
|
|
p1 = []string{"a", "b", "c"}
|
||
|
|
p2 = []string{"c", "b", "a"}
|
||
|
|
o1, o2 = CrossCXString(p1, p2)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossCXString should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossERX(t *testing.T) {
|
||
|
|
var testCases = []struct {
|
||
|
|
p1 []string
|
||
|
|
p2 []string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
p1: []string{"A", "B", "F", "E", "D", "G", "C"},
|
||
|
|
p2: []string{"G", "F", "A", "B", "C", "D", "E"},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, test := range testCases {
|
||
|
|
var o1, o2 = CrossERX(StringSlice(test.p1), StringSlice(test.p2))
|
||
|
|
// Check offsprings have parent's first gene as first gene
|
||
|
|
if o1.At(0).(string) != test.p1[0] || o2.At(0).(string) != test.p2[0] {
|
||
|
|
t.Error("Something went wrong during ERX crossover")
|
||
|
|
}
|
||
|
|
// Check lengths
|
||
|
|
if o1.Len() != len(test.p1) || o2.Len() != len(test.p2) {
|
||
|
|
t.Error("Something went wrong during ERX crossover")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossERXFloat64(t *testing.T) {
|
||
|
|
var (
|
||
|
|
p1 = []float64{1, 2, 3}
|
||
|
|
p2 = []float64{3, 2, 1}
|
||
|
|
o1, o2 = CrossERXFloat64(p1, p2)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossERXFloat64 should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossERXInt(t *testing.T) {
|
||
|
|
var (
|
||
|
|
p1 = []int{1, 2, 3}
|
||
|
|
p2 = []int{3, 2, 1}
|
||
|
|
o1, o2 = CrossERXInt(p1, p2)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossERXInt should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCrossERXString(t *testing.T) {
|
||
|
|
var (
|
||
|
|
p1 = []string{"a", "b", "c"}
|
||
|
|
p2 = []string{"c", "b", "a"}
|
||
|
|
o1, o2 = CrossERXString(p1, p2)
|
||
|
|
)
|
||
|
|
// Check lengths
|
||
|
|
if len(o1) != len(p1) || len(o2) != len(p1) {
|
||
|
|
t.Error("CrossERXString should not produce offsprings with different sizes")
|
||
|
|
}
|
||
|
|
}
|