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)