mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-23 13:54:59 +00:00
test: add test for sanity check with bogus editor model
This commit is contained in:
parent
7e55a8f684
commit
93cb615ffd
1 changed files with 2 additions and 2 deletions
|
@ -1,7 +1,7 @@
|
||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import MagicMock, patch
|
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_model
|
||||||
|
|
||||||
|
|
||||||
class TestModels(unittest.TestCase):
|
class TestModels(unittest.TestCase):
|
||||||
|
@ -74,7 +74,7 @@ class TestModels(unittest.TestCase):
|
||||||
# Set up mock to return False for main model and True (problem) for editor model
|
# Set up mock to return False for main model and True (problem) for editor model
|
||||||
mock_sanity_check_model.side_effect = [False, True]
|
mock_sanity_check_model.side_effect = [False, True]
|
||||||
|
|
||||||
result = models.sanity_check_models(mock_io, main_model)
|
result = sanity_check_models(mock_io, main_model)
|
||||||
|
|
||||||
self.assertTrue(
|
self.assertTrue(
|
||||||
result
|
result
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue