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

do not package tests/

Browse source
This commit is contained in:
Paul Gauthier 2024-07-15 08:40:32 +01:00
parent 7ece31c08a
commit 0860e7b30d
15 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/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()