diff options
Diffstat (limited to 'test/GenomeTest.hs')
| -rw-r--r-- | test/GenomeTest.hs | 94 |
1 files changed, 62 insertions, 32 deletions
diff --git a/test/GenomeTest.hs b/test/GenomeTest.hs index 8448133..d775467 100644 --- a/test/GenomeTest.hs +++ b/test/GenomeTest.hs @@ -2,9 +2,11 @@ module GenomeTest (suite) where import Test.Tasty import Test.Tasty.HUnit -import qualified Mind as M +import Mind (NeuronIndex (..)) import Genome import System.Random +import Data.Ix + suite :: TestTree suite = testGroup "genome tests" $ @@ -18,7 +20,7 @@ mutationTests = testGroup "mutations" $ let rand = mkStdGen 1 -- randomR generates sequence (1, 2, 0, ...) genome = Genome { numInput = 2, numInternal = 1, numOutput = 5, genes = [] } - sourceGene = Gene { source = M.Input 0, sink = M.Output 1, weight = 4 } + sourceGene = Gene { source = Input 0, sink = Output 1, weight = 4 } mutatedGenes = fst $ foldl (\(list, r) g -> let (g', r') = mutateGeneSource genome g r @@ -28,15 +30,15 @@ mutationTests = testGroup "mutations" $ (replicate 3 sourceGene) in mutatedGenes @?= - [ Gene { source = M.Input 1, sink = M.Output 1, weight = 4 } - , Gene { source = M.Internal 0, sink = M.Output 1, weight = 4 } - , Gene { source = M.Input 0, sink = M.Output 1, weight = 4 } + [ Gene { source = Input 1, sink = Output 1, weight = 4 } + , Gene { source = Internal 0, sink = Output 1, weight = 4 } + , Gene { source = Input 0, sink = Output 1, weight = 4 } ] , testCase "mutating the sink of a gene" $ let rand = mkStdGen 1 -- randomR generates sequence (1, 2, 0, ...) genome = Genome { numInput = 2, numInternal = 1, numOutput = 2, genes = [] } - sourceGene = Gene { source = M.Input 0, sink = M.Output 1, weight = 4 } + sourceGene = Gene { source = Input 0, sink = Output 1, weight = 4 } mutatedGenes = fst $ foldl (\(list, r) g -> let (g', r') = mutateGeneSink genome g r @@ -46,15 +48,15 @@ mutationTests = testGroup "mutations" $ (replicate 3 sourceGene) in mutatedGenes @?= - [ Gene { source = M.Input 0, sink = M.Output 0, weight = 4 } - , Gene { source = M.Input 0, sink = M.Output 1, weight = 4 } - , Gene { source = M.Input 0, sink = M.Internal 0, weight = 4 } + [ Gene { source = Input 0, sink = Output 0, weight = 4 } + , Gene { source = Input 0, sink = Output 1, weight = 4 } + , Gene { source = Input 0, sink = Internal 0, weight = 4 } ] , testCase "mutating the weight of a gene" $ let rand = mkStdGen 0 -- randomR generates sequence (7.572357,-1.4116564,-7.2413177, ...) genome = Genome { numInput = 2, numInternal = 1, numOutput = 2, genes = [] } - sourceGene = Gene { source = M.Input 0, sink = M.Output 1, weight = 4 } + sourceGene = Gene { source = Input 0, sink = Output 1, weight = 4 } mutatedGenes = fst $ foldl (\(list, r) g -> let (g', r') = mutateGeneWeight genome g r @@ -63,9 +65,9 @@ mutationTests = testGroup "mutations" $ ([], rand) (replicate 3 sourceGene) expected = - [ Gene { source = M.Input 0, sink = M.Output 1, weight = 7.572357} - , Gene { source = M.Input 0, sink = M.Output 1, weight = -1.4116564 } - , Gene { source = M.Input 0, sink = M.Output 1, weight = -7.2413177 } + [ Gene { source = Input 0, sink = Output 1, weight = 7.572357} + , Gene { source = Input 0, sink = Output 1, weight = -1.4116564 } + , Gene { source = Input 0, sink = Output 1, weight = -7.2413177 } ] approxEqual a b = abs (a-b) < 0.0001 in do @@ -82,9 +84,9 @@ mutationTests = testGroup "mutations" $ , numInternal = 2 , numOutput = 1 , genes = - [ Gene { source = M.Input 0, sink = M.Internal 0, weight = 1.0 } - , Gene { source = M.Internal 0, sink = M.Internal 1, weight = 1.0 } - , Gene { source = M.Internal 1, sink = M.Output 0, weight = 1.0 } + [ Gene { source = Input 0, sink = Internal 0, weight = 1.0 } + , Gene { source = Internal 0, sink = Internal 1, weight = 1.0 } + , Gene { source = Internal 1, sink = Output 0, weight = 1.0 } ] } r = mkStdGen 5 @@ -103,10 +105,10 @@ mutationTests = testGroup "mutations" $ , numInternal = 3 , numOutput = 1 , genes = - [ Gene { source = M.Input 0, sink = M.Internal 0, weight = 1.0 } - , Gene { source = M.Internal 0, sink = M.Internal 1, weight = 1.0 } - , Gene { source = M.Internal 1, sink = M.Internal 2, weight = 1.0 } - , Gene { source = M.Internal 2, sink = M.Output 0, weight = 1.0 } + [ Gene { source = Input 0, sink = Internal 0, weight = 1.0 } + , Gene { source = Internal 0, sink = Internal 1, weight = 1.0 } + , Gene { source = Internal 1, sink = Internal 2, weight = 1.0 } + , Gene { source = Internal 2, sink = Output 0, weight = 1.0 } ] } (genome', r') = mutateGenomeRemoveInternal genome r @@ -126,23 +128,51 @@ mutationTests = testGroup "mutations" $ (numOutput genome''') @?= (numOutput genome) (genes genome') @?= - [ Gene { source = M.Input 0, sink = M.Internal 0, weight = 1.0 } - , Gene { source = M.Internal 0, sink = M.Internal 1, weight = 1.0 } - , Gene { source = M.Internal 0, sink = M.Internal 1, weight = 1.0 } - , Gene { source = M.Internal 1, sink = M.Output 0, weight = 1.0 } + [ Gene { source = Input 0, sink = Internal 0, weight = 1.0 } + , Gene { source = Internal 0, sink = Internal 1, weight = 1.0 } + , Gene { source = Internal 0, sink = Internal 1, weight = 1.0 } + , Gene { source = Internal 1, sink = Output 0, weight = 1.0 } ] (genes genome'') @?= - [ Gene { source = M.Input 0, sink = M.Internal 0, weight = 1.0 } - , Gene { source = M.Internal 0, sink = M.Internal 1, weight = 1.0 } - , Gene { source = M.Internal 1, sink = M.Output 0, weight = 1.0 } - , Gene { source = M.Internal 1, sink = M.Output 0, weight = 1.0 } + [ Gene { source = Input 0, sink = Internal 0, weight = 1.0 } + , Gene { source = Internal 0, sink = Internal 1, weight = 1.0 } + , Gene { source = Internal 1, sink = Output 0, weight = 1.0 } + , Gene { source = Internal 1, sink = Output 0, weight = 1.0 } ] (genes genome''') @?= - [ Gene { source = M.Input 0, sink = M.Internal 0, weight = 1.0 } - , Gene { source = M.Input 0, sink = M.Internal 0, weight = 1.0 } - , Gene { source = M.Internal 0, sink = M.Internal 1, weight = 1.0 } - , Gene { source = M.Internal 1, sink = M.Output 0, weight = 1.0 } + [ Gene { source = Input 0, sink = Internal 0, weight = 1.0 } + , Gene { source = Input 0, sink = Internal 0, weight = 1.0 } + , Gene { source = Internal 0, sink = Internal 1, weight = 1.0 } + , Gene { source = Internal 1, sink = Output 0, weight = 1.0 } ] + , testCase "add new gene" $ + let + genome = Genome + { numInput = 1 + , numInternal = 2 + , numOutput = 1 + , genes = [] + } + r = mkStdGen 5 + (genome', _) = mutateGenomeAddGene genome r + new = last $ genes genome' + -- checking sources + validSource (Input x) = inRange (0, numInput genome') x + validSource (Internal x) = inRange (0, numInternal genome') x + validSource (Output _) = False + -- checking sinks + validSink (Input _) = False + validSink (Internal x) = inRange(0, numInternal genome') x + validSink (Output x) = inRange(0, numOutput genome') x + w = weight new + in do + (numInput genome') @?= (numInput genome) + (numInternal genome') @?= (numInternal genome) + (numOutput genome') @?= (numOutput genome) + (length $ genes genome') @?= 1 + (length $ genes genome) + validSource (source new) @?= True + validSink (sink new) @?= True + (w >= -4) && (w <= 4) @?= True ] |