test_ndlib.py 文件源码

def test_multiple_si_model(self):
        g = nx.erdos_renyi_graph(1000, 0.1)
        model = si.SIModel(g)
        config = mc.Configuration()
        config.add_model_parameter('beta', 0.01)
        config.add_model_parameter("percentage_infected", 0.1)
        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=False)
        self.assertEqual(len(iterations), 10)