def test_remove(self):
root = imdict.node(1, 1)
root = imdict.add(root, 1, 2)
root = imdict.add(root, 2, 2)
root = imdict.add(root, 0, 2)
root = imdict.remove(root, 1)
self.assertEqual(imdict.to_list(root), [[0, 2], [2, 2]])
def test_reduce(self):
def func(k, j):
return k + j
root = imdict.node(1, 1)
root = imdict.add(root, 1, 2)
root = imdict.add(root, 2, 2)
root = imdict.add(root, 0, 2)
sum = imdict.reduce(iter(root), func)
self.assertEqual(sum, 6)
def test_iterator(self):
root = imdict.node(1, 1)
root = imdict.add(root, 1, 2)
root = imdict.add(root, 2, 2)
root = imdict.add(root, 0, 2)
list = imdict.to_list(root)
itor = iter(root)
test = []
while itor.has_next():
test.append(next(itor))
self.assertEqual(test, list)
self.assertRaises(StopIteration, lambda: next(itor))
def test_map(self):
def func(k):
k + 1
root = imdict.node(1, 1)
root = imdict.add(root, 1, 2)
root = imdict.add(root, 2, 2)
root = imdict.add(root, 0, 2)
list = imdict.to_list(root)
list2 = []
for i in list:
i[1] = func(i[1])
list2.append(i)
itor = imdict.map(root, func)
test = []
while itor.has_next():
test.append(next(itor))
self.assertEqual(test, list2)
def test_filter(self):
def func(k):
if k % 2 == 0:
return True
return False
root = imdict.node(1, 1)
root = imdict.add(root, 1, 2)
root = imdict.add(root, 2, 2)
root = imdict.add(root, 0, 2)
list = imdict.to_list(root)
list2 = []
for i in range(len(list)):
if func(list[i][0]):
list2.append(list[i])
itor = imdict.filter(root, func)
test = []
while itor.has_next():
test.append(next(itor))
self.assertEqual(test, list2)
def test_mconcat(self):
t1 = imdict.node(1, 1)
t1 = imdict.add(t1, 1, 2)
t1 = imdict.add(t1, 2, 2)
t1 = imdict.add(t1, 0, 2)
t2 = imdict.node(-1, 1)
t2 = imdict.add(t2, -1, 2)
t2 = imdict.add(t2, 3, 2)
t2 = imdict.add(t2, 1, 3)
t3 = imdict.mconcat(t1, t2)
self.assertEqual(imdict.to_list(t3), [[-1, 2], [0, 2], [1, 2], [2, 2], [3, 2]])
t3 = imdict.mconcat(t2, t1)
self.assertEqual(imdict.to_list(t3), [[-1, 2], [0, 2], [1, 2], [2, 2], [3, 2]])
def test_find(self):
root = imdict.node(1, 1)
root = imdict.add(root, 1, 2)
root = imdict.add(root, 2, 2)
root = imdict.add(root, 0, 2)
self.assertEqual(imdict.find(root, 2), 2)
def test_to_list(self):
root = imdict.node(0, 1)
root = imdict.add(root, 2, 1)
root = imdict.add(root, 3, 1)
self.assertEqual(imdict.to_list(root), [[0, 1], [2, 1], [3, 1]])
def test_size(self):
root = imdict.node(1, 1)
root = imdict.add(root, 1, 2)
root = imdict.add(root, 2, 2)
root = imdict.add(root, 0, 2)
self.assertEqual(imdict.size(root), 3)
