genetic-search

import { describe, it } from '@jest/globals'; import { BaseGenome, Population, PopulationSummary, PopulationSummaryManager } from '../../src'; import { createEmptyGroupedStatSummary, createEmptyRangeStatSummary, createEmptyStatSummary, roundGroupedStatSummary, roundRangeStatSummary, roundStatSummary, } from '../../src/utils'; describe.each([ ...dataProviderForPopulationSummaryManager(), ] as Array<[Population<BaseGenome>, PopulationSummary]>)( 'PopulationSummaryManager Test', (sortedPopulation: Population<BaseGenome>, expected: PopulationSummary) => { it('', async () => { const roundPrecision = 2; const summaryManager = new PopulationSummaryManager(); expect(summaryManager.get()).toEqual({ ageSummary: createEmptyRangeStatSummary(), fitnessSummary: createEmptyStatSummary(), groupedFitnessSummary: createEmptyGroupedStatSummary(), stagnationCounter: 0, }); expect(summaryManager.getRounded(roundPrecision)).toEqual({ ageSummary: createEmptyRangeStatSummary(), fitnessSummary: createEmptyStatSummary(), groupedFitnessSummary: createEmptyGroupedStatSummary(), stagnationCounter: 0, }); summaryManager.update(sortedPopulation); const summary = summaryManager.getRounded(2); expect(roundRangeStatSummary(summary.ageSummary, roundPrecision)).toEqual(expected.ageSummary); expect(roundStatSummary(summary.fitnessSummary, roundPrecision)).toEqual(expected.fitnessSummary); expect(roundGroupedStatSummary(summary.groupedFitnessSummary, roundPrecision)).toEqual(expected.groupedFitnessSummary); expect(summaryManager.getRounded(roundPrecision)).toEqual(summary); const oldStagnationCounter = summaryManager.get().stagnationCounter; summaryManager.update(sortedPopulation); expect(summaryManager.get().stagnationCounter).toEqual(oldStagnationCounter+1); expect(summaryManager.getRounded(4).stagnationCounter).toEqual(oldStagnationCounter+1); }); }, ); function dataProviderForPopulationSummaryManager(): Array<[Population<BaseGenome>, PopulationSummary]> { return [ [ [], { ageSummary: { min: 0, mean: 0, max: 0, }, fitnessSummary: { count: 0, best: 0, second: 0, mean: 0, median: 0, worst: 0, }, groupedFitnessSummary: { initial: { count: 0, best: 0, second: 0, mean: 0, median: 0, worst: 0, }, crossover: { count: 0, best: 0, second: 0, mean: 0, median: 0, worst: 0, }, mutation: { count: 0, best: 0, second: 0, mean: 0, median: 0, worst: 0, }, }, stagnationCounter: 0, }, ], [ [ { id: 1, stats: { age: 1, fitness: 100, phenome: [10], origin: 'initial', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 2, stats: { age: 1, fitness: 90, phenome: [9], origin: 'crossover', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 3, stats: { age: 2, fitness: 80, phenome: [8], origin: 'mutation', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 4, stats: { age: 2, fitness: 70, phenome: [7], origin: 'initial', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 5, stats: { age: 4, fitness: 0, phenome: [6], origin: 'crossover', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, ], { ageSummary: { min: 1, mean: 2, max: 4, }, fitnessSummary: { count: 5, best: 100, second: 90, mean: 68, median: 80, worst: 0, }, groupedFitnessSummary: { initial: { count: 2, best: 100, second: 70, mean: 85, median: 85, worst: 70, }, crossover: { count: 2, best: 90, second: 0, mean: 45, median: 45, worst: 0, }, mutation: { count: 1, best: 80, second: 0, mean: 80, median: 80, worst: 80, }, }, stagnationCounter: 0, }, ], [ [ { id: 1, stats: { age: 2, fitness: 100, phenome: [10], origin: 'initial', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 2, stats: { age: 1, fitness: 90, phenome: [9], origin: 'crossover', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 3, stats: { age: 2, fitness: 80, phenome: [8], origin: 'crossover', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 4, stats: { age: 4, fitness: 70, phenome: [7], origin: 'initial', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, { id: 5, stats: { age: 1, fitness: 0, phenome: [6], origin: 'crossover', originCounters: { crossover: 0, mutation: 0, }, parentIds: [], }, }, ], { ageSummary: { min: 1, mean: 2, max: 4, }, fitnessSummary: { count: 5, best: 100, second: 90, mean: 68, median: 80, worst: 0, }, groupedFitnessSummary: { initial: { count: 2, best: 100, second: 70, mean: 85, median: 85, worst: 70, }, crossover: { count: 3, best: 90, second: 80, mean: 56.67, median: 80, worst: 0, }, mutation: { count: 0, best: 0, second: 0, mean: 0, median: 0, worst: 0, }, }, stagnationCounter: 0, }, ], ]; }