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

fix: correct imports and model usage in test_models.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-10-02 11:19:38 -07:00
parent 1a355cbf74
commit 21adf405a0
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_models.py
View file

@ -1,7 +1,7 @@
import unittest
from unittest.mock import MagicMock, patch
from aider.models import Model, get_model_info, sanity_check_model, sanity_check_models
from aider.models import Model, get_model_info, sanity_check_models
class TestModels(unittest.TestCase):
@ -64,10 +64,10 @@ class TestModels(unittest.TestCase):
def test_sanity_check_models_bogus_editor(self):
mock_io = MagicMock()
main_model = models.Model("gpt-4")
main_model.editor_model = models.Model("bogus-model")
main_model = Model("gpt-4")
main_model.editor_model = Model("bogus-model")
result = models.sanity_check_models(mock_io, main_model)
result = sanity_check_models(mock_io, main_model)
self.assertTrue(
result