24 lines
889 B
Python
24 lines
889 B
Python
|
import unittest
|
||
|
|
|
||
|
|
from compress import compress_data
|
||
|
|
|
||
|
|
|
||
|
|
class TestCompress(unittest.TestCase):
|
||
|
|
def test_compress_abcdef(self):
|
||
|
|
input_data = bytearray(b"abcdef")
|
||
|
|
output_data, delta = compress_data(input_data, skip=0,
|
||
|
|
backwards_mode=False, classic_mode=False, quick_mode=False)
|
||
|
|
|
||
|
|
expected_output_data = bytearray(b'MabcdefUV')
|
||
|
|
self.assertEqual(expected_output_data, output_data)
|
||
|
|
self.assertEqual(2, delta)
|
||
|
|
|
||
|
|
def test_compress_abcabc(self):
|
||
|
|
input_data = bytearray(b"abcabc")
|
||
|
|
output_data, delta = compress_data(input_data, skip=0,
|
||
|
|
backwards_mode=False, classic_mode=False, quick_mode=False)
|
||
|
|
|
||
|
|
expected_output_data = bytearray(b'{abc\xfaUU\x80')
|
||
|
|
self.assertEqual(expected_output_data, output_data)
|
||
|
|
self.assertEqual(3, delta)
|