1from modsim import * 2 3def calc_total_infected(results, system): 4 s_0 = results.s[0] 5 s_end = results.s[system.t_end] 6 return s_0 - s_end 7 8 9