#[cfg(test)]
use crate::support::*;
use genetic_algorithm::chromosome::ChromosomeManager;
use genetic_algorithm::genotype::{
    EvolveGenotype, Genotype, HillClimbGenotype, PermutateGenotype, UniqueGenotype,
};

#[test]
fn mutate_chromosome_single() {
    let mut rng = SmallRng::seed_from_u64(0);
    let mut genotype = UniqueGenotype::builder()
        .with_allele_list(vec![5, 2, 3, 4])
        .build()
        .unwrap();
    genotype.chromosomes_setup();

    let mut chromosome = genotype.chromosome_constructor_random(&mut rng);
    assert_eq!(inspect::chromosome(&chromosome), vec![4, 5, 2, 3]);

    genotype.mutate_chromosome_genes(1, true, &mut chromosome, None, &mut rng);
    assert_eq!(inspect::chromosome(&chromosome), vec![4, 5, 3, 2]);

    genotype.mutate_chromosome_genes(1, true, &mut chromosome, None, &mut rng);
    assert_eq!(inspect::chromosome(&chromosome), vec![2, 5, 3, 4]);
}
#[test]
fn mutate_chromosome_genes_with_duplicates() {
    let mut rng = SmallRng::seed_from_u64(0);
    let mut genotype = UniqueGenotype::builder()
        .with_allele_list(vec![1, 2, 3, 4, 5, 6, 7, 8, 9])
        .build()
        .unwrap();

    let mut chromosome = build::chromosome(vec![1, 2, 3, 4, 5, 6, 7, 8, 9]);
    genotype.mutate_chromosome_genes(3, true, &mut chromosome, None, &mut rng);
    assert_eq!(
        inspect::chromosome(&chromosome),
        vec![1, 2, 3, 5, 9, 6, 7, 8, 4]
    );
}
#[test]
fn mutate_chromosome_genes_without_duplicates() {
    let mut rng = SmallRng::seed_from_u64(0);
    let mut genotype = UniqueGenotype::builder()
        .with_allele_list(vec![1, 2, 3, 4, 5, 6, 7, 8, 9])
        .build()
        .unwrap();

    let mut chromosome = build::chromosome(vec![1, 2, 3, 4, 5, 6, 7, 8, 9]);
    genotype.mutate_chromosome_genes(3, false, &mut chromosome, None, &mut rng);
    assert_eq!(
        inspect::chromosome(&chromosome),
        vec![3, 2, 1, 4, 7, 8, 5, 6, 9]
    );
}

#[test]
#[should_panic]
fn crossover_chromosome_pair_single_gene() {
    let rng = &mut SmallRng::seed_from_u64(0);
    let mut genotype = UniqueGenotype::builder()
        .with_allele_list(vec![5, 2, 3, 4])
        .build()
        .unwrap();

    let mut father = build::chromosome(vec![2, 3, 4, 5]);
    let mut mother = build::chromosome(vec![5, 4, 3, 2]);
    genotype.crossover_chromosome_genes(1, true, &mut father, &mut mother, rng);
}

#[test]
#[should_panic]
fn crossover_chromosome_pair_single_point() {
    let rng = &mut SmallRng::seed_from_u64(0);
    let mut genotype = UniqueGenotype::builder()
        .with_allele_list(vec![5, 2, 3, 4])
        .build()
        .unwrap();

    let mut father = build::chromosome(vec![2, 3, 4, 5]);
    let mut mother = build::chromosome(vec![5, 4, 3, 2]);
    genotype.crossover_chromosome_points(1, true, &mut father, &mut mother, rng);
}

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

    assert_eq!(
        genotype.chromosome_permutations_size(),
        BigUint::from(24u32)
    );
    assert_eq!(
        inspect::chromosomes(
            genotype
                .chromosome_permutations_into_iter()
                .collect::<Vec<_>>()
                .as_slice()
        ),
        vec![
            vec![0, 1, 2, 3],
            vec![0, 1, 3, 2],
            vec![0, 2, 1, 3],
            vec![0, 2, 3, 1],
            vec![0, 3, 1, 2],
            vec![0, 3, 2, 1],
            vec![1, 0, 2, 3],
            vec![1, 0, 3, 2],
            vec![1, 2, 0, 3],
            vec![1, 2, 3, 0],
            vec![1, 3, 0, 2],
            vec![1, 3, 2, 0],
            vec![2, 0, 1, 3],
            vec![2, 0, 3, 1],
            vec![2, 1, 0, 3],
            vec![2, 1, 3, 0],
            vec![2, 3, 0, 1],
            vec![2, 3, 1, 0],
            vec![3, 0, 1, 2],
            vec![3, 0, 2, 1],
            vec![3, 1, 0, 2],
            vec![3, 1, 2, 0],
            vec![3, 2, 0, 1],
            vec![3, 2, 1, 0],
        ]
    );
}

#[test]
fn chromosome_permutations_genes_size_huge() {
    let genotype = UniqueGenotype::builder()
        .with_allele_list((0..30).collect())
        .build()
        .unwrap();
    assert_eq!(
        genotype.chromosome_permutations_size(),
        BigUint::parse_bytes(b"265252859812191058636308480000000", 10).unwrap()
    );

    // ensure lazy
    assert_eq!(
        inspect::chromosomes(
            genotype
                .chromosome_permutations_into_iter()
                .take(1)
                .collect::<Vec<_>>()
                .as_slice()
        ),
        vec![vec![
            0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23,
            24, 25, 26, 27, 28, 29
        ]]
    )
}

#[test]
fn neighbouring_population_2() {
    let mut rng = SmallRng::seed_from_u64(0);
    let mut genotype = UniqueGenotype::builder()
        .with_allele_list(vec![0, 1])
        .build()
        .unwrap();
    genotype.chromosomes_setup();

    let chromosome = genotype.chromosome_constructor_random(&mut rng);
    assert_eq!(inspect::chromosome(&chromosome), vec![0, 1]);

    assert_eq!(genotype.neighbouring_population_size(), BigUint::from(1u32));
    let mut population = Population::new(vec![]);
    genotype.fill_neighbouring_population(&chromosome, &mut population, None, &mut rng);
    assert_eq!(inspect::population(&population), vec![vec![1, 0]]);
}
#[test]
fn neighbouring_population_4() {
    let mut rng = SmallRng::seed_from_u64(0);
    let mut genotype = UniqueGenotype::builder()
        .with_allele_list(vec![0, 1, 2, 3])
        .build()
        .unwrap();
    genotype.chromosomes_setup();

    let chromosome = genotype.chromosome_constructor_random(&mut rng);
    assert_eq!(inspect::chromosome(&chromosome), vec![3, 0, 1, 2]);

    assert_eq!(genotype.neighbouring_population_size(), BigUint::from(6u32));

    let mut population = Population::new(vec![]);
    genotype.fill_neighbouring_population(&chromosome, &mut population, None, &mut rng);
    assert_eq!(
        inspect::population(&population),
        vec![
            vec![0, 3, 1, 2],
            vec![1, 0, 3, 2],
            vec![2, 0, 1, 3],
            vec![3, 1, 0, 2],
            vec![3, 2, 1, 0],
            vec![3, 0, 2, 1],
        ]
    );
}