def test_minimal():
# sample fitness function
def eval_fitness(population):
for individual in population:
individual.fitness = 1.0
# creates the population
local_dir = os.path.dirname(__file__)
config = Config(os.path.join(local_dir, 'test_configuration'))
config.save_best = True
pop = Population(config)
# run the simulation for up to 20 generations
pop.run(eval_fitness, 20)
# Test save_checkpoint with defaults
pop.save_checkpoint()
# get statistics
best_unique = pop.statistics.best_unique_genomes(1)
best_unique = pop.statistics.best_unique_genomes(2)
avg_fitness = pop.statistics.get_average_fitness()
assert len(avg_fitness) == 1
assert all(f == 1 for f in avg_fitness)
# Change fitness threshold and do another run.
config.max_fitness_threshold = 1.1
pop = Population(config)
# runs the simulation for 20 generations
pop.run(eval_fitness, 20)
# get statistics
avg_fitness = pop.statistics.get_average_fitness()
assert len(avg_fitness) == 20
assert all(f == 1 for f in avg_fitness)
def test_checkpoint():
# sample fitness function
def eval_fitness(population):
for individual in population:
individual.fitness = 0.99
# creates the population
local_dir = os.path.dirname(__file__)
config = Config(os.path.join(local_dir, 'test_configuration'))
pop = Population(config)
pop.run(eval_fitness, 20)
t = tempfile.NamedTemporaryFile(delete=False)
t.close()
pop.save_checkpoint(t.name)
pop2 = Population(config)
pop2.load_checkpoint(t.name)
def test_checkpoint():
# sample fitness function
def eval_fitness(population):
for individual in population:
individual.fitness = 0.99
# creates the population
local_dir = os.path.dirname(__file__)
config = Config(os.path.join(local_dir, 'test_configuration'))
pop = Population(config)
pop.run(eval_fitness, 20)
t = tempfile.NamedTemporaryFile(delete=False)
t.close()
pop.save_checkpoint(t.name)
pop2 = Population(config)
pop2.load_checkpoint(t.name)
