Fix tests for Python 2

weasyprint/tests/test_pdf.py

@@ -428,39 +428,38 @@ def test_embedded_files():
     pdf = PdfReader(fdata=pdf_bytes)
     embedded = pdf.Root.Names.EmbeddedFiles.Names
 
-    assert zlib.decompress(
-        embedded[1].EF.F.stream.encode('latin-1')) == b'hi there'
+    def zlib_decompress(stream):
+        if not isinstance(embedded[1].EF.F.stream, bytes):
+            stream = stream.encode('latin-1')
+        return zlib.decompress(stream)
+
+    assert zlib_decompress(embedded[1].EF.F.stream) == b'hi there'
     assert embedded[1].EF.F.Params.CheckSum == (
         '<{}>'.format(hashlib.md5(b'hi there').hexdigest()))
     assert embedded[1].F.decode() == ''
     assert embedded[1].UF.decode() == 'attachment.bin'
     assert embedded[1].Desc.decode() == 'some file attachment äöü'
 
-    assert zlib.decompress(
-        embedded[3].EF.F.stream.encode('latin-1')) == b'12345678'
+    assert zlib_decompress(embedded[3].EF.F.stream) == b'12345678'
     assert embedded[3].EF.F.Params.CheckSum == (
         '<{}>'.format(hashlib.md5(adata).hexdigest()))
     assert embedded[3].UF.decode() == os.path.basename(absolute_tmp_file)
 
-    assert zlib.decompress(
-        embedded[5].EF.F.stream.encode('latin-1')) == b'abcdefgh'
+    assert zlib_decompress(embedded[5].EF.F.stream) == b'abcdefgh'
     assert embedded[5].EF.F.Params.CheckSum == (
         '<{}>'.format(hashlib.md5(rdata).hexdigest()))
     assert embedded[5].UF.decode() == os.path.basename(relative_tmp_file)
 
-    assert zlib.decompress(
-        embedded[7].EF.F.stream.encode('latin-1')) == b'oob attachment'
+    assert zlib_decompress(embedded[7].EF.F.stream) == b'oob attachment'
     assert embedded[7].EF.F.Params.CheckSum == (
         '<{}>'.format(hashlib.md5(b'oob attachment').hexdigest()))
     assert embedded[7].Desc.decode() == 'Hello'
 
-    assert zlib.decompress(
-        embedded[9].EF.F.stream.encode('latin-1')) == b'raw URL'
+    assert zlib_decompress(embedded[9].EF.F.stream) == b'raw URL'
     assert embedded[9].EF.F.Params.CheckSum == (
         '<{}>'.format(hashlib.md5(b'raw URL').hexdigest()))
 
-    assert zlib.decompress(
-        embedded[11].EF.F.stream.encode('latin-1')) == b'file like obj'
+    assert zlib_decompress(embedded[11].EF.F.stream) == b'file like obj'
     assert embedded[11].EF.F.Params.CheckSum == (
         '<{}>'.format(hashlib.md5(b'file like obj').hexdigest()))