From a2bbd1d431fa6952d73abc86b986e3f0e1693d39 Mon Sep 17 00:00:00 2001
From: Rene Luria <rene.luria@infomaniak.com>
Date: Tue, 3 May 2022 23:11:23 +0200
Subject: [PATCH] fix test propagate_infect_cure_all

---
 src/population.rs | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/src/population.rs b/src/population.rs
index 24fbbdb..d508b6f 100644
--- a/src/population.rs
+++ b/src/population.rs
@@ -564,26 +564,38 @@ mod tests {
         println!("stats after init: {:?}", stats);
         assert_eq!(stats.normal, 8);
 
-        // kill every one
+        // infect every one
         propagate_stats = population.propagate();
         stats = humans_stats(&population.humans);
         println!("propate_stats: {:?}", propagate_stats);
         println!("population: {:?}", stats);
         assert_eq!(propagate_stats, [8, 1, 0, 0]);
-        assert_eq!(stats.normal, 8);
-        assert_eq!(stats.infected, 0);
+        assert_eq!(stats.normal, 0);
+        assert_eq!(stats.infected, 8);
         assert_eq!(stats.immune, 1);
         assert_eq!(stats.dead, 0);
 
+        // cure every one
+        propagate_stats = population.propagate();
+        stats = humans_stats(&population.humans);
+        println!("propate_stats: {:?}", propagate_stats);
+        println!("population: {:?}", stats);
+        assert_eq!(propagate_stats, [0, 8, 1, 0]);
+        assert_eq!(stats.normal, 0);
+        assert_eq!(stats.infected, 0);
+        assert_eq!(stats.immune, 9);
+        assert_eq!(stats.dead, 0);
+
+        // then
         for _x in 0..100 {
             propagate_stats = population.propagate();
             stats = humans_stats(&population.humans);
             println!("propate_stats: {:?}", propagate_stats);
             println!("population: {:?}", stats);
-            assert_eq!(propagate_stats, [8, 0, 1, 0]);
-            assert_eq!(stats.normal, 8);
+            assert_eq!(propagate_stats, [0, 0, 9, 0]);
+            assert_eq!(stats.normal, 0);
             assert_eq!(stats.infected, 0);
-            assert_eq!(stats.immune, 1);
+            assert_eq!(stats.immune, 9);
             assert_eq!(stats.dead, 0);
         }
     }