def testDumpAndLoadBase64BloomFilter(self): bloom_filter = BloomFilter(self.BLOOM_CAPACITY, self.BLOOM_ERROR_RATE) for key in self.all_keys: bloom_filter.add(key); dump_str = bloom_filter.dump_to_base64_str(gzipped=True); bloom_filter2 = BloomFilter.load_from_base64_str(dump_str); self.assertEqual(bloom_filter, bloom_filter2) self.check_contains(bloom_filter2)
def testLoadBloomFilterFromBase64File(self): with open(self.BF_DUMP_FILE_BASE_64, "r") as f: dump_str = f.read() bloom_filter = BloomFilter.load_from_base64_str(dump_str) self.check_contains(bloom_filter)