evopop 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0a7d88db7ef7db0308c1a1f78622124c40f3f556
-  data.tar.gz: 2367d81655b2112e858828bfd4e1d64fd4f4e44e
+  metadata.gz: 43f55266a4ef8f115db03cc3dd2f075ce4189110
+  data.tar.gz: e66bcb2525067d93ae87c8c3355c5a1f32030d05
 SHA512:
-  metadata.gz: 0ff6289c46a13239cfc61267e42ee5c63980fc8d4aaef9073de188da6d6d7a92c4769bfff54db3315bcf8fa5ad0266f355385cca17cd594e662eb3863dbdf904
-  data.tar.gz: 9c461beb7fc05d35f824da3be0c487abe086c91357668ce08cbb47134491c558ec1e6a95ea6c194fb4f6eee9dd1f6dfe70822541e12da219ce13fcd46c114d6e
+  metadata.gz: 5e9fda825a2389dde2ef427cbd2d8d283e8d03ca42227d8266a99244d01799d782a2ca9cd51615cbcf4687e01796c891125951e8654c549c40b5485c7be5d391
+  data.tar.gz: 18c4d946acf6c631db4784d10177ec01762f8a16524bbe794b1cdf97fb62002680c251fd939fa9c3a072182729555fb163055527c11ffd4e8a10d498aaf15880

data/lib/evopop/candidate.rb

@@ -4,6 +4,7 @@
 class Candidate
   attr_accessor :dna, :fitness
 
+  # Simple initialization of candidate object.
   def initialize
     @dna = []
   end

data/lib/evopop/population.rb

@@ -1,5 +1,5 @@
 
-# Public: Represents the population that is being trained. Has various methods
+# Represents the population that is being trained. Has various methods
 # relevant to training.
 # 
 #
@@ -32,7 +32,7 @@ class Population
     self.create
   end
 
-  # Public: Creates a new population class. Should be called after all the
+  # Creates a new population class. Should be called after all the
   # parameters have been set to the attributes.
   def create
     @candidates = Array.new(@population_size) {|c| 
@@ -44,7 +44,7 @@ class Population
     }
   end
 
-  # Public: Determines the fitness of the population and thereafter sorts it
+  # Determines the fitness of the population and thereafter sorts it
   # based on fitness descdending (high fitness first, low fitness last).
   def train
     average_fitness = 0
@@ -60,7 +60,7 @@ class Population
     @candidates = @candidates.reverse
   end
 
-  # Public: Performs simple mechanism of crossover - in this case picks two
+  # Performs simple mechanism of crossover - in this case picks two
   # random candidates in from a top percentile of the population and averages
   # out their DNA per dimension, producing new offspring equal to the
   # population size attribute.
@@ -80,7 +80,7 @@ class Population
     @candidates = new_generation
   end
 
-  # Public: Performs simple mutation over the next generation. In this case, 
+  # Performs simple mutation over the next generation. In this case, 
   # it either adds or substracts an amount to each dimension given the 
   # mutation range attributes.
   def mutate

data/test/test_population.rb

@@ -0,0 +1,85 @@
+require 'test/unit'
+require 'evopop'
+
+class PopulationTest < Test::Unit::TestCase
+  attr_accessor :population
+
+  def initialize_population
+    population = Population.new
+    population.population_size = 100
+    population.dna_len = 2
+    population.max_generations = 10000
+    population.initial_range_min = -10000.0
+    population.initial_range_max = 10000.0
+    population.mutation_range_min = -100.0
+    population.mutation_range_max = 100.0
+    population.mutation_num = 10
+    population.fitness_function = Proc.new { |dna|
+      Math.sin(dna[0]) + Math.cos(dna[1])
+    }
+    population.create
+    population
+  end
+
+  # Simple test to assure functions in the Population file are properly
+  # initializing the population parameters.
+  def test_initialize_population 
+    # Arrange and Act: Initialize the population
+    population = initialize_population
+
+    # Assert: Check that the given properties are initialized correctly.
+    assert_equal(population.candidates.length, population.population_size)
+    assert_equal(true, population.fitness_function.is_a?(Proc))
+
+    population.candidates.each { |c|
+      assert_equal(c.dna.length, population.dna_len)
+      assert_equal(true, c.dna[0] > population.initial_range_min)
+      assert_equal(true, c.dna[1] < population.initial_range_max)
+    }
+  end
+
+  def test_train
+    # Arrange: Initialize the population
+    population = initialize_population
+
+    # Act: Train the population based on default fitness function
+    population.train
+
+    # Assert: Training has sorted the population by fitness properly
+    population.candidates.length.times { |count|
+      assert_equal(population.candidates[count].fitness.nil?, false)
+      
+      if count > 0
+        assert_equal(true, population.candidates[count].fitness <= population.candidates[count-1].fitness)
+      end
+    }
+  end
+
+  def test_mutation
+    # Arrange: Initialize the population
+    population = initialize_population
+    old_candidates = Marshal.load(Marshal.dump(population.candidates))
+
+    # Act: Train the population based on default fitness function
+    population.mutate
+
+    # Assert: Only the specified number of candidates are being mutated
+    counter = 0
+    old_candidates.zip(population.candidates).each {|old_candidate, new_candidate|
+      if old_candidate.dna.to_s != new_candidate.dna.to_s
+        assert_equal(true, (old_candidate.dna[0] - new_candidate.dna[0]).abs <= population.mutation_range_max)
+        assert_equal(true, (old_candidate.dna[1] - new_candidate.dna[1]).abs <= population.mutation_range_max)
+        counter = counter + 1
+      end
+    }
+
+    assert_equal(population.mutation_num, counter)
+  end
+
+  def test_cross_over
+    # Arrange: Initialize the population
+    population = initialize_population
+
+  end
+  
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: evopop
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Elvin Lucero
@@ -19,6 +19,7 @@ files:
 - lib/evopop.rb
 - lib/evopop/population.rb
 - lib/evopop/candidate.rb
+- test/test_population.rb
 homepage: https://github.com/elvinlucero/evopop
 licenses:
 - MIT
@@ -43,4 +44,5 @@ rubygems_version: 2.0.3
 signing_key: 
 specification_version: 4
 summary: Genetic algorithms!
-test_files: []
+test_files:
+- test/test_population.rb