mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-03 16:08:02 +00:00
@@ -1,20 +1,20 @@
|
||||
from nose.tools import assert_equal
|
||||
import unittest
|
||||
|
||||
|
||||
class TestQueue(object):
|
||||
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()
|
||||
assert_equal(queue.dequeue(), None)
|
||||
self.assertEqual(queue.dequeue(), None)
|
||||
|
||||
print('Test: Enqueue to an empty queue')
|
||||
queue.enqueue(1)
|
||||
|
||||
print('Test: Dequeue a queue with one element')
|
||||
assert_equal(queue.dequeue(), 1)
|
||||
self.assertEqual(queue.dequeue(), 1)
|
||||
|
||||
print('Test: Enqueue to a non-empty queue')
|
||||
queue.enqueue(2)
|
||||
@@ -22,9 +22,9 @@ class TestQueue(object):
|
||||
queue.enqueue(4)
|
||||
|
||||
print('Test: Dequeue a queue with more than one element')
|
||||
assert_equal(queue.dequeue(), 2)
|
||||
assert_equal(queue.dequeue(), 3)
|
||||
assert_equal(queue.dequeue(), 4)
|
||||
self.assertEqual(queue.dequeue(), 2)
|
||||
self.assertEqual(queue.dequeue(), 3)
|
||||
self.assertEqual(queue.dequeue(), 4)
|
||||
|
||||
print('Success: test_end_to_end')
|
||||
|
||||
@@ -35,4 +35,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user