Skip to content

Commit

Permalink
update test suite
Browse files Browse the repository at this point in the history
  • Loading branch information
@GiulioRossetti
GiulioRossetti committed Aug 9, 2017
1 parent 4981783 commit 3acd974
Showing 1 changed file with 79 additions and 0 deletions.
79 changes: 79 additions & 0 deletions ndlib/test/test_ndlib.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@ def test_voter_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_sznajd_model(self):
g = nx.complete_graph(100)
Expand All @@ -46,6 +48,8 @@ def test_sznajd_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_majorityrule_model(self):
g = nx.complete_graph(100)
Expand All @@ -56,6 +60,8 @@ def test_majorityrule_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_qvoter_model(self):
g = nx.complete_graph(100)
Expand All @@ -66,6 +72,8 @@ def test_qvoter_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_cognitive_model(self):
g = nx.complete_graph(100)
Expand All @@ -83,6 +91,8 @@ def test_cognitive_model(self):

iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_si_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
Expand All @@ -93,6 +103,8 @@ def test_si_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_sir_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
Expand All @@ -104,6 +116,8 @@ def test_sir_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_seir_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
Expand All @@ -117,6 +131,17 @@ def test_seir_model(self):
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)

g = g.to_directed()
model = seir.SEIRModel(g)
config = mc.Configuration()
config.add_model_parameter('beta', 0.5)
config.add_model_parameter('gamma', 0.8)
config.add_model_parameter('alpha', 0.5)
config.add_model_parameter("percentage_infected", 0.1)
model.set_initial_status(config)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_seis_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
model = seis.SEISModel(g)
Expand All @@ -129,6 +154,17 @@ def test_seis_model(self):
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)

g = g.to_directed()
model = seis.SEISModel(g)
config = mc.Configuration()
config.add_model_parameter('beta', 0.5)
config.add_model_parameter('lambda', 0.8)
config.add_model_parameter('alpha', 0.5)
config.add_model_parameter("percentage_infected", 0.1)
model.set_initial_status(config)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_sis_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
model = sis.SISModel(g)
Expand All @@ -139,6 +175,8 @@ def test_sis_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_kertesz_model(self):
g = nx.complete_graph(100)
Expand All @@ -155,6 +193,8 @@ def test_kertesz_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_multiple_si_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
Expand All @@ -165,6 +205,8 @@ def test_multiple_si_model(self):
model.set_initial_status(config)
executions = ut.multi_runs(model, execution_number=10, iteration_number=50)
self.assertEqual(len(executions), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_threshold_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
Expand All @@ -179,6 +221,8 @@ def test_threshold_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_profile_threshold_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
Expand All @@ -197,6 +241,25 @@ def test_profile_threshold_model(self):
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)

model = pt.ProfileThresholdModel(g)
config = mc.Configuration()
config.add_model_parameter('percentage_infected', 0.1)
config.add_model_parameter("blocked", 0.1)
config.add_model_parameter("adopter_rate", 0.001)

threshold = 0.2
profile = 0.1
for i in g.nodes():
config.add_node_configuration("threshold", i, threshold)
config.add_node_configuration("profile", i, profile)

model.set_initial_status(config)

iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_profile_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
model = pr.ProfileModel(g)
Expand All @@ -211,6 +274,20 @@ def test_profile_model(self):
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)

model = pr.ProfileModel(g)
config = mc.Configuration()
config.add_model_parameter('percentage_infected', 0.1)
config.add_model_parameter("blocked", 0.1)
config.add_model_parameter("adopeter_rate", 0.001)

profile = 0.1
for i in g.nodes():
config.add_node_configuration("profile", i, profile)

model.set_initial_status(config)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_independent_cascade_model(self):
g = nx.erdos_renyi_graph(1000, 0.1)
model = ids.IndependentCascadesModel(g)
Expand All @@ -223,6 +300,8 @@ def test_independent_cascade_model(self):
model.set_initial_status(config)
iterations = model.iteration_bunch(10)
self.assertEqual(len(iterations), 10)
iterations = model.iteration_bunch(10, node_status=True)
self.assertEqual(len(iterations), 10)

def test_kertesz_model_predefined_blocked(self):
g = nx.complete_graph(100)
Expand Down

0 comments on commit 3acd974

Please sign in to comment.