blob: 0ca64f4e559230a595bdf92103bf0edd82d6ceeb (
plain) (
blame)
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
|
#! /usr/bin/env python2
#-*- coding:utf-8 -*-
from __future__ import print_function
from builtins import range
import unittest
class TestUtils(unittest.TestCase):
def test_boundedDict(self):
from miasm.core.utils import BoundedDict
# Use a callback
def logger(key):
print("DELETE", key)
# Create a 5/2 dictionary
bd = BoundedDict(5, 2, initialdata={"element": "value"},
delete_cb=logger)
bd["element2"] = "value2"
assert("element" in bd)
assert("element2" in bd)
self.assertEqual(bd["element"], "value")
self.assertEqual(bd["element2"], "value2")
# Increase 'element2' use
_ = bd["element2"]
for i in range(6):
bd[i] = i
print("Insert %d -> %s" % (i, bd))
assert(len(bd) == 2)
assert("element2" in bd)
self.assertEqual(bd["element2"], "value2")
if __name__ == '__main__':
testsuite = unittest.TestLoader().loadTestsFromTestCase(TestUtils)
report = unittest.TextTestRunner(verbosity=2).run(testsuite)
exit(len(report.errors + report.failures))
|