added test cases for total_infection and limited_infection