def test_one_poll(self):
qs = QueueSet(['a'])
assert len(qs) == 1
assert qs.front() == 'a'
k = qs.poll()
assert k == 'a'
assert len(qs) == 0
assert qs.front() is None
def test_poll(self):
qs = QueueSet(['one','two','three'])
assert len(qs) == 3
assert qs.front() == 'one'
k = qs.poll()
assert k == 'one'
assert len(qs) == 2
assert qs.front() == 'two'
def test_enqueue_one_thing_twice(self):
qs = QueueSet()
qs.enqueue('home')
qs.enqueue('home')
assert len(qs) == 1
assert qs.front() == 'home'
k = qs.poll()
assert k == 'home'
assert len(qs) == 0
assert qs.front() is None
def test_enqueue_two(self):
qs = QueueSet()
qs.enqueue('a')
qs.enqueue('b')
assert len(qs) == 2
assert qs.front() == 'a'
k = qs.poll()
assert k == 'a'
assert len(qs) == 1
assert qs.front() == 'b'
k = qs.poll()
assert k == 'b'
assert len(qs) == 0
assert qs.front() is None
def test_many_polls(self):
qs = QueueSet()
self.fill_queue(qs, 10000)
for i in xrange(0, 9000):
qs.poll()
assert len(qs) == 1000
assert qs.front() == 9000
def test_enqueue_multiple_replace_one(self):
qs = QueueSet()
qs.enqueue('a')
qs.enqueue('b')
qs.enqueue('c')
qs.enqueue('d')
qs.enqueue('b')
assert len(qs) == 4
assert qs.front() == 'a'
qs.poll()
k = qs.poll()
assert k == 'b'
assert len(qs) == 2
def test_enq_multi_replace_multi(self):
qs = QueueSet()
self.fill_queue(qs, 100)
qs.enqueue(23)
qs.enqueue(47)
qs.enqueue(34)
qs.enqueue(194)
assert 23 in qs
assert 47 in qs
assert 34 in qs
assert 194 in qs
assert len(qs) == 101
assert qs.front() == 0
def test_cons3(self):
qs = QueueSet([1,2,3,4,5])
assert len(qs) == 5
assert qs.front() == 1
def test_cons1(self):
qs = QueueSet()
assert len(qs) == 0
assert qs.front() is None
