Skip to content

Commit a1c84a1

Browse files
committed
Add tests for PriorityItem
1 parent 543cd35 commit a1c84a1

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

tests/test_queue.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from pytest import raises
22

3-
from desmod.queue import PriorityQueue, Queue
3+
from desmod.queue import PriorityItem, PriorityQueue, Queue
44

55

66
def test_mq(env):
@@ -105,16 +105,17 @@ def test_priority_mq(env):
105105
queue = PriorityQueue(env)
106106

107107
def producer(env):
108-
for i in reversed(range(5)):
109-
yield queue.put(i)
108+
for priority in reversed(range(5)):
109+
item = set([priority]) # unhashable
110+
yield queue.put(PriorityItem(priority, item))
110111
yield env.timeout(1)
111112

112113
def consumer(env):
113114
yield env.timeout(5)
114115
for i in range(5):
115116
msg = yield queue.get()
117+
assert msg.item == set([i])
116118
yield env.timeout(1)
117-
assert msg == i
118119

119120
env.process(producer(env))
120121
env.process(consumer(env))

0 commit comments

Comments
 (0)