From 702f8999b3311a997b104c650885436c09ad751d Mon Sep 17 00:00:00 2001 From: Julien Lengrand-Lambert Date: Thu, 21 Nov 2013 11:40:29 +0100 Subject: [PATCH] Adds custom size of hash map table --- 02_hash_maps/hm.py | 3 ++- 02_hash_maps/hm_test.py | 8 ++++++++ 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/02_hash_maps/hm.py b/02_hash_maps/hm.py index 9cf3a4e..5daa599 100644 --- a/02_hash_maps/hm.py +++ b/02_hash_maps/hm.py @@ -5,7 +5,8 @@ HashMap Table Implementation """ class HashMap(): - def __init__(self): + def __init__(self, hash_size=513): + self._hash_size = hash_size self._size = 0 def size(self): diff --git a/02_hash_maps/hm_test.py b/02_hash_maps/hm_test.py index 5f16779..d986053 100644 --- a/02_hash_maps/hm_test.py +++ b/02_hash_maps/hm_test.py @@ -11,6 +11,14 @@ import unittest 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)