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)
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)
