def test_Simulation(self):
print("--- Simulation ---")
ls = self.test_IMM()
simu = MI.Simu(ls, "test_simu.txt", 50)
cs = MI.GeneralCascade()
cs.Build(self.gf)
simu.toSimulate(cs)
def test_graph_general_cascade(self):
print("--- General Cascade ---")
fact = MI.GraphFactory()
fin = MI.PyHelper().GetInputFileStream(TEST_GRAPH)
gf = fact.Build(fin)
gc = MI.GeneralCascade()
gc.Build(gf)
def test_Greedy(self):
print("--- Greedy ---")
alg = MI.Greedy()
cascade = MI.GeneralCascade()
cascade.Build(self.gf)
alg.Build(self.gf, 50, cascade)
ls = alg.GetSeedList()
print [ls[i] for i in xrange(len(ls))]
def setUp(self):
fact = MI.GraphFactory()
fin = MI.PyHelper().GetInputFileStream(TEST_GRAPH)
self.gf = fact.Build(fin)
self.gc = MI.GeneralCascade()
self.gc.Build(self.gf)