def test_transmit_infections(): v0 = Village(loc=(0.2, 0.3), N=10, vaccinated_fraction=0.5, neighbors=set()) v0.challenge(2) generation = 0 while v0.S > 0 and v0.infecteds: print('\n=========\ngeneration-%d\n=========' % generation) v0.update() v0.transmit() print(v0.summary()) generation += 1
def test_transmit_neighbors(): v0 = Village(loc=(0.2, 0.3), N=10, vaccinated_fraction=0.5) v1 = Village(loc=(0.2, 0.3), N=10, vaccinated_fraction=0.5) v0.neighbors = [v1] # only v0-towards-v1 transmissions between villages v0.challenge(2) generation = 0 while (v0.S + v1.S) > 0 and (v0.infecteds or v1.infecteds): print('\n=========\ngeneration-%d\n=========' % generation) for v in (v0, v1): v.update() for v in (v0, v1): v.transmit() print(v.summary()) print('=========') generation += 1