type hints requested by boot

This commit is contained in:
Isidro Arias 2023-04-06 14:17:32 +02:00
parent 392616713d
commit 280ffa0564

View File

@ -10,7 +10,7 @@ class Bloom:
# number of hash functions is fixed
HASH_FUNCTIONS = (sha256, md5)
def __init__(self, size=8):
def __init__(self, size: int = 8) -> None:
self.bitstring = 0b0
self.size = size
@ -52,7 +52,7 @@ class Bloom:
return res
def test_movies():
def test_movies() -> None:
b = Bloom()
b.add("Titanic")
b.add("Avatar")
@ -66,11 +66,11 @@ def test_movies():
assert b.exists("Pulp Fiction") in (True, False)
def random_string(size):
def random_string(size: int) -> str:
return "".join(choices(ascii_lowercase + " ", k=size))
def test_probability(m=64, n=20):
def test_probability(m: int = 64, n: int = 20) -> None:
b = Bloom(size=m)
k = len(b.HASH_FUNCTIONS)