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")