mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-02 23:48:02 +00:00
39 lines
977 B
Python
39 lines
977 B
Python
import unittest
|
|
|
|
|
|
class TestQueue(unittest.TestCase):
|
|
|
|
# TODO: It would be better if we had unit tests for each
|
|
# method in addition to the following end-to-end test
|
|
def test_end_to_end(self):
|
|
print('Test: Dequeue an empty queue')
|
|
queue = Queue()
|
|
self.assertEqual(queue.dequeue(), None)
|
|
|
|
print('Test: Enqueue to an empty queue')
|
|
queue.enqueue(1)
|
|
|
|
print('Test: Dequeue a queue with one element')
|
|
self.assertEqual(queue.dequeue(), 1)
|
|
|
|
print('Test: Enqueue to a non-empty queue')
|
|
queue.enqueue(2)
|
|
queue.enqueue(3)
|
|
queue.enqueue(4)
|
|
|
|
print('Test: Dequeue a queue with more than one element')
|
|
self.assertEqual(queue.dequeue(), 2)
|
|
self.assertEqual(queue.dequeue(), 3)
|
|
self.assertEqual(queue.dequeue(), 4)
|
|
|
|
print('Success: test_end_to_end')
|
|
|
|
|
|
def main():
|
|
test = TestQueue()
|
|
test.test_end_to_end()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|