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
|
"""Stress test diskcache.persistent.Index."""
import collections as co
import itertools as it
import random
import diskcache as dc
KEYS = 100
OPERATIONS = 25000
SEED = 0
functions = []
def register(function):
functions.append(function)
return function
@register
def stress_get(mapping, index):
key = random.randrange(KEYS)
assert mapping.get(key, None) == index.get(key, None)
@register
def stress_set(mapping, index):
key = random.randrange(KEYS)
value = random.random()
mapping[key] = value
index[key] = value
register(stress_set)
register(stress_set)
register(stress_set)
@register
def stress_pop(mapping, index):
key = random.randrange(KEYS)
assert mapping.pop(key, None) == index.pop(key, None)
@register
def stress_popitem(mapping, index):
if len(mapping) == len(index) == 0:
return
elif random.randrange(2):
assert mapping.popitem() == index.popitem()
else:
assert mapping.popitem(last=False) == index.popitem(last=False)
@register
def stress_iter(mapping, index):
iterator = it.islice(zip(mapping, index), 5)
assert all(alpha == beta for alpha, beta in iterator)
@register
def stress_reversed(mapping, index):
reversed_mapping = reversed(mapping)
reversed_index = reversed(index)
pairs = it.islice(zip(reversed_mapping, reversed_index), 5)
assert all(alpha == beta for alpha, beta in pairs)
@register
def stress_len(mapping, index):
assert len(mapping) == len(index)
def stress(mapping, index):
for count in range(OPERATIONS):
function = random.choice(functions)
function(mapping, index)
if count % 1000 == 0:
print('\r', len(mapping), ' ' * 7, end='')
print()
def test():
random.seed(SEED)
mapping = co.OrderedDict(enumerate(range(KEYS)))
index = dc.Index(enumerate(range(KEYS)))
stress(mapping, index)
assert mapping == index
if __name__ == '__main__':
test()
|
