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
