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
