File size: 1,543 Bytes
3315103 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | # -*- coding: utf-8 -*-
from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals
from io import BytesIO
from unittest import TestCase
import os
import zlib
from hwp5.compressed import ZLibIncrementalDecoder
from hwp5.compressed import decompress
from hwp5.compressed import decompress_gen
from hwp5.utils import cached_property
class TestUncompress(TestCase):
@cached_property
def original_data(self):
return os.urandom(16384)
@cached_property
def compressed_data(self):
return zlib.compress(self.original_data)
def test_incremental_decode(self):
compressed_data = self.compressed_data
dec = ZLibIncrementalDecoder(wbits=-15)
data = dec.decode(compressed_data[2:2048])
data += dec.decode(compressed_data[2048:2048 + 1024])
data += dec.decode(compressed_data[2048 + 1024:2048 + 1024 + 4096])
data += dec.decode(compressed_data[2048 + 1024 + 4096:], True)
self.assertEqual(self.original_data, data)
def test_decompress(self):
gen = decompress_gen(BytesIO(self.compressed_data[2:]))
self.assertEqual(self.original_data, b''.join(gen))
# print '-----'
f = decompress(BytesIO(self.compressed_data[2:]))
g = BytesIO(self.original_data)
self.assertEqual(f.read(2048), g.read(2048))
self.assertEqual(f.read(1024), g.read(1024))
self.assertEqual(f.read(4096), g.read(4096))
self.assertEqual(f.read(), g.read())
|