def test_peek(self):
   """
   Test for peek method
   """
   queue = Queue()
   self.assertEqual(queue.peek(), None)
   queue.enqueue(1)
   queue.enqueue(2)
   queue.enqueue(3)
   self.assertEqual(queue.peek(), 1)
   self.assertEqual(queue.size(), 3)
    def test_peek(self):
        print('Test: test_peek')
        queue = Queue()

        print('Test: Peek to an empty queue')
        self.assertEqual(queue.peek(), None)

        print('Test: General case')
        queue.enqueue(1)
        queue.enqueue(2)
        self.assertEqual(queue.peek(), 1)
        print('Success: test_peek')
Beispiel #3
0
    def _level_order_print(self, start):
        if start is None:
            return

        queue = Queue()
        queue.enqueue(start)

        traversal = ""
        while len(queue) > 0:
            traversal += str(queue.peek().value) + "-"
            node = queue.dequeue()

            if node.left:
                queue.enqueue(node.left)
            if node.right:
                queue.enqueue(node.right)

        return traversal