forked from panda3d/panda3d
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_PythonUtil.py
More file actions
159 lines (128 loc) · 4.19 KB
/
test_PythonUtil.py
File metadata and controls
159 lines (128 loc) · 4.19 KB
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
from direct.showbase import PythonUtil
import pytest
def test_queue():
q = PythonUtil.Queue()
assert q.isEmpty()
q.clear()
assert q.isEmpty()
q.push(10)
assert not q.isEmpty()
q.push(20)
assert not q.isEmpty()
assert len(q) == 2
assert q.front() == 10
assert q.back() == 20
assert q.top() == 10
assert q.top() == 10
assert q.pop() == 10
assert len(q) == 1
assert not q.isEmpty()
assert q.pop() == 20
assert len(q) == 0
assert q.isEmpty()
def test_flywheel():
f = PythonUtil.flywheel(['a','b','c','d'], countList=[11,20,3,4])
obj2count = {}
for obj in f:
obj2count.setdefault(obj, 0)
obj2count[obj] += 1
assert obj2count['a'] == 11
assert obj2count['b'] == 20
assert obj2count['c'] == 3
assert obj2count['d'] == 4
f = PythonUtil.flywheel([1,2,3,4], countFunc=lambda x: x*2)
obj2count = {}
for obj in f:
obj2count.setdefault(obj, 0)
obj2count[obj] += 1
assert obj2count[1] == 2
assert obj2count[2] == 4
assert obj2count[3] == 6
assert obj2count[4] == 8
f = PythonUtil.flywheel([1,2,3,4], countFunc=lambda x: x, scale = 3)
obj2count = {}
for obj in f:
obj2count.setdefault(obj, 0)
obj2count[obj] += 1
assert obj2count[1] == 1 * 3
assert obj2count[2] == 2 * 3
assert obj2count[3] == 3 * 3
assert obj2count[4] == 4 * 3
def test_unescape_html_string():
assert PythonUtil.unescapeHtmlString('asdf') == 'asdf'
assert PythonUtil.unescapeHtmlString('as+df') == 'as df'
assert PythonUtil.unescapeHtmlString('as%32df') == 'as2df'
assert PythonUtil.unescapeHtmlString('asdf%32') == 'asdf2'
def test_priority_callbacks():
l = []
def a(l=l):
l.append('a')
def b(l=l):
l.append('b')
def c(l=l):
l.append('c')
pc = PythonUtil.PriorityCallbacks()
pc.add(a)
pc()
assert l == ['a']
del l[:]
bItem = pc.add(b)
pc()
assert 'a' in l
assert 'b' in l
assert len(l) == 2
del l[:]
pc.remove(bItem)
pc()
assert l == ['a']
del l[:]
pc.add(c, 2)
bItem = pc.add(b, 10)
pc()
assert l == ['a', 'c', 'b']
del l[:]
pc.remove(bItem)
pc()
assert l == ['a', 'c']
del l[:]
pc.clear()
pc()
assert len(l) == 0
def test_weighted_choice():
# Test PythonUtil.weightedChoice() with no valid list.
with pytest.raises(IndexError):
PythonUtil.weightedChoice([])
# Create a sample choice list.
# This contains a few tuples containing only a weight
# and an arbitrary item.
choicelist = [(3, 'item1'), (1, 'item2'), (7, 'item3')]
# These are the items that we expect.
items = ['item1', 'item2', 'item3']
# Test PythonUtil.weightedChoice() with our choice list.
item = PythonUtil.weightedChoice(choicelist)
# Assert that what we got was at least an available item.
assert item in items
# Create yet another sample choice list, but with a couple more items.
choicelist = [(2, 'item1'), (25, 'item2'), (14, 'item3'), (5, 'item4'),
(7, 'item5'), (3, 'item6'), (6, 'item7'), (50, 'item8')]
# Set the items that we expect again.
items = ['item1', 'item2', 'item3', 'item4', 'item5', 'item6', 'item7', 'item8']
# The sum of all of the weights is 112.
weightsum = 2 + 25 + 14 + 5 + 7 + 3 + 6 + 50
# Test PythonUtil.weightedChoice() with the sum.
item = PythonUtil.weightedChoice(choicelist, sum=weightsum)
# Assert that we got a valid item (most of the time this should be 'item8').
assert item in items
# Test PythonUtil.weightedChoice(), but with an invalid sum.
item = PythonUtil.weightedChoice(choicelist, sum=1)
# Assert that we got 'item1'.
assert item == items[0]
# Create a bogus random function.
rnd = lambda: 0.5
# Test PythonUtil.weightedChoice() with the bogus function.
item = PythonUtil.weightedChoice(choicelist, rng=rnd, sum=weightsum)
# Assert that we got 'item6'.
# We expect 'item6' because 0.5 multiplied by 112 is 56.0.
# When subtracting that number by each weight, it will reach 0
# by the time it hits 'item6' in the iteration.
assert item == items[5]