class BloomFilterTestCase(unittest.TestCase):
def setUp(self):
self.bf = BloomFilter(262144, 14)
lines = open("/usr/share/dict/american-english").read().splitlines()
for line in lines:
self.bf.update(line)
def tearDown(self):
pass
def test_probably(self):
self.assertEqual(self.bf.lookup("Max"), "Probably")
self.assertEqual(self.bf.lookup("mice"), "Probably")
def test_nope(self):
self.assertEqual(self.bf.lookup("3"), "Nope")
self.assertEqual(self.bf.lookup("google"), "Nope")
