def test_accumulate_susceptibles(): v1 = Village(loc=(0.2, 0.3), N=100, vaccinated_fraction=0.2, neighbors=set()) v1.S = 0 # start from transiently low susceptibles n_generations = 26 * 300 S_timeseries = [] for t in range(n_generations): v1.update() S_timeseries.append(v1.S) plt.plot(range(n_generations), S_timeseries)