def test_iterators(self):
gs1 = GraphSet([g0, g12, g13])
gs2 = GraphSet()
for g in gs1:
self.assertTrue(isinstance(g, list))
gs2 = gs2 | GraphSet([g])
self.assertEqual(gs1, GraphSet([g0, g12, g13]))
self.assertEqual(gs1, gs2)
gs2 = GraphSet()
for g in gs1:
self.assertTrue(isinstance(g, list))
gs2 = gs2 | GraphSet([g])
self.assertEqual(gs1, gs2)
gs1 = GraphSet([g0, g12, g13])
gs2 = GraphSet()
for g in gs1.rand_iter():
self.assertTrue(isinstance(g, list))
gs2 = gs2 | GraphSet([g])
self.assertEqual(gs1, gs2)
gen = gs1.rand_iter()
self.assertTrue(isinstance(gen.next(), list))
gs = GraphSet([g0, g1, g12, g123, g1234, g134, g14, g4])
r = []
for g in gs.max_iter():
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g14)
self.assertEqual(r[1], g134)
self.assertEqual(r[2], g4)
r = []
for g in gs.max_iter({e1: -.3, e2: .2, e3: .2, e4: -.4}):
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g123)
self.assertEqual(r[1], g0)
self.assertEqual(r[2], g12)
r = []
for g in gs.min_iter():
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g123)
self.assertEqual(r[1], g0)
self.assertEqual(r[2], g12)
r = []
for g in gs.min_iter({e1: -.3, e2: .2, e3: .2, e4: -.4}):
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g14)
self.assertEqual(r[1], g134)
self.assertEqual(r[2], g4)
def test_iterators(self):
gs1 = GraphSet([g0, g12, g13])
gs2 = GraphSet()
for g in gs1:
self.assertTrue(isinstance(g, list))
gs2 = gs2 | GraphSet([g])
self.assertEqual(gs1, GraphSet([g0, g12, g13]))
self.assertEqual(gs1, gs2)
gs2 = GraphSet()
for g in gs1:
self.assertTrue(isinstance(g, list))
gs2 = gs2 | GraphSet([g])
self.assertEqual(gs1, gs2)
gs1 = GraphSet([g0, g12, g13])
gs2 = GraphSet()
for g in gs1.rand_iter():
self.assertTrue(isinstance(g, list))
gs2 = gs2 | GraphSet([g])
self.assertEqual(gs1, gs2)
gen = gs1.rand_iter()
self.assertTrue(isinstance(next(gen), list))
gs = GraphSet([g0, g1, g12, g123, g1234, g134, g14, g4])
r = []
for g in gs.max_iter():
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g14)
self.assertEqual(r[1], g134)
self.assertEqual(r[2], g4)
r = []
for g in gs.max_iter({e1: -.3, e2: .2, e3: .2, e4: -.4}):
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g123)
self.assertEqual(r[1], g0)
self.assertEqual(r[2], g12)
r = []
for g in gs.min_iter():
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g123)
self.assertEqual(r[1], g0)
self.assertEqual(r[2], g12)
r = []
for g in gs.min_iter({e1: -.3, e2: .2, e3: .2, e4: -.4}):
self.assertTrue(isinstance(g, list))
r.append(g)
self.assertEqual(len(r), 8)
self.assertEqual(r[0], g14)
self.assertEqual(r[1], g134)
self.assertEqual(r[2], g4)
universe = []
for i in range(n):
universe.append((i,i+1))
for i in universe:
print i
GraphSet.set_universe(universe)
ng = 20
g_univ = GraphSet({})
g1 = []
i = 1
for sg in g_univ.rand_iter():
g1.append(sg)
if i == ng: break
i += 1
G1 = GraphSet(g1)
print(G1.dumps())
data = map (lambda x:x.split(" "),G1.dumps().strip().split("\n"))
data.pop()
data_fl = list(chain.from_iterable(data))
len_data_fl = len(data_fl)
num_v = len_data_fl/4
