dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-23 05:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fixed test

Browse source
This commit is contained in:
Paul Gauthier 2023-06-20 17:27:48 -07:00
parent 4956c35870
commit 124ab7dd99
1 changed files with 3 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_commands.py
View file

@ -2,8 +2,8 @@ import os
import shutil import shutil
import tempfile import tempfile
from unittest import TestCase from unittest import TestCase
from unittest.mock import patch
from aider import models
from aider.commands import Commands from aider.commands import Commands
from aider.io import InputOutput from aider.io import InputOutput
@ -14,22 +14,16 @@ class TestCommands(TestCase):
self.tempdir = tempfile.mkdtemp() self.tempdir = tempfile.mkdtemp()
os.chdir(self.tempdir) os.chdir(self.tempdir)
self.patcher = patch("aider.coder.Coder.check_model_availability")
self.mock_check = self.patcher.start()
self.mock_check.return_value = True
def tearDown(self): def tearDown(self):
os.chdir(self.original_cwd) os.chdir(self.original_cwd)
shutil.rmtree(self.tempdir) shutil.rmtree(self.tempdir)
self.patcher.stop()
def test_cmd_add(self): def test_cmd_add(self):
# Initialize the Commands and InputOutput objects # Initialize the Commands and InputOutput objects
io = InputOutput(pretty=False, yes=True) io = InputOutput(pretty=False, yes=True)
from aider.coder import Coder from aider.coders import Coder
coder = Coder(io, openai_api_key="deadbeef") coder = Coder.create(models.GPT35, io, openai_api_key="deadbeef")
commands = Commands(io, coder) commands = Commands(io, coder)
# Call the cmd_add method with 'foo.txt' and 'bar.txt' as a single string # Call the cmd_add method with 'foo.txt' and 'bar.txt' as a single string