#!/usr/bin/env perl6

use v6;

use Test;

use Algorithm::Evolutionary::Simple;

my $length = 32;

my $population-size = 32;
my @initial-population;
for 1..$population-size -> $p {
    @initial-population.push: random-chromosome( $length );
    cmp-ok( @initial-population[$p-1], "==", mutation( @initial-population[$p-1] ), "Mutation works" );
}

my @another-population = initialize( size => $population-size,
				     genome-length => $length );

cmp-ok( @another-population.elems, "==", $population-size );

# Crossover
my @χs = crossover( @initial-population[0], @initial-population[1]);
cmp-ok( @initial-population[$_], &[!eqv], @χs[$_], "$_ chromosome xovered" ) for 0..1;

# Evaluation
my %fitness-of;
my $population = evaluate( population => @initial-population,
			   fitness-of => %fitness-of,
			   evaluator => &max-ones );


my $one-of-them = $population.roll();
ok( %fitness-of{$one-of-them}, "Evaluated to " ~ %fitness-of{$one-of-them});

my $best = $population.sort(*.value).reverse.[0..1].Mix;
my $initial-fitness = $best.values.[0];

my @pool = get-pool-roulette-wheel( $population, $population-size-2);
cmp-ok( @pool.elems, "==", $population-size-2, "Correct number of elements" );

# Reproduce
my @new-population= produce-offspring( @pool );
cmp-ok( @new-population.elems, "==", $population-size-2, "Correct number of elements in reproduction" );

$population =  (evaluate( population => @new-population,
			     fitness-of => %fitness-of,
			     evaluator => &max-ones ) ∪ $best).Mix;

cmp-ok( $population.elems, "<=", $population-size, "Correct number of elements in new generation" );

my $now-fitness = $population.sort(*.value).reverse.[0].value;

cmp-ok( $now-fitness, ">=", $initial-fitness, "Improving fitness " );

$population = generation( population => $population,
			  fitness-of => %fitness-of,
			  evaluator => &max-ones,
			  population-size => $population-size);

my $evolved-fitness = $population.sort(*.value).reverse.[0].value;

cmp-ok( $evolved-fitness, ">=", $now-fitness, "Improving fitness by evolving " );



# Merge populations
my $another-population =  evaluate( population => @another-population,
				    fitness-of => %fitness-of,
				    evaluator => &max-ones,
                                    auto-t => True);

$population =  generation( population => $another-population,
			   fitness-of => %fitness-of,
			   evaluator => &max-ones,
			   population-size => $population-size,
                           auto-t => True);
$evolved-fitness = $population.sort(*.value).reverse.[0].value;

my $merged = mix( $population, $another-population, $population-size);
cmp-ok( best-fitness($merged), ">=", $evolved-fitness, "Improving fitness by mixing " );

# Merge populations

$another-population = $population;
$evolved-fitness = $population.sort(*.value).reverse.[0].value;

$population =  generation( population => $another-population,
			   fitness-of => %fitness-of,
			   evaluator => &max-ones,
			   population-size => $population-size,
                           :no-mutation);

$merged = mix( $population, $another-population, $population-size);
cmp-ok( best-fitness($merged), ">=", $evolved-fitness, "Improving fitness by mixing " );


done-testing;