def test_bz2(self):
"""Successful bz2 compression"""
data = self._data
compress = verify_and_get_compress_func("bz2")
import bz2
self.assertEqual(data, bz2.decompress(compress(data)))
def test_zlib(self):
"""Successful zlib compression"""
data = self._data
compress = verify_and_get_compress_func("deflate")
import zlib
self.assertEqual(data, zlib.decompress(compress(data)))
def test_lz4(self):
data = self._data
compress = verify_and_get_compress_func("lz4")
self.assertIsNotNone(compress)
import lz4.frame as lz4
self.assertEqual(data, lz4.decompress(compress(data)))
def test_zstandard(self):
data = self._data
compress = verify_and_get_compress_func("zstandard")
self.assertIsNotNone(compress)
import zstandard
decompressor = zstandard.ZstdDecompressor()
self.assertEqual(data, decompressor.decompress(compress(data)))
def test_compression_none(self):
data = self._data
compress = verify_and_get_compress_func("none")
self.assertIsNotNone(compress)
self.assertEqual(data, compress(data))
compress_func, decompress_func = get_compress_and_decompress_func(
"none")
self.assertIsNotNone(compress_func)
self.assertIsNotNone(decompress_func)
self.assertEqual(data, compress_func(data))
self.assertEqual(data, decompress_func(data))
self.assertEqual(data, decompress_func(compress_func(data)))
def _test(compression_type):
self.assertIsNone(verify_and_get_compress_func(compression_type))
def test_bad_compression_type(self):
"""User enters unsupported compression type"""
self.assertIsNone(verify_and_get_compress_func("bad_compression_type"))
