def test_multipush(self): i = [1, 2, 3, 4, 5] g = PushBack.PushToBack(i.__iter__()) g.pushback(6) assert g.next() == 1 g.pushback(7) assert g.next() == 2 l = list(g) assert l == [3, 4, 5, 6, 7] g.pushback(8) l = list(g) assert l == [8] l = list(g) assert l == []
def test_simplepush(self): i = [1, 2, 3, 4, 5] g = PushBack.PushToBack(i.__iter__()) g.pushback(6) l = list(g) assert l == [1, 2, 3, 4, 5, 6]
def test_wrap(self): i = [1, 2, 3, 4, 5] g = PushBack.PushToBack(i.__iter__()) l = list(g) assert i == l