HuffmanCompressionUnitTest

Complexity

Total Complexity

2

Size/Duplication

Total Lines	15
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	text_huffman_string()	0	3	1
A	test_huffman_binary()	0	9	1

import unittest

from pycompressor.huffman import HuffmanCompressor


class HuffmanCompressionUnitTest(unittest.TestCase):

    def test_huffman_binary(self):
        huffman = HuffmanCompressor()
        text = 'Hello World'
        bit_stream = huffman.compress_to_binary(text)
        self.assertFalse(bit_stream.is_empty())
        print(bit_stream.size())
        decompressed = huffman.decompress_from_binary(bit_stream)
        print(decompressed)
        self.assertEqual(text, decompressed)

    def text_huffman_string(self):
        huffman = HuffmanCompressor()
        text = 'Hello World'


if __name__ == '__main__':
    unittest.main()