def test_peek(self):
i = peek_insert_iter(iter([1, 2, 3]))
self.assertEqual(i.peek(), 1)
self.assertEqual(i.next(), 1)
self.assertEqual(i.peek(), 2)
self.assertEqual(i.next(), 2)
self.assertEqual(i.peek(), 3)
self.assertEqual(i.next(), 3)
self.assertIs(i.peek(), EOF)
def test_insert(self):
i = peek_insert_iter(iter([1, 2, 3]))
i.insert([10])
self.assertEqual(i.next(), 10)
self.assertEqual(i.next(), 1)
i.insert([4, 5])
i.insert([6, 7])
self.assertEqual(i.next(), 6)
self.assertEqual(i.next(), 7)
self.assertEqual(i.next(), 4)
self.assertEqual(i.next(), 5)
self.assertEqual(i.next(), 2)
self.assertEqual(i.next(), 3)
with self.assertRaises(StopIteration):
i.next()
i.insert([8])
self.assertEqual(i.next(), 8)
with self.assertRaises(StopIteration):
i.next()
def test_peek_insert(self):
i = peek_insert_iter(iter([1, 2, 3]))
i.insert([10])
self.assertEqual(i.peek(), 10)
self.assertEqual(i.next(), 10)
self.assertEqual(i.peek(), 1)
self.assertEqual(i.next(), 1)
i.insert([4, 5])
self.assertEqual(i.peek(), 4)
i.insert([6, 7])
self.assertEqual(i.peek(), 6)
self.assertEqual(i.next(), 6)
self.assertEqual(i.next(), 7)
self.assertEqual(i.next(), 4)
self.assertEqual(i.next(), 5)
self.assertEqual(i.next(), 2)
self.assertEqual(i.next(), 3)
self.assertIs(i.peek(), EOF)
i.insert([8])
self.assertEqual(i.peek(), 8)
self.assertEqual(i.next(), 8)
self.assertIs(i.peek(), EOF)
