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
|
import pytest
from unittestmock import UnitTestMock
from cykhash import Int64toInt64Map, Int32toInt32Map, Float64toInt64Map, Float32toInt32Map, PyObjectMap
from cykhash import Int64toFloat64Map, Int32toFloat32Map, Float64toFloat64Map, Float32toFloat32Map
@pytest.mark.parametrize(
"map_type",
[
Int64toInt64Map, Int32toInt32Map, Float64toInt64Map, Float32toInt32Map,
Int64toFloat64Map, Int32toFloat32Map, Float64toFloat64Map, Float32toFloat32Map,
PyObjectMap]
)
class TestMapIterator(UnitTestMock):
def test_iterate(self, map_type):
cy_map = map_type()
py_map = dict()
for i in range(10):
cy_map[i] = i+1
py_map[i] = i+1
clone = dict()
for x in cy_map.items():
clone[x[0]] = x[1]
self.assertEqual(py_map, clone)
def test_iterate_after_clear(self, map_type):
cy_map = map_type(zip(range(1000), range(1000)))
it = iter(cy_map.items())
for x in range(1000):
next(it)
cy_map.clear()
with pytest.raises(StopIteration):
next(it)
def test_iterate_after_growing(self, map_type):
cy_map = map_type()
it = iter(cy_map.keys())
#change bucket size
for i in range(1000):
cy_map[i]=i
#make sure new size is used
lst = []
with pytest.raises(StopIteration):
for x in range(1001):
lst.append(next(it))
self.assertEqual(set(lst), set(range(1000)))
def test_iterate_after_growing2(self, map_type):
cy_map = map_type()
cy_map[42]=42
it = iter(cy_map.keys())
next(it) #iterator shows to end now
cy_map[11]=11
cy_map[15]=15
cy_map[13]=13
#old end is no longer an end
try:
self.assertTrue(next(it) in {42,11,15,13})
except StopIteration:
pass # stop iteration could be an acceptable outcome
|
