Fix tests

kkroening · kkroening · commit 6a2d3381b79d · 2018-05-20T01:43:43.000-07:00
diff --git a/ffmpeg/tests/test_ffmpeg.py b/ffmpeg/tests/test_ffmpeg.py
@@ -329,20 +329,20 @@ def test__run__capture_out(mocker, capture_stdout, capture_stderr):
     stream = _get_simple_example()
     out, err = ffmpeg.run(stream, capture_stdout=capture_stdout, capture_stderr=capture_stderr)
     if capture_stdout:
-        assert out == 'test\n'
+        assert out == 'test\n'.encode()
     else:
         assert out is None
     if capture_stderr:
-        assert err == ''
+        assert err == ''.encode()
     else:
         assert err is None
 
 
 def test__run__input_output(mocker):
     mocker.patch.object(ffmpeg._run, 'compile', return_value=['cat'])
     stream = _get_simple_example()
-    out, err = ffmpeg.run(stream, input='test', capture_stdout=True)
-    assert out == 'test'
+    out, err = ffmpeg.run(stream, input='test'.encode(), capture_stdout=True)
+    assert out == 'test'.encode()
     assert err is None
 
 
@@ -357,11 +357,11 @@ def test__run__error(mocker, capture_stdout, capture_stderr):
     out = excinfo.value.stdout
     err = excinfo.value.stderr
     if capture_stdout:
-        assert out == ''
+        assert out == ''.encode()
     else:
         assert out is None
     if capture_stderr:
-        assert err.startswith('ffmpeg version')
+        assert err.decode().startswith('ffmpeg version')
     else:
         assert err is None
 
