Added reset() to smallqueue.

This commit is contained in:
Jeremy Fincher 2003-10-09 03:59:36 +00:00
parent b0f5674f91
commit ae61e178ec
2 changed files with 20 additions and 0 deletions

View File

@ -294,6 +294,12 @@ class smallqueue(list):
def peek(self):
return self[0]
def __repr__(self):
return 'smallqueue([%s])' % ', '.join(map(repr, self))
def reset(self):
self[:] = []
class MaxLengthQueue(queue):
__slots__ = ('length',)

View File

@ -199,6 +199,13 @@ class RingBufferTestCase(unittest.TestCase):
class QueueTest(unittest.TestCase):
def testReset(self):
q = queue()
q.enqueue(1)
self.assertEqual(len(q), 1)
q.reset()
self.assertEqual(len(q), 0)
def testGetitem(self):
q = queue()
n = 10
@ -357,6 +364,13 @@ class QueueTest(unittest.TestCase):
queue = smallqueue
class SmallQueueTest(unittest.TestCase):
def testReset(self):
q = queue()
q.enqueue(1)
self.assertEqual(len(q), 1)
q.reset()
self.assertEqual(len(q), 0)
def testGetitem(self):
q = queue()
n = 10