class BloomFilter(object):

    def __init__(self, hash_funcs, n_bits=32):
        self._bitarray = BitArray()
        self._hash_funcs = hash_funcs
        self._n_bits = n_bits

    def insert(self, val):
        for f in self._hash_funcs:
            self._bitarray.set(f(val) % self._n_bits)

    def exists(self, val):
        return all(self._bitarray.get(f(val) % self._n_bits)
                   for f in self._hash_funcs)
Exemple #2
0
 def setUp(self):
     self.array = BitArray()
Exemple #3
0
class TestBitArray(unittest.TestCase):

    def setUp(self):
        self.array = BitArray()

    def test_get_on_empty_array(self):
        self.assertEquals(self.array.get(0), 0)

    def test_get_on_one_value_array(self):
        self.array.set(1)

        self.assertEquals(self.array.get(0), 0)
        self.assertEquals(self.array.get(1), 1)
        self.assertEquals(self.array.get(2), 0)

    def test_get_on_multi_value_array(self):
        self.array.set(3)
        self.array.set(5)
        self.array.set(5000)

        self.assertEquals(self.array.get(3), 1)
        self.assertEquals(self.array.get(4), 0)
        self.assertEquals(self.array.get(5), 1)
        self.assertEquals(self.array.get(6), 0)

        self.assertEquals(self.array.get(4999), 0)
        self.assertEquals(self.array.get(5000), 1)
        self.assertEquals(self.array.get(5001), 0)

    def test_set_on_empty_array(self):
        self.array.set(0)

        self.assertEquals(self.array.get(0), 1)


    def test_toggle_on_empty_array(self):
        self.array.toggle(0)

        self.assertEquals(self.array.get(0), 1)

    def test_clear_on_empty_array(self):
        self.array.clear(0)

        self.assertEquals(self.array.get(0), 0)

    def test_is_set_on_empty_array(self):
        self.assertFalse(self.array.is_set(0))
 def __init__(self, hash_funcs, n_bits=32):
     self._bitarray = BitArray()
     self._hash_funcs = hash_funcs
     self._n_bits = n_bits
Exemple #5
0
 def setUp(self):
     self.array = BitArray()
Exemple #6
0
class TestBitArray(unittest.TestCase):
    def setUp(self):
        self.array = BitArray()

    def test_get_on_empty_array(self):
        self.assertEquals(self.array.get(0), 0)

    def test_get_on_one_value_array(self):
        self.array.set(1)

        self.assertEquals(self.array.get(0), 0)
        self.assertEquals(self.array.get(1), 1)
        self.assertEquals(self.array.get(2), 0)

    def test_get_on_multi_value_array(self):
        self.array.set(3)
        self.array.set(5)
        self.array.set(5000)

        self.assertEquals(self.array.get(3), 1)
        self.assertEquals(self.array.get(4), 0)
        self.assertEquals(self.array.get(5), 1)
        self.assertEquals(self.array.get(6), 0)

        self.assertEquals(self.array.get(4999), 0)
        self.assertEquals(self.array.get(5000), 1)
        self.assertEquals(self.array.get(5001), 0)

    def test_set_on_empty_array(self):
        self.array.set(0)

        self.assertEquals(self.array.get(0), 1)

    def test_toggle_on_empty_array(self):
        self.array.toggle(0)

        self.assertEquals(self.array.get(0), 1)

    def test_clear_on_empty_array(self):
        self.array.clear(0)

        self.assertEquals(self.array.get(0), 0)

    def test_is_set_on_empty_array(self):
        self.assertFalse(self.array.is_set(0))