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

use pytest.ini testpaths to order testing

Browse source
This commit is contained in:
Paul Gauthier 2024-07-16 10:33:42 +01:00
parent 2d09af61ba
commit 896e79bcd1
16 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/basic/test_models.py Normal file
View file

@ -0,0 +1,28 @@
import unittest
from aider.models import Model
class TestModels(unittest.TestCase):
def test_max_context_tokens(self):
model = Model("gpt-3.5-turbo")
self.assertEqual(model.info["max_input_tokens"], 16385)
model = Model("gpt-3.5-turbo-16k")
self.assertEqual(model.info["max_input_tokens"], 16385)
model = Model("gpt-3.5-turbo-1106")
self.assertEqual(model.info["max_input_tokens"], 16385)
model = Model("gpt-4")
self.assertEqual(model.info["max_input_tokens"], 8 * 1024)
model = Model("gpt-4-32k")
self.assertEqual(model.info["max_input_tokens"], 32 * 1024)
model = Model("gpt-4-0613")
self.assertEqual(model.info["max_input_tokens"], 8 * 1024)
if __name__ == "__main__":
unittest.main()