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

refactor: update test to use ModelInfoManager instead of get_model_info

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-10-30 13:21:43 -07:00
parent a3d78e0944
commit a045bda171
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/basic/test_models.py
View file

@ -1,12 +1,13 @@
import unittest
from unittest.mock import ANY, MagicMock, patch
from aider.models import Model, get_model_info, sanity_check_model, sanity_check_models
from aider.models import Model, ModelInfoManager, sanity_check_model, sanity_check_models
class TestModels(unittest.TestCase):
def test_get_model_info_nonexistent(self):
info = get_model_info("non-existent-model")
manager = ModelInfoManager()
info = manager.get_model_info("non-existent-model")
self.assertEqual(info, {})
def test_max_context_tokens(self):