summaryrefslogtreecommitdiff
path: root/test/GenomeTest.hs
diff options
context:
space:
mode:
Diffstat (limited to 'test/GenomeTest.hs')
-rw-r--r--test/GenomeTest.hs20
1 files changed, 13 insertions, 7 deletions
diff --git a/test/GenomeTest.hs b/test/GenomeTest.hs
index 0a27a94..a09e742 100644
--- a/test/GenomeTest.hs
+++ b/test/GenomeTest.hs
@@ -62,11 +62,17 @@ mutationTests = testGroup "mutations" $
)
([], rand)
(replicate 3 sourceGene)
- in
- mutatedGenes @?=
- [ 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.Internal 1, weight = -7.2413177 }
- ]
-
+ 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 }
+ ]
+ approxEqual a b = abs (a-b) < 0.0001
+ in do
+ (map source mutatedGenes) @?= (map source expected)
+ (map sink mutatedGenes) @?= (map sink expected)
+ True @?= foldl (&&) True (zipWith approxEqual
+ (map weight mutatedGenes)
+ (map weight expected)
+ )
]