diff --git a/aider/tests/test_coder.py b/aider/tests/test_coder.py index ca3014ff9..cb021cd17 100644 --- a/aider/tests/test_coder.py +++ b/aider/tests/test_coder.py @@ -219,7 +219,7 @@ class TestCoder(unittest.TestCase): files = [file1, file2] # Initialize the Coder object with the mocked IO and mocked repo - coder = Coder.create(self.GPT35, None, io=InputOutput(), fnames=files) + coder = Coder.create(self.GPT35, None, io=InputOutput(), fnames=files, pretty=False) def mock_send(*args, **kwargs): coder.partial_response_content = "ok" @@ -246,7 +246,7 @@ class TestCoder(unittest.TestCase): files = [file1, file2] # Initialize the Coder object with the mocked IO and mocked repo - coder = Coder.create(self.GPT35, None, io=InputOutput(), fnames=files) + coder = Coder.create(self.GPT35, None, io=InputOutput(), fnames=files, pretty=False) def mock_send(*args, **kwargs): coder.partial_response_content = "ok" @@ -337,7 +337,7 @@ class TestCoder(unittest.TestCase): fname = Path("file.txt") io = InputOutput(yes=True) - coder = Coder.create(self.GPT35, "diff", io=io, fnames=[str(fname)]) + coder = Coder.create(self.GPT35, "diff", io=io, fnames=[str(fname)], pretty=False) self.assertTrue(fname.exists()) @@ -394,7 +394,9 @@ new fname1.write_text("ONE\n") io = InputOutput(yes=True) - coder = Coder.create(self.GPT35, "diff", io=io, fnames=[str(fname1), str(fname2)]) + coder = Coder.create( + self.GPT35, "diff", io=io, fnames=[str(fname1), str(fname2)], pretty=False + ) def mock_send(*args, **kwargs): coder.partial_response_content = f""" @@ -447,7 +449,7 @@ TWO fname2.write_text("OTHER\n") io = InputOutput(yes=True) - coder = Coder.create(self.GPT35, "diff", io=io, fnames=[str(fname)]) + coder = Coder.create(self.GPT35, "diff", io=io, fnames=[str(fname)], pretty=False) def mock_send(*args, **kwargs): coder.partial_response_content = f""" @@ -525,7 +527,7 @@ three repo.git.commit("-m", "initial") io = InputOutput(yes=True) - coder = Coder.create(self.GPT35, "diff", io=io, fnames=[str(fname)]) + coder = Coder.create(self.GPT35, "diff", io=io, fnames=[str(fname)], pretty=False) def mock_send(*args, **kwargs): coder.partial_response_content = f""" diff --git a/aider/tests/test_editblock.py b/aider/tests/test_editblock.py index 40a0d4572..92083c0b1 100644 --- a/aider/tests/test_editblock.py +++ b/aider/tests/test_editblock.py @@ -297,7 +297,7 @@ These changes replace the `subprocess.run` patches with `subprocess.check_output files = [file1] # Initialize the Coder object with the mocked IO and mocked repo - coder = Coder.create(self.GPT35, "diff", io=InputOutput(), fnames=files) + coder = Coder.create(self.GPT35, "diff", io=InputOutput(), fnames=files, pretty=False) def mock_send(*args, **kwargs): coder.partial_response_content = f""" @@ -340,6 +340,7 @@ new io=InputOutput(dry_run=True), fnames=files, dry_run=True, + pretty=False, ) def mock_send(*args, **kwargs): diff --git a/aider/tests/test_wholefile.py b/aider/tests/test_wholefile.py index 10e996dec..56742c71d 100644 --- a/aider/tests/test_wholefile.py +++ b/aider/tests/test_wholefile.py @@ -309,7 +309,6 @@ Do this: return [] coder.send = MagicMock(side_effect=mock_send) - coder.live_incremental_response = lambda x: x # Call the run method with a message coder.run(with_message="hi")