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)