--- a/contrib/python-zstandard/tests/test_roundtrip.py Tue Apr 18 11:22:42 2017 -0400
+++ /dev/null Thu Jan 01 00:00:00 1970 +0000
@@ -1,64 +0,0 @@
-import io
-
-try:
- import unittest2 as unittest
-except ImportError:
- import unittest
-
-try:
- import hypothesis
- import hypothesis.strategies as strategies
-except ImportError:
- raise unittest.SkipTest('hypothesis not available')
-
-import zstd
-
-
-compression_levels = strategies.integers(min_value=1, max_value=22)
-
-
-class TestRoundTrip(unittest.TestCase):
- @hypothesis.given(strategies.binary(), compression_levels)
- def test_compress_write_to(self, data, level):
- """Random data from compress() roundtrips via write_to."""
- cctx = zstd.ZstdCompressor(level=level)
- compressed = cctx.compress(data)
-
- buffer = io.BytesIO()
- dctx = zstd.ZstdDecompressor()
- with dctx.write_to(buffer) as decompressor:
- decompressor.write(compressed)
-
- self.assertEqual(buffer.getvalue(), data)
-
- @hypothesis.given(strategies.binary(), compression_levels)
- def test_compressor_write_to_decompressor_write_to(self, data, level):
- """Random data from compressor write_to roundtrips via write_to."""
- compress_buffer = io.BytesIO()
- decompressed_buffer = io.BytesIO()
-
- cctx = zstd.ZstdCompressor(level=level)
- with cctx.write_to(compress_buffer) as compressor:
- compressor.write(data)
-
- dctx = zstd.ZstdDecompressor()
- with dctx.write_to(decompressed_buffer) as decompressor:
- decompressor.write(compress_buffer.getvalue())
-
- self.assertEqual(decompressed_buffer.getvalue(), data)
-
- @hypothesis.given(strategies.binary(average_size=1048576))
- @hypothesis.settings(perform_health_check=False)
- def test_compressor_write_to_decompressor_write_to_larger(self, data):
- compress_buffer = io.BytesIO()
- decompressed_buffer = io.BytesIO()
-
- cctx = zstd.ZstdCompressor(level=5)
- with cctx.write_to(compress_buffer) as compressor:
- compressor.write(data)
-
- dctx = zstd.ZstdDecompressor()
- with dctx.write_to(decompressed_buffer) as decompressor:
- decompressor.write(compress_buffer.getvalue())
-
- self.assertEqual(decompressed_buffer.getvalue(), data)