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
|
# cython: c_string_type=unicode, c_string_encoding=utf8
from util.generic.map cimport TMap
from util.generic.string cimport TString
import pytest
import unittest
from libcpp.pair cimport pair
from cython.operator cimport dereference as deref
def _check_convert(TMap[TString, int] x):
return x
class TestMap(unittest.TestCase):
def test_constructors_and_assignments(self):
cdef TMap[TString, int] c1
c1["one"] = 1
c1["two"] = 2
cdef TMap[TString, int] c2 = TMap[TString, int](c1)
self.assertEqual(2, c1.size())
self.assertEqual(2, c2.size())
self.assertEqual(1, c1.at("one"))
self.assertTrue(c1.contains("two"))
self.assertTrue(c2.contains("one"))
self.assertEqual(2, c2.at("two"))
c2["three"] = 3
c1 = c2
self.assertEqual(3, c1.size())
self.assertEqual(3, c2.size())
self.assertEqual(3, c1.at("three"))
def test_equality_operator(self):
cdef TMap[TString, int] base
base["one"] = 1
base["two"] = 2
cdef TMap[TString, int] c1 = TMap[TString, int](base)
self.assertTrue(c1==base)
cdef TMap[TString, int] c2
c2["one"] = 1
c2["two"] = 2
self.assertTrue(c2 == base)
c2["three"] = 3
self.assertTrue(c2 != base)
cdef TMap[TString, int] c3 = TMap[TString, int](base)
c3["one"] = 0
self.assertTrue(c3 != base)
def test_insert_erase(self):
cdef TMap[TString, int] tmp
self.assertTrue(tmp.insert(pair[TString, int]("one", 0)).second)
self.assertFalse(tmp.insert(pair[TString, int]("one", 1)).second)
self.assertTrue(tmp.insert(pair[TString, int]("two", 2)).second)
cdef TString one = "one"
cdef TString two = "two"
self.assertEqual(tmp.erase(one), 1)
self.assertEqual(tmp.erase(two), 1)
self.assertEqual(tmp.size(), 0)
self.assertTrue(tmp.empty())
def test_iterators_and_find(self):
cdef TMap[TString, int] tmp
self.assertTrue(tmp.begin() == tmp.end())
self.assertTrue(tmp.find("1") == tmp.end())
tmp["1"] = 1
self.assertTrue(tmp.begin() != tmp.end())
cdef TMap[TString, int].iterator it = tmp.find("1")
self.assertTrue(it != tmp.end())
self.assertEqual(deref(it).second, 1)
def test_convert(self):
src = {'foo': 1, 'bar': 42}
self.assertEqual(_check_convert(src), src)
bad_src = {'foo': 1, 'bar': 'baz'}
with self.assertRaises(TypeError):
_check_convert(bad_src)
|
