diff --git a/tests/test_models.py b/tests/test_models.py index e69de29bb..84d04f6d7 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -0,0 +1,10 @@ +import unittest +from aider.models import Model + +class TestModels(unittest.TestCase): + def test_max_context_tokens(self): + model = Model('gpt-3.5') + self.assertEqual(model.max_context_tokens, 4*1024) + +if __name__ == '__main__': + unittest.main()