Add Speciator runtime error in GA.Enhance

Author: MaxHalford

ga_test.go

@@ -271,3 +271,18 @@ func TestGAEnhanceModelRuntimeError(t *testing.T) {
 	}
 	ga.Model = model
 }
+
+func TestGAEnhanceSpeciatorRuntimeError(t *testing.T) {
+	var speciator = ga.Speciator
+	ga.Speciator = SpecRuntimeError{}
+	// Check invalid speciator doesn't raise error
+	if ga.Validate() != nil {
+		t.Errorf("Expected %s, got %s", nil, ga.Validate())
+	}
+	// Enhance
+	var err = ga.Enhance()
+	if err == nil {
+		t.Error("An error should have been raised")
+	}
+	ga.Speciator = speciator
+}

setup_test.go

@@ -87,3 +87,12 @@ type ModRuntimeError struct{}
 
 func (mod ModRuntimeError) Apply(pop *Population) error { return errors.New("") }
 func (mod ModRuntimeError) Validate() error             { return nil }
+
+// Runtime error speciator
+
+type SpecRuntimeError struct{}
+
+func (spec SpecRuntimeError) Apply(indis Individuals, rng *rand.Rand) ([]Individuals, error) {
+	return []Individuals{indis, indis}, errors.New("")
+}
+func (spec SpecRuntimeError) Validate() error { return nil }