def test_generalisedthreshold_model(self): g = nx.erdos_renyi_graph(1000, 0.1) model = epd.GeneralisedThresholdModel(g) config = mc.Configuration() config.add_model_parameter('fraction_infected', 0.1) config.add_model_parameter('tau', 5) config.add_model_parameter('mu', 5) threshold = 0.2 for i in g.nodes(): config.add_node_configuration("threshold", i, threshold) model.set_initial_status(config) iterations = model.iteration_bunch(50) self.assertEqual(len(iterations), 50) iterations = model.iteration_bunch(50, node_status=False) self.assertEqual(len(iterations), 50)
def test_generalisedthreshold_model(self): for g in get_graph(True): model = epd.GeneralisedThresholdModel(g) config = mc.Configuration() config.add_model_parameter('fraction_infected', 0.1) config.add_model_parameter('tau', 5) config.add_model_parameter('mu', 5) threshold = 0.2 if isinstance(g, nx.Graph): nodes = g.nodes else: nodes = g.vs['name'] for i in nodes: config.add_node_configuration("threshold", i, threshold) model.set_initial_status(config) iterations = model.iteration_bunch(50) self.assertEqual(len(iterations), 50) iterations = model.iteration_bunch(50, node_status=False) self.assertEqual(len(iterations), 50)