MirroredRepos/thefuck
1
0
Fork 0
mirror of https://github.com/nvbn/thefuck.git synced 2025-03-20 09:39:01 +00:00
Code Issues Releases Wiki Activity

Change files using black and flake8 style check

Browse Source
This commit is contained in:
Boon Wenjie 2019-10-28 17:54:52 +00:00
parent d7e2d4e3c6
commit 648d63869b
2 changed files with 24 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
tests/rules/test_remove_shell_prompt_literal.py
View File

@ -3,30 +3,31 @@ from thefuck.rules.remove_shell_prompt_literal import match, get_new_command
from thefuck.types import Command from thefuck.types import Command
@pytest.mark.parametrize('command', @pytest.mark.parametrize(
"command",
[ [
Command('$ cd newdir', '$: command not found'), Command("$ cd newdir", "$: command not found"),
Command(' $ cd newdir', '$: command not found'), Command(" $ cd newdir", "$: command not found"),
]) ],
)
def test_match(command): def test_match(command):
assert match(command) assert match(command)
@pytest.mark.parametrize('command', @pytest.mark.parametrize(
[ "command",
Command('$', ''), [Command("$", ""), Command("$?", ""), Command(" $?", ""), Command("", "")],
Command('$?', ''), )
Command(' $?', ''),
Command('', ''),
])
def test_not_match(command): def test_not_match(command):
assert not match(command) assert not match(command)
@pytest.mark.parametrize('command, new_command', @pytest.mark.parametrize(
"command, new_command",
[ [
(Command('$ cd newdir', ''), 'cd newdir'), (Command("$ cd newdir", ""), "cd newdir"),
(Command('$ python3 -m virtualenv env', ''), 'python3 -m virtualenv env'), (Command("$ python3 -m virtualenv env", ""), "python3 -m virtualenv env"),
]) ],
)
def test_get_new_command(command, new_command): def test_get_new_command(command, new_command):
assert get_new_command(command) == new_command assert get_new_command(command) == new_command

9
thefuck/rules/remove_shell_prompt_literal.py
View File

@ -10,11 +10,16 @@ bash: $: command not found...
import re import re
def match(command): def match(command):
return ("$: command not found" in command.output and return (
re.search(r"^[\s]*\$ [\S]+", command.script) is not None) "$: command not found" in command.output
and re.search(r"^[\s]*\$ [\S]+", command.script) is not None
)
def get_new_command(command): def get_new_command(command):
return command.script.replace("$", "", 1).strip() return command.script.replace("$", "", 1).strip()
requires_output = True requires_output = True