diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index b5a0912ca..300558b71 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -192,16 +192,15 @@ class TestMain(TestCase): self.assertIn(abs_ignored_file, coder.abs_fnames) # Test with --add-gitignore-files set to False - with patch("sys.stdout", new_callable=StringIO) as mock_stdout: - coder = main( - ["--no-add-gitignore-files", "--exit", "--yes", abs_ignored_file], - input=DummyInput(), - output=DummyOutput(), - return_coder=True, - force_git_root=git_dir, - ) - # Verify the ignored file is not in the chat - self.assertNotIn(abs_ignored_file, coder.abs_fnames) + coder = main( + ["--no-add-gitignore-files", "--exit", "--yes", abs_ignored_file], + input=DummyInput(), + output=DummyOutput(), + return_coder=True, + force_git_root=git_dir, + ) + # Verify the ignored file is not in the chat + self.assertNotIn(abs_ignored_file, coder.abs_fnames) def test_add_command_gitignore_files_flag(self): with GitTemporaryDirectory() as git_dir: @@ -251,20 +250,19 @@ class TestMain(TestCase): self.assertIn(abs_ignored_file, coder.abs_fnames) # Test with --add-gitignore-files set to False - with patch("sys.stdout", new_callable=StringIO) as mock_stdout: - coder = main( - ["--no-add-gitignore-files", "--exit", "--yes"], - input=DummyInput(), - output=DummyOutput(), - return_coder=True, - force_git_root=git_dir, - ) + coder = main( + ["--no-add-gitignore-files", "--exit", "--yes"], + input=DummyInput(), + output=DummyOutput(), + return_coder=True, + force_git_root=git_dir, + ) - with patch.object(coder.io, "confirm_ask", return_value=True): - coder.commands.cmd_add(rel_ignored_file) + with patch.object(coder.io, "confirm_ask", return_value=True): + coder.commands.cmd_add(rel_ignored_file) - # Verify the ignored file is not in the chat - self.assertNotIn(abs_ignored_file, coder.abs_fnames) + # Verify the ignored file is not in the chat + self.assertNotIn(abs_ignored_file, coder.abs_fnames) def test_main_args(self): with patch("aider.coders.Coder.create") as MockCoder: