Skip to content

Commit

Permalink
Add integer package
Browse files Browse the repository at this point in the history
  • Loading branch information
cbarrick committed Dec 7, 2015
1 parent 19e7e21 commit d24a1e8
Show file tree
Hide file tree
Showing 2 changed files with 72 additions and 0 deletions.
26 changes: 26 additions & 0 deletions integer/cross.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
package integer

import "math/rand"

// UniformX performs a uniform crossover of some parents into a child.
func UniformX(child []int, parents ...[]int) {
n := len(parents)
for i := range child {
child[i] = parents[rand.Intn(n)][i]
}
}

// PointX performs n-point crossover of two parents into a child.
func PointX(n int, child, mom, dad []int) {
if rand.Intn(2) == 0 {
mom, dad = dad, mom
}
for 0 < n {
i := rand.Intn(len(child)-n) + 1
copy(child, mom[:i])
child = child[i:]
mom, dad = dad[i:], mom[i:]
n--
}
copy(child, mom)
}
46 changes: 46 additions & 0 deletions integer/int_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
package integer_test

import (
"testing"

"github.com/cbarrick/evo/integer"
)

// cross.go
// -------------------------

func TestUniformX(t *testing.T) {
mom := make([]int, 8)
dad := make([]int, 8)
for i := range mom {
mom[i] = 1
}
for i := range dad {
dad[i] = 2
}
child := make([]int, 8)
integer.UniformX(child, mom, dad)
for i := range child {
if child[i] != mom[i] && child[i] != dad[i] {
t.Fail()
}
}
}

func TestPointX(t *testing.T) {
mom := make([]int, 8)
dad := make([]int, 8)
for i := range mom {
mom[i] = 1
}
for i := range dad {
dad[i] = 2
}
child := make([]int, 8)
integer.PointX(7, child, mom, dad)
for i := range child {
if child[i] != mom[i] && child[i] != dad[i] {
t.Fail()
}
}
}

0 comments on commit d24a1e8

Please sign in to comment.