/
test_priorityq.py
65 lines (47 loc) · 1.39 KB
/
test_priorityq.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#-*- coding:UTF-8 -*-
from __future__ import unicode_literals
from priorityq import PriorityQ
import pytest
@pytest.fixture
def full_priorityq():
priorityq = PriorityQ()
priorityq.insert(3, 8)
priorityq.insert(5, 2)
priorityq.insert(6, 3)
priorityq.insert(4, 10)
priorityq.insert(7, 3)
return priorityq
def test_init_empty():
priorityq = PriorityQ()
assert priorityq.list == []
def test_insert():
priorityq = PriorityQ()
priorityq.insert(3, 8)
assert priorityq.list[0].value == 3
def test_insert_more():
priorityq = PriorityQ()
priorityq.insert(3, 8)
priorityq.insert(5, 2)
assert priorityq.list[0].value == 5
assert priorityq.list[1].value == 3
def test_pop_first(full_priorityq):
assert full_priorityq.pop() == 5
def test_pop_more(full_priorityq):
assert full_priorityq.pop() == 5
assert full_priorityq.pop() == 6
assert full_priorityq.pop() == 7
def test_pop_last():
priorityq = PriorityQ()
priorityq.insert(5, 2)
assert priorityq.pop() == 5
assert priorityq.list == []
def pop_empty():
priorityq = PriorityQ()
with pytest.raises(IndexError):
priorityq.pop()
def test_burn_the_whole_thing(full_priorityq):
assert full_priorityq.pop() == 5
assert full_priorityq.pop() == 6
assert full_priorityq.pop() == 7
assert full_priorityq.pop() == 3
assert full_priorityq.pop() == 4