#[cfg(test)]
use crate::support::*;
use genetic_algorithm::genotype::{BinaryGenotype, Genotype, ListGenotype, RangeGenotype};
use genetic_algorithm::mutate::{Mutate, MutateSingleGene};
use genetic_algorithm::population::Population;
use genetic_algorithm::strategy::evolve::{EvolveConfig, EvolveState};
use genetic_algorithm::strategy::StrategyReporterNoop;

#[test]
fn binary_genotype() {
    let mut genotype = BinaryGenotype::builder()
        .with_genes_size(3)
        .build()
        .unwrap();

    let population: Population<BinaryChromosome> = build::population(vec![
        vec![true, true, true],
        vec![true, true, true],
        vec![true, true, true],
        vec![true, true, true],
    ]);

    let mut state = EvolveState::new(&genotype);
    state.population = population;
    let config = EvolveConfig::new();
    let mut reporter = StrategyReporterNoop::new();
    let mut rng = SmallRng::seed_from_u64(0);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);

    assert_eq!(
        inspect::population(&state.population),
        vec![
            vec![true, false, true],
            vec![true, true, true],
            vec![true, true, false],
            vec![true, true, true],
        ]
    );
}

#[test]
fn list_genotype() {
    let mut genotype = ListGenotype::builder()
        .with_genes_size(3)
        .with_allele_list(vec![0, 1, 2, 3])
        .build()
        .unwrap();

    let population = build::population(vec![
        vec![0, 0, 0],
        vec![0, 0, 0],
        vec![0, 0, 0],
        vec![0, 0, 0],
    ]);

    let mut state = EvolveState::new(&genotype);
    state.population = population;
    let config = EvolveConfig::new();
    let mut reporter = StrategyReporterNoop::new();
    let mut rng = SmallRng::seed_from_u64(0);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);

    assert_eq!(
        inspect::population(&state.population),
        vec![vec![0, 3, 0], vec![0, 0, 3], vec![0, 0, 0], vec![0, 3, 0],]
    );
}

#[test]
fn range_float_genotype_unscaled() {
    let mut genotype = RangeGenotype::builder()
        .with_genes_size(3)
        .with_allele_range(0.0..=1.0)
        .with_allele_mutation_range(-0.1..=0.1)
        .build()
        .unwrap();

    let population = build::population(vec![
        vec![0.5, 0.5, 0.5],
        vec![0.5, 0.5, 0.5],
        vec![0.5, 0.5, 0.5],
        vec![0.5, 0.5, 0.5],
    ]);

    let mut state = EvolveState::new(&genotype);
    state.population = population;
    let config = EvolveConfig::new();
    let mut reporter = StrategyReporterNoop::new();
    let mut rng = SmallRng::seed_from_u64(0);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);

    assert!(relative_population_eq(
        inspect::population(&state.population),
        vec![
            vec![0.5, 0.595, 0.5],
            vec![0.5, 0.5, 0.588],
            vec![0.5, 0.5, 0.5],
            vec![0.5, 0.563, 0.5],
        ],
        0.001,
    ));

    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);

    assert!(relative_population_eq(
        inspect::population(&state.population),
        vec![
            vec![0.5, 0.595, 0.528],
            vec![0.572, 0.586, 0.533],
            vec![0.557, 0.456, 0.594],
            vec![0.5, 0.563, 0.487],
        ],
        0.001
    ));
}

#[test]
fn range_float_genotype_scaled() {
    let mut genotype = RangeGenotype::builder()
        .with_genes_size(3)
        .with_allele_range(0.0..=1.0)
        .with_allele_mutation_scaled_range(vec![-0.1..=0.1, -0.01..=0.01, -0.001..=0.001])
        .build()
        .unwrap();

    let population = build::population(vec![
        vec![0.5, 0.5, 0.5],
        vec![0.5, 0.5, 0.5],
        vec![0.5, 0.5, 0.5],
        vec![0.5, 0.5, 0.5],
    ]);

    let mut state = EvolveState::new(&genotype);
    state.population = population;
    let config = EvolveConfig::new();
    let mut reporter = StrategyReporterNoop::new();
    let mut rng = SmallRng::seed_from_u64(0);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);
    assert!(relative_population_eq(
        inspect::population(&state.population),
        vec![
            vec![0.5, 0.4, 0.5],
            vec![0.5, 0.5, 0.4],
            vec![0.5, 0.5, 0.5],
            vec![0.5, 0.4, 0.5],
        ],
        0.001,
    ));

    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);
    assert!(relative_population_eq(
        inspect::population(&state.population),
        vec![
            vec![0.5, 0.4, 0.4],
            vec![0.5, 0.5, 0.5],
            vec![0.4, 0.5, 0.5],
            vec![0.5, 0.4, 0.5],
        ],
        0.001,
    ));

    state.current_scale_index = Some(1);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);
    assert!(relative_population_eq(
        inspect::population(&state.population),
        vec![
            vec![0.5, 0.4, 0.4],
            vec![0.49, 0.5, 0.5],
            vec![0.4, 0.5, 0.49],
            vec![0.5, 0.4, 0.51],
        ],
        0.001
    ));
}

#[test]
fn range_integer_genotype() {
    let mut genotype = RangeGenotype::builder()
        .with_genes_size(3)
        .with_allele_range(-9..=9)
        .with_allele_mutation_range(-1..=1)
        .build()
        .unwrap();

    let population = build::population(vec![
        vec![0, 0, 0],
        vec![0, 0, 0],
        vec![0, 0, 0],
        vec![0, 0, 0],
    ]);

    let mut state = EvolveState::new(&genotype);
    state.population = population;
    let config = EvolveConfig::new();
    let mut reporter = StrategyReporterNoop::new();
    let mut rng = SmallRng::seed_from_u64(0);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);

    assert_eq!(
        inspect::population(&state.population),
        vec![vec![0, 1, 0], vec![0, 0, 1], vec![0, 0, 0], vec![0, 1, 0]],
    );

    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);
    MutateSingleGene::new(0.5).call(&mut genotype, &mut state, &config, &mut reporter, &mut rng);

    assert_eq!(
        inspect::population(&state.population),
        vec![vec![0, 1, 0], vec![1, 1, 0], vec![1, -1, 1], vec![0, 1, 0]]
    );
}