add tests

This commit is contained in:
2022-05-03 13:50:10 +02:00
parent 5eddad02ee
commit 4b88165c31
2 changed files with 11 additions and 1 deletions
+1 -1
View File
@@ -1,7 +1,7 @@
// use crate::prelude::*;
// #[derive(Copy, Clone, PartialEq)]
#[derive(PartialEq)]
#[derive(PartialEq, Debug)]
pub enum State {
Normal,
Infected,
+10
View File
@@ -21,6 +21,16 @@ pub fn human_idx(x: i32, y: i32, width: i32) -> usize {
((y * width) + x)as usize
}
#[test]
fn popuplation_gen() {
let disease = Disease::new(20,10,5,String::from("Covid 44"));
let population = Population::new(20,10,5,5,7,disease);
assert_eq!(population.humans.len(), 5 * 7);
for human in population.humans.iter() {
assert_eq!(human.present_state, State::Normal);
}
}
impl Population{
pub fn new(start_infected_ratio:u32,start_immune_ratio:u32,start_dead_ratio:u32,width:i32,height:i32,plague:Disease)->Self{
let mut the_humans: Vec<Human> = Vec::with_capacity((width*height) as usize);