Implementação de um HashMap simples em Python

hashmap.py

class HashMap(object):
    def __init__(self, bucket_size=10):
        self.bucket_size = bucket_size
        self.buckets = [None] * self.bucket_size

    def insert(self, key, value):
        hashed_key = hash(key)
        bucket_index = hashed_key % self.bucket_size
        self.buckets[bucket_index] = value

    def retrieve(self, key):
        hashed_key = hash(key)
        bucket_index = hashed_key % self.bucket_size
        return self.buckets[bucket_index]


hm = HashMap()
hm.insert("Teste", 12345)
print(hm.retrieve("Teste"))
print(hm.retrieve("Teste2"))