-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_fizzbuzztree.py
52 lines (42 loc) · 1.27 KB
/
test_fizzbuzztree.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
from .fizzbuzztree import fizzbuzztree
def test_empty_bst_fizzbuzztree_in_order_traverse(new_bst):
new_bst = fizzbuzztree(new_bst)
lst = []
new_bst.in_order(lst.append)
assert lst == []
def test_data_bst_fizzbuzztree_in_order_traverse(filled_bst):
filled_bst = fizzbuzztree(filled_bst)
lst = []
filled_bst.in_order(lst.append)
assert lst == ["1", "2", "fizz", "4", "fizz", "8", "fizz", "fizz"]
def test_data_bst_fizzbuzztree_pre_order_traverse(filled_bst):
filled_bst = fizzbuzztree(filled_bst)
lst = []
filled_bst.pre_order(lst.append)
assert lst == ["4", "fizz", "2", "1", "8", "fizz", "fizz", "fizz"]
def test_fizz_bst_fizzbuzztree_in_order_traverse(fizz_buzz_bst):
fizz_buzz_bst = fizzbuzztree(fizz_buzz_bst)
lst = []
fizz_buzz_bst.in_order(lst.append)
assert lst == [
"fizzbuzz",
"fizz",
"buzz",
"7",
"fizzbuzz",
"fizzbuzz",
"fizzbuzz",
]
def test_fizz_bst_fizzbuzztree_pre_order_traverse(fizz_buzz_bst):
fizz_buzz_bst = fizzbuzztree(fizz_buzz_bst)
lst = []
fizz_buzz_bst.pre_order(lst.append)
assert lst == [
"fizzbuzz",
"fizz",
"fizzbuzz",
"buzz",
"7",
"fizzbuzz",
"fizzbuzz",
]