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)