def test_visualize_prevalence(self):
g = nx.erdos_renyi_graph(1000, 0.1)
model = epd.SIRModel(g)
config = mc.Configuration()
config.add_model_parameter('beta', 0.001)
config.add_model_parameter('gamma', 0.01)
config.add_model_parameter("percentage_infected", 0.05)
model.set_initial_status(config)
iterations = model.iteration_bunch(200)
trends = model.build_trends(iterations)
# Visualization
viz = DiffusionPrevalence(model, trends)
viz.plot("prev.pdf")
os.remove("prev.pdf")
def test_visualize_dynamic(self):
dg = dn.DynGraph()
for t in past.builtins.xrange(0, 4):
g = nx.erdos_renyi_graph(200, 0.05)
dg.add_interactions_from(g.edges(), t)
model = dyn.DynSIModel(dg)
config = mc.Configuration()
config.add_model_parameter('beta', 0.1)
config.add_model_parameter("percentage_infected", 0.1)
model.set_initial_status(config)
iterations = model.execute_snapshots()
trends = model.build_trends(iterations)
# Visualization
viz = DiffusionPrevalence(model, trends)
viz.plot("prevd.pdf")
os.remove("prevd.pdf")