def test_snappy_read(self, tmpdir): comp = snappy.StreamCompressor() # generate two chunks with their own framing compressed = comp.compress(b"hello, ") + comp.compress(b"world") file_path = tmpdir.join("foo").strpath with open(file_path, "wb") as fp: fp.write(compressed) out = [] with SnappyFile(open(file_path, "rb"), "rb") as fp: while True: chunk = fp.read() if not chunk: break out.append(chunk) full = b"".join(out) assert full == b"hello, world"
def compress(self, data): return snappy.StreamCompressor().compress(data)