76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
import tempfile
|
|
import unittest
|
|
from os.path import join
|
|
|
|
from int2intmaplike import Int2IntMapLike
|
|
|
|
|
|
class Int2IntMapLikeTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.test_dir = tempfile.TemporaryDirectory()
|
|
self.tmpfile = join(self.test_dir.name, "my.log")
|
|
self.map = Int2IntMapLike(self.tmpfile)
|
|
|
|
def tearDown(self):
|
|
self.map.close()
|
|
self.test_dir.cleanup()
|
|
|
|
def test_empty_map(self):
|
|
map = self.map
|
|
self.assertEqual(None, map.find(0))
|
|
|
|
def test_one_line_one_byte(self):
|
|
map = self.map
|
|
map.add(10, 1, 1) # add only the key 10
|
|
self.assertEqual(None, map.find(9)) # directly before
|
|
self.assertEqual(1, map.find(10))
|
|
self.assertEqual(None, map.find(11)) # directly after
|
|
|
|
def test_one_line_two_bytes(self):
|
|
map = self.map
|
|
map.add(10, 2, 1) # added keys 10 and 11
|
|
self.assertEqual(None, map.find(9)) # directly before
|
|
self.assertEqual(1, map.find(10))
|
|
self.assertEqual(1, map.find(11))
|
|
self.assertEqual(None, map.find(12)) # directly after
|
|
|
|
def test_two_lines(self):
|
|
map = self.map
|
|
map.add(10, 1, 1) # added key 10
|
|
map.add(12, 1, 2) # added key 12
|
|
self.assertEqual(None, map.find(9)) # directly before
|
|
self.assertEqual(1, map.find(10))
|
|
self.assertEqual(None, map.find(11)) # between
|
|
self.assertEqual(2, map.find(12))
|
|
self.assertEqual(None, map.find(13)) # directly after
|
|
|
|
def test_fill_map(self):
|
|
map = self.map
|
|
map.blocksize = 64
|
|
|
|
# fill map with
|
|
# 10,5,1
|
|
# 20,5,2
|
|
# 30,5,3
|
|
# ...
|
|
for i in range(1, 20):
|
|
map.add(i * 10, 5, i)
|
|
|
|
self.assertEqual(2, map.find(20))
|
|
self.assertEqual(7, map.find(71))
|
|
self.assertEqual(13, map.find(134))
|
|
self.assertEqual(19, map.find(194))
|
|
|
|
# values that are not in the map
|
|
self.assertEqual(None, map.find(0))
|
|
self.assertEqual(None, map.find(9))
|
|
self.assertEqual(None, map.find(15))
|
|
self.assertEqual(None, map.find(16))
|
|
self.assertEqual(None, map.find(107)) # a value in the second block
|
|
self.assertEqual(None, map.find(188)) # a value in the third block
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|