Added test for MaxLengthQueue.

This commit is contained in:
Jeremy Fincher 2003-04-09 18:41:41 +00:00
parent 5da85d4f2f
commit 63279606d9

View File

@ -77,6 +77,10 @@ class QueueTest(unittest.TestCase):
self.assertEqual(1, len(q), 'queue len not 1 after dequeue') self.assertEqual(1, len(q), 'queue len not 1 after dequeue')
q.dequeue() q.dequeue()
self.assertEqual(0, len(q), 'queue len not 0 after dequeue') self.assertEqual(0, len(q), 'queue len not 0 after dequeue')
for i in range(10):
L = range(i)
q = queue(L)
self.assertEqual(len(q), i)
def testEq(self): def testEq(self):
q1 = queue() q1 = queue()
@ -184,6 +188,22 @@ class QueueTest(unittest.TestCase):
self.assertEqual(q, pickle.loads(pickle.dumps(q))) self.assertEqual(q, pickle.loads(pickle.dumps(q)))
class MaxLengthQueueTestCase(unittest.TestCase):
def testMaxLength(self):
q = MaxLengthQueue(3)
q.enqueue(1)
self.assertEqual(len(q), 1)
q.enqueue(2)
self.assertEqual(len(q), 2)
q.enqueue(3)
self.assertEqual(len(q), 3)
q.enqueue(4)
self.assertEqual(len(q), 3)
self.assertEqual(q.peek(), 2)
q.enqueue(5)
self.assertEqual(len(q), 3)
self.assertEqual(q[0], 3)
class FunctionsTest(unittest.TestCase): class FunctionsTest(unittest.TestCase):
def testCatch(self): def testCatch(self):
def f(): def f():