import pytest from thefuck.rules.touch import match, get_new_command from thefuck.types import Command @pytest.fixture def output(is_bsd): print(is_bsd) if is_bsd: return "touch: /a/b/c: No such file or directory" return "touch: cannot touch '/a/b/c': No such file or directory" @pytest.mark.parametrize('script, is_bsd', [ ('touch /a/b/c', False), ('touch /a/b/c', True)]) def test_match(script, is_bsd, output): command = Command(script, output) assert match(command) @pytest.mark.parametrize('command', [ Command('touch /a/b/c', ''), Command('ls /a/b/c', output(False))]) def test_not_match(command): assert not match(command) @pytest.mark.parametrize('script, is_bsd', [ ('touch /a/b/c', False), ('touch /a/b/c', True)]) def test_get_new_command(script, is_bsd, output): command = Command(script, output) fixed_command = get_new_command(command) assert fixed_command == 'mkdir -p /a/b && touch /a/b/c'