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