#[cfg(test)]
use crate::support::*;
use genetic_algorithm::fitness::placeholders::CountTrue;
use genetic_algorithm::fitness::{Fitness, FitnessOrdering};
use genetic_algorithm::genotype::{BinaryGenotype, Genotype};
use genetic_algorithm::population::Population;
use genetic_algorithm::select::{Select, SelectElite};
use genetic_algorithm::strategy::evolve::{EvolveConfig, EvolveState};
use genetic_algorithm::strategy::StrategyReporterNoop;
#[test]
fn maximize() {
let genotype = BinaryGenotype::builder()
.with_genes_size(3)
.build()
.unwrap();
let population: Population<bool> = build::population(vec![
vec![false, false, false],
vec![false, false, true],
vec![false, true, false],
vec![false, true, true],
vec![true, false, false],
vec![true, false, true],
vec![true, true, false],
vec![true, true, true],
]);
assert_eq!(population.chromosomes.capacity(), 8);
let mut state = EvolveState::new(&genotype);
state.population = population;
let mut reporter = StrategyReporterNoop::<BinaryGenotype>::new();
let mut rng = SmallRng::seed_from_u64(0);
CountTrue.call_for_population(&mut state.population, &genotype, None, None);
let config = EvolveConfig {
fitness_ordering: FitnessOrdering::Maximize,
target_population_size: 6,
..Default::default()
};
SelectElite::new(0.5, 0.02).call(&genotype, &mut state, &config, &mut reporter, &mut rng);
assert_eq!(
inspect::population(&state.population),
vec![
vec![true, true, true],
vec![false, true, true],
vec![true, false, true],
vec![true, true, false],
vec![false, false, true],
vec![false, true, false]
]
);
assert_eq!(state.population.chromosomes.capacity(), 8);
}
#[test]
fn minimize() {
let genotype = BinaryGenotype::builder()
.with_genes_size(3)
.build()
.unwrap();
let population: Population<bool> = build::population(vec![
vec![false, false, false],
vec![false, false, true],
vec![false, true, false],
vec![false, true, true],
vec![true, false, false],
vec![true, false, true],
vec![true, true, false],
vec![true, true, true],
]);
let mut state = EvolveState::new(&genotype);
state.population = population;
let mut reporter = StrategyReporterNoop::<BinaryGenotype>::new();
let mut rng = SmallRng::seed_from_u64(0);
CountTrue.call_for_population(&mut state.population, &genotype, None, None);
let config = EvolveConfig {
fitness_ordering: FitnessOrdering::Minimize,
target_population_size: 6,
..Default::default()
};
SelectElite::new(0.5, 0.02).call(&genotype, &mut state, &config, &mut reporter, &mut rng);
assert_eq!(
inspect::population(&state.population),
vec![
vec![false, false, false],
vec![false, false, true],
vec![false, true, false],
vec![true, false, false],
vec![false, true, true],
vec![true, false, true]
]
);
}
#[test]
fn fitness_ordering_with_none_fitness() {
let genotype = BinaryGenotype::builder()
.with_genes_size(3)
.build()
.unwrap();
let population: Population<bool> = build::population_with_fitness_scores(vec![
(vec![false, false, false], Some(0)),
(vec![false, false, true], Some(1)),
(vec![false, true, true], Some(2)),
(vec![true, true, true], Some(3)),
(vec![true, true, false], None),
]);
let mut state = EvolveState::new(&genotype);
state.population = population;
let mut reporter = StrategyReporterNoop::<BinaryGenotype>::new();
let mut rng = SmallRng::seed_from_u64(0);
let config = EvolveConfig {
fitness_ordering: FitnessOrdering::Maximize,
target_population_size: 5,
..Default::default()
};
SelectElite::new(0.5, 0.02).call(&genotype, &mut state, &config, &mut reporter, &mut rng);
assert_eq!(
inspect::population_with_fitness_scores(&state.population),
vec![
(vec![true, true, true], Some(3)),
(vec![false, true, true], Some(2)),
(vec![false, false, true], Some(1)),
(vec![false, false, false], Some(0)),
(vec![true, true, false], None),
]
);
let config = EvolveConfig {
fitness_ordering: FitnessOrdering::Minimize,
target_population_size: 5,
..Default::default()
};
SelectElite::new(0.5, 0.02).call(&genotype, &mut state, &config, &mut reporter, &mut rng);
assert_eq!(
inspect::population_with_fitness_scores(&state.population),
vec![
(vec![false, false, false], Some(0)),
(vec![false, false, true], Some(1)),
(vec![false, true, true], Some(2)),
(vec![true, true, true], Some(3)),
(vec![true, true, false], None),
]
);
}
#[test]
fn extreme_elitism_rates() {
let genotype = BinaryGenotype::builder()
.with_genes_size(3)
.build()
.unwrap();
let population: Population<bool> = build::population_with_fitness_scores(vec![
(vec![false, false, false], Some(0)),
(vec![false, false, true], Some(1)),
(vec![false, true, true], Some(2)),
(vec![true, true, true], Some(3)),
(vec![true, true, false], None),
]);
let mut state = EvolveState::new(&genotype);
state.population = population;
let mut reporter = StrategyReporterNoop::<BinaryGenotype>::new();
let mut rng = SmallRng::seed_from_u64(0);
let config = EvolveConfig {
fitness_ordering: FitnessOrdering::Maximize,
target_population_size: 5,
..Default::default()
};
SelectElite::new(0.5, 0.0).call(&genotype, &mut state, &config, &mut reporter, &mut rng);
assert_eq!(
inspect::population_with_fitness_scores(&state.population),
vec![
(vec![true, true, true], Some(3)),
(vec![false, true, true], Some(2)),
(vec![false, false, true], Some(1)),
(vec![false, false, false], Some(0)),
(vec![true, true, false], None),
]
);
let config = EvolveConfig {
fitness_ordering: FitnessOrdering::Minimize,
target_population_size: 5,
..Default::default()
};
SelectElite::new(0.5, 1.0).call(&genotype, &mut state, &config, &mut reporter, &mut rng);
assert_eq!(
inspect::population_with_fitness_scores(&state.population),
vec![
(vec![false, false, false], Some(0)),
(vec![false, false, true], Some(1)),
(vec![false, true, true], Some(2)),
(vec![true, true, true], Some(3)),
(vec![true, true, false], None),
]
);
}