data-structures/test_priority_queue.py at master · Jakeand3rson/data-structures · GitHub

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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
from priority_queue import Queue
import pytest


def test_enqueue_first_item():
    queue = Queue()
    queue.insert("Bacon", 1)
    assert queue.last_item.val == "Bacon"


def test_enqueue_multi_last_item():
    queue = Queue()
    queue.insert("Bacon", 1)
    queue.insert("Steak", 2)
    queue.insert("Beer", 1)
    assert queue.last_item.val == "Beer"


def test_insert_with_spaces():
    queue = Queue()
    queue.insert("Breakfast Foods", 32)
    queue.insert("Water Bottles", 3)
    queue.insert("Non awkward backrubs", 1)
    assert queue.size() == 3


def test_dequeue_empty():
    queue = Queue()
    with pytest.raises(ValueError):
        queue.pop()


def test_dequeue():
    queue = Queue()
    queue.insert("Bacon", 1)
    assert queue.pop() == "Bacon"
    assert queue.size() == 0


def test_dequeue_multi():
    queue = Queue()
    queue.insert("Bacon", 1)
    queue.insert("Beer", 5)
    assert queue.pop() == "Bacon"
    assert queue.first_item.val == "Beer"
    assert queue.size() == 1


def test_size():
    queue = Queue()
    queue.insert("Bacon", 65)
    queue.insert("Beer", 2)
    assert queue.size() == 2


def test_size_with_remove():
    queue = Queue()
    queue.insert("Bacon", 1)
    queue.insert("Beer", 7)
    queue.insert("Cow", 5)
    queue.insert("Whiskey", 2)
    queue.pop()
    assert queue.size() == 3


def test_pop_multi():
    queue = Queue()
    queue.insert("Pickles", 3)
    queue.insert("Apples", 10)
    queue.insert("lemons", 14)
    queue.insert("human", 1)
    queue.insert("pineapple", 7)
    assert queue.pop() == "human"
    assert queue.pop() == "Pickles"
    assert queue.pop() == "pineapple"
    assert queue.size() == 2


def test_peek():
    queue = Queue()
    queue.insert("Bacon", 25)
    queue.insert("Eggs", 63)
    queue.insert("Fajitas", 67)
    queue.insert("Chicken", 5)
    assert queue.peek() == "Chicken"


def test_peek_2():
    queue = Queue()
    queue.insert("Cabbage", 5)
    queue.insert("mayo", 88)
    queue.insert("Slaw", 4)
    queue.insert("Village People", 7)
    queue.insert("Coffee", 1)
    assert queue.peek() == "Coffee"
    queue.pop()
    assert queue.peek() == ("Slaw")