style: format code with linter in test_main.py

This commit is contained in:
Paul Gauthier (aider) 2024-08-27 13:53:27 -07:00
parent 3bc1b7c2b7
commit 59cc799b0d

View file

@ -638,24 +638,30 @@ class TestMain(TestCase):
with patch("aider.io.InputOutput.confirm_ask") as mock_confirm_ask:
main(
["--apply", "shell.md", "--no-suggest-shell-commands"],
#input=DummyInput(),
#output=DummyOutput(),
# input=DummyInput(),
# output=DummyOutput(),
)
# Make sure confirm_ask is not called when --no-suggest-shell-commands is used
mock_confirm_ask.assert_called_once()
print("confirm_ask args (with --no-suggest-shell-commands):", mock_confirm_ask.call_args)
print(
"confirm_ask args (with --no-suggest-shell-commands):",
mock_confirm_ask.call_args,
)
with patch("aider.io.InputOutput.confirm_ask") as mock_confirm_ask:
main(
["--apply", "shell.md"],
#input=DummyInput(),
#output=DummyOutput(),
# input=DummyInput(),
# output=DummyOutput(),
)
# Make sure confirm_ask IS called when --no-suggest-shell-commands is not used
mock_confirm_ask.assert_called_once()
print("confirm_ask args (without --no-suggest-shell-commands):", mock_confirm_ask.call_args)
print(
"confirm_ask args (without --no-suggest-shell-commands):",
mock_confirm_ask.call_args,
)
# Check that the file was not created in either case
self.assertFalse(Path("no_suggest_file.txt").exists())