diff --git a/tests/test_main.py b/tests/test_main.py index 696b390f..f2c2843e 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -53,8 +53,8 @@ def test_get_command(): return_value=True): Popen.return_value.stdout.read.return_value = b'stdout' Popen.return_value.stderr.read.return_value = b'stderr' - assert main.get_command(Mock(), [b'thefuck', b'apt-get', - b'search', b'vim']) \ + assert main.get_command(Mock(), ['thefuck', 'apt-get', + 'search', 'vim']) \ == main.Command('apt-get search vim', 'stdout', 'stderr') Popen.assert_called_once_with('apt-get search vim', shell=True,