py3k fix

alex · Nov 9, 2013 · 5849481 · 5849481
1 parent e1a8540
commit 5849481
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_zero_buffer.py b/test_zero_buffer.py
@@ -337,7 +337,7 @@ def test_write_to(self, buf, tmpdir):
         with tmpdir.join("t.txt").open("wb") as f:
             res = view.write_to(f.fileno())
         assert res == 4
-        assert tmpdir.join("t.txt").read() == b"abcd"
+        assert tmpdir.join("t.txt").read("rb") == b"abcd"
 
     def test_write_to_badfd(self, buf):
         buf.add_bytes(b"abcd")
@@ -351,7 +351,7 @@ def test_write_to_empty_bytes(self, buf, tmpdir):
         with tmpdir.join("t.txt").open("wb") as f:
             res = view.write_to(f.fileno())
         assert res == 0
-        assert tmpdir.join("t.txt").read() == b""
+        assert tmpdir.join("t.txt").read("rb") == b""