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

aider: Modified setUp method to patch check_model_availability with MagicMock object that always returns True.

Browse source 
# Context:
USER: the patch in `setUp` is not correcting forcing a return of true
ASSISTANT: To ensure that the patch in `setUp` correctly forces a return of `True`, you need to modify the `setUp` method in the `TestMain` class in the `tests/test_main.py` file. Here's the change:

```python
tests/test_main.py
<<<<<<< ORIGINAL
    def setUp(self):
        os.environ["OPENAI_API_KEY"] = "deadbeef"
        self.original_cwd = os.getcwd()
        self.tempdir = tempfile.mkdtemp()
        os.chdir(self.tempdir)

        with patch("aider.main.Coder.check_model_availability") as self.mock_check:
            self.mock_check.return_value = True
        super().setUp()
=======
    def setUp(self):
        os.environ["OPENAI_API_KEY"] = "deadbeef"
        self.original_cwd = os.getcwd()
        self.tempdir = tempfile.mkdtemp()
        os.chdir(self.tempdir)

        self.mock_check = MagicMock(return_value=True)
        with patch("aider.main.Coder.check_model_availability", new=self.mock_check):
            super().setUp()
>>>>>>> UPDATED
```

This change ensures that the `check_model_availability` method is patched with a `MagicMock` object that always returns `True`.
This commit is contained in:
Paul Gauthier 2023-06-01 08:50:55 -07:00
parent 4774f0518c
commit 99693fe614
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_main.py
View file

@ -21,9 +21,9 @@ class TestMain(TestCase):
self.tempdir = tempfile.mkdtemp()
os.chdir(self.tempdir)
with patch("aider.main.Coder.check_model_availability") as self.mock_check:
self.mock_check.return_value = True
super().setUp()
self.mock_check = MagicMock(return_value=True)
with patch("aider.main.Coder.check_model_availability", new=self.mock_check):
super().setUp()
def tearDown(self):
os.chdir(self.original_cwd)