wip

1 files changed, 14 insertions, 10 deletions

diff --git a/test/test_encoding.py b/test/test_encoding.py
index 612aea89..9da3a38d 100644
--- a/test/test_encoding.py
+++ b/test/test_encoding.py
@@ -9,25 +9,29 @@ def test_identity():
 
 
 def test_gzip():
-    assert "string" == encoding.decode(
+    assert b"string" == encoding.decode(
         "gzip",
         encoding.encode(
             "gzip",
-            "string"))
-    assert None == encoding.decode("gzip", "bogus")
+            b"string"
+        )
+    )
+    assert encoding.decode("gzip", b"bogus") is None
 
 
 def test_deflate():
-    assert "string" == encoding.decode(
+    assert b"string" == encoding.decode(
         "deflate",
         encoding.encode(
             "deflate",
-            "string"))
-    assert "string" == encoding.decode(
+            b"string"
+        )
+    )
+    assert b"string" == encoding.decode(
         "deflate",
         encoding.encode(
             "deflate",
-            "string")[
-            2:-
-            4])
-    assert None == encoding.decode("deflate", "bogus")
+            b"string"
+        )[2:-4]
+    )
+    assert encoding.decode("deflate", b"bogus") is None