from gifprime import lzw


def test_decompress_simple():
    assert ''.join(lzw.decompress('D\x01', 2)) == '\x00\x00\x00'


def test_compress_simple():
    # TODO: this fails (uses data from whitepixel gif)
    assert ''.join(lzw.compress('\x00\x00\x00', 2)) == 'D\x01'

@@ -263,8 +263,380 @@
 'D%5Cx01'%0A
+%0A%0Adef test_decompress_5pixels():%0A    # TODO: fails with exception inside decompress (not completely sure what the result should be)%0A    assert ''.join(lzw.decompress('D%5Cx1e%5Cx05', 2)) == '%5Cx00%5Cx01%5Cx01%5Cx01%5Cx01'%0A%0A%0Adef test_compress_5pixels():%0A    # TODO: not completely sure about the correct output%0A    assert ''.join(lzw.compress('%5Cx00%5Cx01%5Cx01%5Cx01%5Cx01', 2)) == 'D%5Cx1e%5Cx05'%0A

Add 2 more tests