123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428 |
- 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")
- }
- }
|