dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-28 08:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fixed test_editblock and _wholefile

Browse source
This commit is contained in:
Paul Gauthier 2024-04-18 16:02:50 -07:00
parent 512b991a51
commit 2a3eb8ac35
2 changed files with 18 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/test_editblock.py
View file

@ -5,21 +5,16 @@ import unittest
from pathlib import Path
from unittest.mock import MagicMock, patch
from aider import models
from aider.coders import Coder
from aider.coders import editblock_coder as eb
from aider.dump import dump # noqa: F401
from aider.io import InputOutput
from aider.models import Model
class TestUtils(unittest.TestCase):
def setUp(self):
self.patcher = patch("aider.coders.base_coder.check_model_availability")
self.mock_check = self.patcher.start()
self.mock_check.return_value = True
def tearDown(self):
self.patcher.stop()
self.GPT35 = Model("gpt-3.5-turbo")
# fuzzy logic disabled v0.11.2-dev
def __test_replace_most_similar_chunk(self):
@ -302,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(models.GPT4, "diff", io=InputOutput(), fnames=files)
coder = Coder.create(self.GPT35, "diff", io=InputOutput(), fnames=files)
def mock_send(*args, **kwargs):
coder.partial_response_content = f"""
@ -339,7 +334,7 @@ new
# Initialize the Coder object with the mocked IO and mocked repo
coder = Coder.create(
models.GPT4,
self.GPT35,
"diff",
io=InputOutput(dry_run=True),
fnames=files,