def test_voter_model(self):
g = nx.complete_graph(100)
model = opn.VoterModel(g)
config = mc.Configuration()
config.add_model_parameter("percentage_infected", 0.2)
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=False)
self.assertEqual(len(iterations), 10)
def test_voter_model(self):
for g in get_graph():
model = opn.VoterModel(g)
config = mc.Configuration()
config.add_model_parameter("fraction_infected", 0.2)
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=False)
self.assertEqual(len(iterations), 10)
SEISModel = ep.SEISModel(g.copy())
SEISModel.set_initial_status(get_seis_params())
SEIS_iterations = SEISModel.iteration_bunch(num_iterations)
SEIS_trends = SEISModel.build_trends(SEIS_iterations)
visualize(SEISModel, SEIS_trends, sub_dir='epidemics')
SEIRModel = ep.SEIRModel(g.copy())
SEIRModel.set_initial_status(get_seir_params())
SEIR_iterations = SEIRModel.iteration_bunch(num_iterations)
SEIR_trends = SEIRModel.build_trends(SEIR_iterations)
visualize(SEIRModel, SEIR_trends, sub_dir='epidemics')
###############################################################
voter_model = op.VoterModel(g.copy())
voter_model.set_initial_status(get_voter_params())
voter_iterations = voter_model.iteration_bunch(num_iterations)
voter_trends = voter_model.build_trends(voter_iterations)
visualize(voter_model, voter_trends, sub_dir='opinions')
QVoter_model = op.QVoterModel(g.copy())
QVoter_model.set_initial_status(get_qvoter_params())
QVoter_iterations = QVoter_model.iteration_bunch(num_iterations)
QVoter_trends = QVoter_model.build_trends(QVoter_iterations)
visualize(QVoter_model, QVoter_trends, sub_dir='opinions')
majority_rule_model = op.MajorityRuleModel(g.copy())
majority_rule_model.set_initial_status(get_majority_rules_params())
majority_rule_iterations = majority_rule_model.iteration_bunch(
num_iterations)
