fix test propagate_infect_cure_all

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.normal, 0);
-        assert_eq!(stats.infected, 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!(propagate_stats, [0, 0, 9, 0]);
-            assert_eq!(stats.normal, 8);
+            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);
         }
     }