individual_test.go

package eaopt

import (
	"fmt"
	"testing"
)

func TestIndividualString(t *testing.T) {
	var testCases = []struct {
		indi Individual
		str  string
	}{
		{
			indi: Individual{
				Genome:    Vector{0, 1, 2},
				Fitness:   42,
				Evaluated: true,
				ID:        "bob",
			},
			str: "bob - 42.000 - [0 1 2]",
		},
		{
			indi: Individual{
				Genome:    Vector{0, 1, 2},
				Fitness:   42,
				Evaluated: false,
				ID:        "ALICE",
			},
			str: "ALICE - ??? - [0 1 2]",
		},
	}
	for i, tc := range testCases {
		t.Run(fmt.Sprintf("TC %d", i), func(t *testing.T) {
			if tc.indi.String() != tc.str {
				t.Errorf("Expected %s, got %s", tc.str, tc.indi.String())
			}
		})
	}
}

func TestCloneIndividual(t *testing.T) {
	var (
		rng    = newRand()
		genome = NewVector(rng)
		indi1  = NewIndividual(genome, rng)
		indi2  = indi1.Clone(rng)
	)
	if &indi1 == &indi2 || &indi1.Genome == &indi2.Genome {
		t.Error("Individual was not deep copied")
	}
}

func TestEvaluateIndividual(t *testing.T) {
	var (
		rng    = newRand()
		genome = NewVector(rng)
		indi   = NewIndividual(genome, rng)
	)
	if indi.Evaluated {
		t.Error("Individual shouldn't have Evaluated set to True")
	}
	if indi.Evaluate() != nil {
		t.Error("No error should have been raised")
	}
	if !indi.Evaluated {
		t.Error("Individual should have Evaluated set to True")
	}
}

func TestEvaluateIndividualWithError(t *testing.T) {
	var (
		rng    = newRand()
		genome = NewRuntimeErrorGenome(rng)
		indi   = NewIndividual(genome, rng)
	)
	if indi.Evaluate() == nil {
		t.Error("An error should have been raised")
	}
	if indi.Evaluated {
		t.Error("The individual shouldn't have Evaluated set to true")
	}
}

func TestMutateIndividual(t *testing.T) {
	var (
		rng    = newRand()
		genome = NewVector(rng)
		indi   = NewIndividual(genome, rng)
	)
	indi.Evaluate()
	indi.Mutate(rng)
	if indi.Evaluated {
		t.Error("Individual shouldn't have Evaluated set to True")
	}
}

func TestCrossoverIndividual(t *testing.T) {
	var (
		rng        = newRand()
		indi1      = NewIndividual(NewVector(rng), rng)
		indi2      = NewIndividual(NewVector(rng), rng)
		offspring1 = indi1.Clone(rng)
		offspring2 = indi2.Clone(rng)
	)
	indi1.Crossover(indi2, rng)
	if offspring1.Evaluated || offspring2.Evaluated {
		t.Error("Offsprings shouldn't have Evaluated set to True")
	}
	if &offspring1 == &indi1 || &offspring1 == &indi2 || &offspring2 == &indi1 || &offspring2 == &indi2 {
		t.Error("Offsprings shouldn't share pointers with parents")
	}
}