added types

data_structures/hashing/bloom_filter.py

@@ -72,25 +72,25 @@ def __init__(self, size: int = 8) -> None:
         self.size = size
 
     def add(self, value: str) -> None:
-        h = self.hash_(value)
+        h = self.hash(value)
         self.bitarray |= h
 
     def exists(self, value: str) -> bool:
-        h = self.hash_(value)
+        h = self.hash(value)
         return (h & self.bitarray) == h
 
-    def __contains__(self, other):
+    def __contains__(self, other: str) -> bool:
         return self.exists(other)
 
     def format_bin(self, bitarray: int) -> str:
         res = bin(bitarray)[2:]
         return res.zfill(self.size)
 
     @property
-    def bitstring(self):
+    def bitstring(self) -> None:
         return self.format_bin(self.bitarray)
 
-    def hash_(self, value: str) -> int:
+    def hash(self, value: str) -> int:
         res = 0b0
         for func in HASH_FUNCTIONS:
             b = func(value.encode()).digest()
@@ -99,9 +99,9 @@ def hash_(self, value: str) -> int:
         return res
 
     def format_hash(self, value: str) -> str:
-        return self.format_bin(self.hash_(value))
+        return self.format_bin(self.hash(value))
 
-    def estimated_error_rate(self):
+    def estimated_error_rate(self) -> float:
         n_ones = bin(self.bitarray).count("1")
         k = len(HASH_FUNCTIONS)
         return (n_ones / self.size) ** k