mirror of
https://github.com/jlengrand/Coding4Interviews.git
synced 2026-03-10 08:11:24 +00:00
80 lines
1.4 KiB
Python
80 lines
1.4 KiB
Python
"""
|
|
Unit tests for the Hash Map table implementation
|
|
@jlengrand
|
|
2013/11
|
|
"""
|
|
|
|
from hm import HashMap
|
|
from hm import HMTableCollision
|
|
|
|
import unittest
|
|
|
|
class test_hash_map_table_collision(unittest.TestCase):
|
|
|
|
def test_add(self):
|
|
hm = HMTableCollision()
|
|
|
|
hm.add("a", "Ibiza")
|
|
self.assertEqual(hm.size(), 1)
|
|
|
|
hm.add("a", "Ibiza")
|
|
self.assertEqual(hm.size(), 2)
|
|
|
|
class test_hash_map(unittest.TestCase):
|
|
|
|
def test_hash_size(self):
|
|
hm = HashMap()
|
|
self.assertEqual(hm._hash_size, 513)
|
|
|
|
hm = HashMap(1025)
|
|
self.assertEqual(hm._hash_size, 1025)
|
|
|
|
|
|
def test_size(self):
|
|
hm = HashMap()
|
|
self.assertEqual(hm.size(), 0)
|
|
|
|
def test__hash(self):
|
|
hm = HashMap()
|
|
|
|
value = "a"
|
|
self.assertEqual(hm._hash(value), ord(value))
|
|
|
|
value = "test"
|
|
self.assertEqual(hm._hash(value), 208)
|
|
|
|
value = ""
|
|
self.assertRaises(Exception, lambda x : hm._hash(value))
|
|
|
|
|
|
def test_add(self):
|
|
hm = HashMap()
|
|
|
|
hm.add("a", "Ibiza")
|
|
self.assertEqual(hm.size(), 1)
|
|
|
|
# Tests Collision
|
|
self.assertRaises(Exception, lambda x : hm.add("a", "Ibiza"))
|
|
self.assertEqual(hm.size(), 1)
|
|
|
|
def test_get(self):
|
|
|
|
hm = HashMap()
|
|
value = ""
|
|
self.assertRaises(Exception, lambda x : hm.get(value))
|
|
|
|
key = "One"
|
|
value = "Ibiza"
|
|
hm.add(key, value)
|
|
|
|
hm.add("Two", "NY")
|
|
hm.add("Three", "Berlin")
|
|
hm.add("Four", "Chicago")
|
|
|
|
self.assertEqual(hm.size(), 4)
|
|
|
|
self.assertEqual(hm.get(key), value)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|