thefuck/tests/rules/test_python_command.py

from thefuck.rules.python_command import match, get_new_command
from tests.utils import Command


def test_match():
    assert match(Command('temp.py', stderr='Permission denied'), None)
    assert not match(Command(), None)


def test_get_new_command():
    assert get_new_command(Command('./test_sudo.py'), None)\
           == 'python ./test_sudo.py'